Java中如何使用ReentrantLock實(shí)現(xiàn)長輪詢

這篇文章主要介紹“Java中如何使用ReentrantLock實(shí)現(xiàn)長輪詢”，在日常操作中，相信很多人在Java中如何使用ReentrantLock實(shí)現(xiàn)長輪詢問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”Java中如何使用ReentrantLock實(shí)現(xiàn)長輪詢”的疑惑有所幫助！接下來，請跟著小編一起來學(xué)習(xí)吧！

Java代碼

1. ReentrantLock

加鎖阻塞，一個(gè)condition對應(yīng)一個(gè)線程，以便于喚醒時(shí)使用該condition一定會喚醒該線程

/**
     * 獲取探測點(diǎn)數(shù)據(jù),長輪詢實(shí)現(xiàn)
     * @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) {
                // 等待超時(shí)， 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. 回調(diào)

當(dāng)異步數(shù)據(jù)返回，使用上一步的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. 喚醒

執(zhí)行回調(diào)操作

public void distribute(Message message, ChannelHandlerContext ctx) {
   MessageType messageType = message.getMessageType();
   switch (messageType) {
       case TOUT_DATA_RESPONSE:
           // 數(shù)據(jù)響應(yīng)
           toutService.callback(message);
           break;
   }

}

4. 調(diào)用

調(diào)用時(shí)，判斷返回的值是否為空，如果為空，與前端約定，當(dāng)返回該狀態(tài)值時(shí)，應(yīng)再次發(fā)起相同請求

/**
* 獲取探測數(shù)據(jù)(使用長輪詢實(shí)現(xiàn))
* @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.前端實(shí)現(xiàn)

簡單使用遞歸實(shí)現(xiàn)了當(dāng)數(shù)據(jù)返回?zé)o效時(shí)再次發(fā)起請求

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()
             }
         })
     }
 }

 // 遞歸循環(huán)調(diào)用
 getData()

到此，關(guān)于“Java中如何使用ReentrantLock實(shí)現(xiàn)長輪詢”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注億速云網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬?shí)用的文章！