Java如何使用ReentrantLock实现长轮询 Java如何使用ReentrantLock实现长轮

Java代码 1. ReentrantLock
加锁阻塞，一个condition对应一个线程，以便于唤醒时使用该condition一定会唤醒该线程

/*** 获取探测点数据,长轮询实现* @param messageId* @return*/public JSONObject getToutData(String messageId) {Message message = toutMessageCache.get(messageId); if (message == null) {// 等待lock.lock(); try {Condition condition = lock.newCondition(); conditionMap.put(messageId + "_data", condition); condition.await(CONNECTION_HOLD_TIMEOUT, TimeUnit.SECONDS); // 等待60s} catch (InterruptedException e) {// 等待超时， do nothing} finally {lock.unlock(); }}// 再次尝试获取message = toutMessageCache.get(messageId); if (message == null) {// 如果还没有，返回空对象return null; }byte[] bytes = message.getDataBytes(); if (bytes == null) {return null; }String resStr = new String(bytes, StandardCharsets.UTF_8); //log.info("resStr: {}", resStr); JSONObject resObj; try {resObj = new JSONObject(resStr); resObj.put("invokeTime", DateUtil.format(new Date(resObj.getLong("invokeTime")), DatePattern.NORM_DATETIME_MS_PATTERN)); } catch (Exception e) {resObj = new JSONObject(); }return resObj; }

2. 回调
当异步数据返回，使用上一步的condition唤醒线程

public void callback(Message message) {String messageId = message.getId(); toutMessageCache.put(message.getId(), message); String messageDataId = messageId + "_data"; if (conditionMap.containsKey(messageDataId)) {lock.lock(); try {Condition condition = conditionMap.get(messageDataId); condition.signal(); } catch (Exception e) {e.printStackTrace(); } finally {lock.unlock(); conditionMap.remove(messageDataId); }}}

3. 唤醒
执行回调操作

public void distribute(Message message, ChannelHandlerContext ctx) {MessageType messageType = message.getMessageType(); switch (messageType) {case TOUT_DATA_RESPONSE:// 数据响应toutService.callback(message); break; }}

4. 调用
调用时，判断返回的值是否为空，如果为空，与前端约定，当返回该状态值时，应再次发起相同请求

/*** 获取探测数据(使用长轮询实现)* @param linkId* @return*/@GetMapping("/data")public ResultVO getToutData(String linkId) {JSONObject resObj = toutService.getToutData(linkId); if (resObj == null || resObj.isEmpty()) {return ResultVOUtil.error(ResultEnum.NO_MESSAGE_HOLD_CONNECTION); }return ResultVOUtil.success(resObj); }

5.前端实现
简单使用递归实现了当数据返回无效时再次发起请求

let that = thisfunction getData() {if (toutStatus === statusEnum.start) {getToutData({linkId}).then(res => {if (res.code === ERROR_CODE_OK) {that.toutData = https://www.it610.com/article/res.datatoutStatus = statusEnum.resDatathat._btnStatus()} else {getData()}})} } // 递归循环调用 getData()

【Java如何使用ReentrantLock实现长轮询】以上就是如何使用ReentrantLock实现长轮询的详细内容，更多关于ReentrantLock长轮询的资料请关注脚本之家其它相关文章！