From 8c733c9c51ebcf4fbb6b785bd9afbf11087600e8 Mon Sep 17 00:00:00 2001
From: shenyijian <1178253012@qq.com>
Date: Sat, 02 Dec 2023 16:09:37 +0800
Subject: [PATCH] 无人机 -操作消息同步至RabbitMq
---
src/main/java/com/dji/sample/component/mqtt/service/impl/MessageSenderServiceImpl.java | 42 +++++++++++++++++++++++++++++++++++++++++-
src/main/java/com/dji/sample/amap/model/ReceiverData.java | 11 +++++++++++
src/main/java/com/dji/sample/component/rabbitmq/config/MqttMsgProxyProducer.java | 7 +++++++
3 files changed, 59 insertions(+), 1 deletions(-)
diff --git a/src/main/java/com/dji/sample/amap/model/ReceiverData.java b/src/main/java/com/dji/sample/amap/model/ReceiverData.java
new file mode 100644
index 0000000..8804348
--- /dev/null
+++ b/src/main/java/com/dji/sample/amap/model/ReceiverData.java
@@ -0,0 +1,11 @@
+package com.dji.sample.amap.model;
+
+import lombok.Data;
+
+@Data
+public class ReceiverData {
+ private String tid;
+ private String bid;
+ private Object receiver;
+ private String error;
+}
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 237713c..ba3211a 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,17 @@
package com.dji.sample.component.mqtt.service.impl;
+import com.dji.sample.amap.model.ReceiverData;
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.dji.sample.component.rabbitmq.config.MqttMsgProxyProducer;
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.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
@@ -31,14 +34,42 @@
@Autowired
private ObjectMapper mapper;
+ @Value("${spring.rabbitmq.is-open}")
+ private Boolean sendRabbitMQ;
+
+ @Autowired
+ private MqttMsgProxyProducer mqttMsgProxyProducer;
+
public void publish(String topic, CommonTopicResponse response) {
this.publish(topic, 1, response);
+ }
+ public void sendRabbitMQ(String topic, Object data) {
+ try {
+ if (sendRabbitMQ) {
+ sendRabbitMQ(topic.replace("/", ".") + ".response", mapper.writeValueAsBytes(data));
+ }
+ } catch (Exception e) {
+ log.error("消息发送失败2：", e);
+ }
+ }
+
+ private void sendRabbitMQ(String topic, byte[] bytes) {
+ //发送操作消息至mqtt同时发送到rabbitMQ
+ try {
+ if (sendRabbitMQ) {
+ mqttMsgProxyProducer.publish(topic, bytes);
+ }
+ } catch (Exception e) {
+ log.error("消息发送失败：", e);
+ }
}
public void publish(String topic, int qos, CommonTopicResponse response) {
try {
log.info("send topic: {}, payload: {}", topic, response.toString());
- messageGateway.publish(topic, mapper.writeValueAsBytes(response), qos);
+ byte[] bytes = mapper.writeValueAsBytes(response);
+ sendRabbitMQ(topic, bytes);
+ messageGateway.publish(topic, bytes, qos);
} catch (JsonProcessingException e) {
log.info("Failed to publish the message. {}", response.toString());
e.printStackTrace();
@@ -51,6 +82,9 @@
public <T> T publishWithReply(Class<T> clazz, String topic, CommonTopicResponse response, int retryTime) {
AtomicInteger time = new AtomicInteger(0);
+ ReceiverData receiverData = new ReceiverData();
+ receiverData.setTid(response.getTid());
+ receiverData.setBid(response.getBid());
// Retry three times
while (time.getAndIncrement() <= retryTime) {
this.publish(topic, response);
@@ -62,15 +96,21 @@
// Need to match tid and bid.
if (Objects.nonNull(receiver) && receiver.getTid().equals(response.getTid()) &&
receiver.getBid().equals(response.getBid())) {
+ receiverData.setReceiver(receiver.getData());
if (clazz.isAssignableFrom(receiver.getData().getClass())) {
+ sendRabbitMQ(topic, receiverData);
return receiver.getData();
}
+ receiverData.setError("类型转换异常：" + clazz.getTypeName());
+ sendRabbitMQ(topic, receiverData);
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());
}
+ receiverData.setError("没有收到消息回复");
+ sendRabbitMQ(topic, receiverData);
throw new RuntimeException("没有收到消息回复。");
}
diff --git a/src/main/java/com/dji/sample/component/rabbitmq/config/MqttMsgProxyProducer.java b/src/main/java/com/dji/sample/component/rabbitmq/config/MqttMsgProxyProducer.java
index 5c22565..9e9ff7e 100644
--- a/src/main/java/com/dji/sample/component/rabbitmq/config/MqttMsgProxyProducer.java
+++ b/src/main/java/com/dji/sample/component/rabbitmq/config/MqttMsgProxyProducer.java
@@ -44,6 +44,13 @@
rabbitTemplate.send(EXCHANGE_NAME,routingKey, message);
}
+ public void publish(String routingKey, byte[] content) {
+ Message message = MessageBuilder.withBody(content)
+ .setExpiration("1000")
+ .build();
+ rabbitTemplate.send(EXCHANGE_NAME,routingKey, message);
+ }
+
/* *
* 使用json序列化机制，进行消息转换
* @return -*/
--
Gitblit v1.9.3