摘要：背景通過接口實現調用發送數據，接口返回值為發送數據的對應結果。接口為同步阻塞，為異步回調方式。接收數據回調接收到數據后，通過閉鎖釋放阻塞的線程，同時設置結果返回給調用者
背景
通過HTTP接口實現調用MQTT Client發送數據，HTTP接口返回值為MQTT Client發送數據的對應結果。 HTTP接口為同步阻塞，MQTT Client 為異步回調方式。
如何實現在HTTP接口中調用MQTT Client發送數據后，能夠阻塞等待MQTT返回結果，然后將結果返回？
解決方法
CountDownLatch + Callbale+FutureTask
1.CountDownLatch作用
CountDownLatch實現在MQTT Client 發送數據后 到接收數據后這段時間的阻塞。
HTTP每次請求，新建一個CountDownLatch，然后將CountDownLatch作為值和deviceId作為KEY保存到Map中，
調用MQTT Client 發送數據后，countDownLatch.await(),進行同步等待
在MQTT Client接收數據的回調方法中更加deviceId取出CountDwonLatch然后計數減一
2.Callbale+FutureTask作用
將調用MQTT Client發送數據的過程，封裝成Callable,投遞發送任務時，通過返回的FutureTask的get()方法，
同步阻塞，直到結果返回。
關鍵代碼
1.Map保存CountDownObj用于同步阻塞等待MQTT Client返回結果，以及將返回結果傳遞個FutureTask
private final static ConcurrentMap countDownLatchMap = new ConcurrentHashMap<>();
//線程池
private final ThreadPoolExecutor threadPoolExecutor =
new ThreadPoolExecutor(3, 5, 30, TimeUnit.SECONDS, new ArrayBlockingQueue<>(100), runnable -> {
Thread thread = new Thread(runnable, "mqtt thread");
return thread;
});
2.HTTP API 調用的發送MQTT 消息數據的接口
/**
* HTTP API 調用的發送MQTT 消息數據的接口
* 同步阻塞
*/
public Integer send(Long packageId, String deviceId) throws Exception {
......
FutureTask futureTask = sendTask(publishDto));
return futureTask.get()
}
3.投遞發送MQTT指令的task方法
/**
* 投遞MQTT發送指令任務
* 同步阻塞
*/
private FutureTask sendTask(PublishDto publishDto) throws Exception {
FutureTask futureTask = new FutureTask<>(new GetDatapointValueCallable(publishDto));
threadPoolExecutor.execute(futureTask);
//阻塞線程
return futureTask;
}
4.封裝CountDownLatch 和 Integer的對象，用于CountDownLatch阻塞控制和返回結果
/**
* 封裝CountDownLatch 和 Integer
* 用于CountDownLatch阻塞控制和返回結果
*/
private class CountDownObj {
private final CountDownLatch countDownLatch;
private volatile Integer value;
private CountDownObj(CountDownLatch countDownLatch) {
this.countDownLatch = countDownLatch;
}
public CountDownLatch getCountDownLatch() {
return countDownLatch;
}
public Integer getValue() {
return value;
}
public void setValue(Integer value) {
this.value = value;
}
}
5.具體發送MQTT數據的Callbale線程Task，會新建CountDownLatch，并通過CountDownLatch.await()方法阻塞，直到MQTT回調接收到數據或者超時。
/**
* 發送MQTT消息的任務Callable
*/
private class GetDatapointValueCallable implements Callable {
private final PublishDto publishDto;
GetDatapointValueCallable(PublishDto publishDto) {
this.publishDto = publishDto;
}
@Override
public Integer call() throws Exception {
//mqtt client 發送數據,此處具體代碼省略
......
CountDownLatch countDownLatch = new CountDownLatch(1);
countDownLatchMap.putIfAbsent(publishDto.getDeviceId(), new CountDownObj(countDownLatch));
//阻塞，超時時間3s
countDownLatch.await(3, TimeUnit.SECONDS);
//返回mqtt指令對應的結果或者null
return countDownLatchMap.remove(publishDto.getDeviceId()).getValue();
}
}
6.MQTT接收數據回調，這里通過deviceId從MAP里面取到CountDownObj，釋放閉鎖（結束callable線程的等待）和設置MQTT返回的結果（即callable中call()返回的結果，也就是FutureTask的get()方法返回的結果）。
/**
* MQTT 接收數據回調
*/
void mqttReceiveCallback(String deviceId, String datapointId, String value) {
......
//接收到數據后，通過閉鎖釋放阻塞的線程，同時設置結果返回給調用者
CountDownObj countDownObj=countDownLatchMap.get(deviceId);
if(countDownObj!=null) {
countDownObj.setValue(Integer.parseInt(value));
countDownObj.getCountDownLatch().countDown();
}
.......
}
文章版權歸作者所有，未經允許請勿轉載,若此文章存在違規行為，您可以聯系管理員刪除。
轉載請注明本文地址：http://specialneedsforspecialkids.com/yun/68941.html
