From bf4990ffb53e2a0e7b2ec0c7a8a054bdf8a57db5 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Thu, 16 Nov 2023 10:02:13 +0800
Subject: [PATCH] 无人机 -设备分页新增保护区ID查询条件

---
 src/main/java/com/dji/sample/component/mqtt/service/impl/MessageSenderServiceImpl.java |  133 +++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 117 insertions(+), 16 deletions(-)

diff --git a/src/main/java/com/dji/sample/component/mqtt/service/impl/MessageSenderServiceImpl.java b/src/main/java/com/dji/sample/component/mqtt/service/impl/MessageSenderServiceImpl.java
index 2510ead..237713c 100644
--- a/src/main/java/com/dji/sample/component/mqtt/service/impl/MessageSenderServiceImpl.java
+++ b/src/main/java/com/dji/sample/component/mqtt/service/impl/MessageSenderServiceImpl.java
@@ -1,14 +1,20 @@
 package com.dji.sample.component.mqtt.service.impl;
 
-import com.dji.sample.component.mqtt.model.CommonTopicResponse;
+import com.dji.sample.component.mqtt.model.*;
 import com.dji.sample.component.mqtt.service.IMessageSenderService;
 import com.dji.sample.component.mqtt.service.IMqttMessageGateway;
-import com.fasterxml.jackson.annotation.JsonInclude;
 import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.TypeMismatchException;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
+
+import java.util.Objects;
+import java.util.UUID;
+import java.util.concurrent.atomic.AtomicInteger;
 
 /**
  * @author sean.zhou
@@ -22,29 +28,124 @@
     @Autowired
     private IMqttMessageGateway messageGateway;
 
-    public void publish(String topic, CommonTopicResponse response) {
-        try {
-            ObjectMapper mapper = new ObjectMapper();
-            // Only parameters whose value is not null will be serialised.
-            mapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
+    @Autowired
+    private ObjectMapper mapper;
 
-            messageGateway.publish(topic, mapper.writeValueAsBytes(response));
-        } catch (JsonProcessingException e) {
-            log.info("Failed to publish the message. {}", response.toString());
-            e.printStackTrace();
-        }
+    public void publish(String topic, CommonTopicResponse response) {
+        this.publish(topic, 1, response);
     }
 
     public void publish(String topic, int qos, CommonTopicResponse response) {
         try {
-            ObjectMapper mapper = new ObjectMapper();
-            // Only parameters whose value is not null will be serialised.
-            mapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
-
+            log.info("send topic: {}, payload: {}", topic, response.toString());
             messageGateway.publish(topic, mapper.writeValueAsBytes(response), qos);
         } catch (JsonProcessingException e) {
             log.info("Failed to publish the message. {}", response.toString());
             e.printStackTrace();
         }
     }
+
+    public <T> T publishWithReply(Class<T> clazz, String topic, CommonTopicResponse response) {
+        return this.publishWithReply(clazz, topic, response, 2);
+    }
+
+    public <T> T publishWithReply(Class<T> clazz, String topic, CommonTopicResponse response, int retryTime) {
+        AtomicInteger time = new AtomicInteger(0);
+        // Retry three times
+        while (time.getAndIncrement() <= retryTime) {
+            this.publish(topic, response);
+
+            Chan<CommonTopicReceiver<T>> chan = Chan.getInstance();
+            // If the message is not received in 0.5 seconds then resend it again.
+            CommonTopicReceiver<T> receiver = chan.get(response.getTid());
+
+            // Need to match tid and bid.
+            if (Objects.nonNull(receiver) && receiver.getTid().equals(response.getTid()) &&
+                    receiver.getBid().equals(response.getBid())) {
+                if (clazz.isAssignableFrom(receiver.getData().getClass())) {
+                    return receiver.getData();
+                }
+                throw new TypeMismatchException(receiver.getData(), clazz);
+            }
+            // It must be guaranteed that the tid and bid of each message are different.
+            response.setBid(UUID.randomUUID().toString());
+            response.setTid(UUID.randomUUID().toString());
+        }
+        throw new RuntimeException("没有收到消息回复。");
+    }
+
+    @Override
+    public <T> ServiceReply<T> publishServicesTopic(TypeReference<T> clazz, String sn, String method, Object data, String bid) {
+        String topic = TopicConst.THING_MODEL_PRE + TopicConst.PRODUCT + sn + TopicConst.SERVICES_SUF;
+        ServiceReply reply = this.publishWithReply(ServiceReply.class, topic,
+                CommonTopicResponse.builder()
+                        .tid(UUID.randomUUID().toString())
+                        .bid(StringUtils.hasText(bid) ? bid : UUID.randomUUID().toString())
+                        .timestamp(System.currentTimeMillis())
+                        .method(method)
+                        .data(Objects.requireNonNullElse(data, ""))
+                        .build());
+        if (Objects.isNull(clazz)) {
+            return reply;
+        }
+        // put together in "output"
+        if (Objects.nonNull(reply.getInfo())) {
+            reply.setOutput(mapper.convertValue(reply.getInfo(), clazz));
+        }
+        if (Objects.nonNull(reply.getOutput())) {
+            reply.setOutput(mapper.convertValue(reply.getOutput(), clazz));
+        }
+        return reply;
+    }
+
+    @Override
+    public ServiceReply publishServicesTopic(String sn, String method, Object data, String bid) {
+        return this.publishServicesTopic(null, sn, method, data, bid);
+    }
+
+    @Override
+    public <T> ServiceReply<T> publishServicesTopic(TypeReference<T> clazz, String sn, String method, Object data) {
+        return this.publishServicesTopic(clazz, sn, method, data, null);
+    }
+
+    @Override
+    public ServiceReply publishServicesTopic(String sn, String method, Object data) {
+        return this.publishServicesTopic(null, sn, method, data, null);
+    }
+
+    @Override
+    public  <T> ServiceReply<T> publishRequestsTopic(TypeReference<T> clazz, String sn, String method, Object data, String bid) {
+        String topic = TopicConst.THING_MODEL_PRE + TopicConst.PRODUCT + sn + TopicConst.REQUESTS_SUF;
+        ServiceReply reply = this.publishWithReply(ServiceReply.class, topic,
+                CommonTopicResponse.builder()
+                        .tid(UUID.randomUUID().toString())
+                        .bid(StringUtils.hasText(bid) ? bid : UUID.randomUUID().toString())
+                        .timestamp(System.currentTimeMillis())
+                        .gateway(sn)
+                        .method(method)
+                        .data(Objects.requireNonNullElse(data, ""))
+                        .build());
+        if (Objects.isNull(clazz)) {
+            return reply;
+        }
+        // put together in "output"
+        if (Objects.nonNull(reply.getInfo())) {
+            reply.setOutput(mapper.convertValue(reply.getInfo(), clazz));
+        }
+        if (Objects.nonNull(reply.getOutput())) {
+            reply.setOutput(mapper.convertValue(reply.getOutput(), clazz));
+        }
+        return reply;
+    }
+
+    @Override
+    public ServiceReply publishRequestsTopic(String sn, String method, Object data, String bid) {
+        return this.publishRequestsTopic(null, sn, method, data, bid);
+    }
+
+    @Override
+    public ServiceReply publishRequestsTopic(String sn, String method, Object data) {
+        return this.publishRequestsTopic(null, sn, method, data, null);
+    }
+
 }
\ No newline at end of file

--
Gitblit v1.9.3