MqttMessageSender.java 2.31 KB
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.model.SendModel;
import com.diligrp.mqtt.core.service.MqttMessageService;
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 void sendMessage(SendModel sendModel) {
        try {
            String payload = JsonUtils.toJsonString(sendModel.getPayload());
            var message = MessageBuilder.withPayload(payload)
                    .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("消息发送成功 - :{}", message);
                applicationEventPublisher.publishEvent(new SendSuccessEvent(this, sendModel));
            }, () -> {
                log.error("消息发送失败 - :{}", message);
                applicationEventPublisher.publishEvent(new SendFailEvent(this, sendModel));
            });
        } catch (Exception e) {
            log.error("发送Mqtt消息失败", e);
            applicationEventPublisher.publishEvent(new SendFailEvent(this, sendModel));
        }
    }
}