MqttMessageSender.java
3.54 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package com.diligrp.mqtt.integration.handler;
import com.diligrp.mqtt.core.event.SendFailEvent;
import com.diligrp.mqtt.core.event.SendSuccessEvent;
import com.diligrp.mqtt.core.exception.MqttServiceException;
import com.diligrp.mqtt.core.model.SendModel;
import com.diligrp.mqtt.core.service.MqttMessageService;
import com.diligrp.mqtt.core.type.CharsetType;
import com.diligrp.mqtt.core.util.JsonUtils;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.integration.support.MessageBuilder;
import org.springframework.messaging.MessageChannel;
import org.springframework.messaging.MessageHeaders;
import org.springframework.stereotype.Service;
import org.springframework.util.MimeTypeUtils;
import java.util.Optional;
/**
* MQTT消息发送服务
*/
@Slf4j
@Service
public class MqttMessageSender implements MqttMessageService {
public static final String MQTT_TOPIC = "mqtt_topic";
public static final String MQTT_QOS = "mqtt_qos";
public static final String MQTT_RETAINED = "mqtt_retained";
@Resource
private MessageChannel mqttOutboundChannel;
@Resource
private ApplicationEventPublisher applicationEventPublisher;
@Override
public Boolean sendMessage(SendModel sendModel) {
Optional.ofNullable(sendModel).orElseThrow(()->new MqttServiceException("消息结构体不能为空"));
Optional.ofNullable(sendModel.getTopic()).filter(e -> !e.isEmpty()).orElseThrow(()->new MqttServiceException("【消息主题】不能为空"));
Optional.ofNullable(sendModel.getPayload()).orElseThrow(()->new MqttServiceException("【消息内容】不能为空"));
Optional.ofNullable(sendModel.getQos()).filter(e -> e >= 0 && e <= 2).orElseThrow(()->new MqttServiceException("【QOS编码】错误"));
Optional.ofNullable(sendModel.getRetained()).orElseThrow(()->new MqttServiceException("【消息是否保留】不能为空"));
Optional.ofNullable(sendModel.getCharsetType()).orElseThrow(()->new MqttServiceException("【发送消息字符集】不能为空"));
Optional.ofNullable(sendModel.getSendEventType()).orElseThrow(()->new MqttServiceException("【发送事件标识】不能为空"));
try {
String payload = JsonUtils.toJsonString(sendModel.getPayload());
byte[] bytes = CharsetType.of(sendModel.getCharsetType()).getBytes(payload);
var message = MessageBuilder.withPayload(bytes)
.setHeader(MessageHeaders.CONTENT_TYPE, MimeTypeUtils.TEXT_PLAIN)
.setHeader(MQTT_TOPIC, sendModel.getTopic())
.setHeader(MQTT_QOS, sendModel.getQos())
.setHeader(MQTT_RETAINED, sendModel.getRetained())
.build();
boolean result = mqttOutboundChannel.send(message);
Optional.of(result).filter(e -> e).ifPresentOrElse(send -> {
log.info("消息发送成功 - :{}", JsonUtils.toJsonString(message));
applicationEventPublisher.publishEvent(new SendSuccessEvent(this, sendModel));
}, () -> {
log.error("消息发送失败 - :{}", JsonUtils.toJsonString(message));
applicationEventPublisher.publishEvent(new SendFailEvent(this, sendModel));
});
return result;
} catch (Exception e) {
log.error("发送Mqtt消息失败", e);
applicationEventPublisher.publishEvent(new SendFailEvent(this, sendModel));
}
return false;
}
}