Java如何使用ReentrantLock实现长轮询

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

Java代码

1. ReentrantLock

加锁阻塞，一个condition对应一个线程，以便于唤醒时使用该condition一定会唤醒该线程



    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. 调用

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



@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 = this
function getData() {
     if (toutStatus === statusEnum.start) {
         getToutData({
             linkId
         }).then(res => {
             if (res.code === ERROR_CODE_OK) {
                 that.toutData = res.data
                 toutStatus = statusEnum.resData
                 that._btnStatus()
             } else {
                 getData()
             }
         })
     }
 }

 // 递归循环调用
 getData()

以上就是如何使用ReentrantLock实现长轮询的详细内容，更多关于ReentrantLock长轮询的资料请关注编程网其它相关文章！

免责声明：

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

阅读原文内容投诉