springboot整合mqtt進行消息的發佈及訂閱

一、.導入maven包

		<!--mqtt-->
        <dependency>
            <groupId>org.springframework.integration</groupId>
            <artifactId>spring-integration-mqtt</artifactId>
        </dependency>

二、添加mqtt的配置

#mqtt的配置
mqtt:
  server:
    url: 192.168.137.59,192.168.137.4
    port: 1883
    username: admin
    password: 111111
  client:
    consumerId: consumerCo
    publishId: publishCo
  default:
    topic: topic
    completionTimeout: 3000

三、mqtt客戶端配置

MqttConfig訂閱消息（動態添加topic，不用寫死）：

import com.cecjx.common.utils.RedisUtil;
import com.cecjx.device.domain.TopicDO;
import lombok.extern.slf4j.Slf4j;
import org.eclipse.paho.client.mqttv3.MqttConnectOptions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.integration.annotation.ServiceActivator;
import org.springframework.integration.channel.DirectChannel;
import org.springframework.integration.core.MessageProducer;
import org.springframework.integration.mqtt.core.DefaultMqttPahoClientFactory;
import org.springframework.integration.mqtt.core.MqttPahoClientFactory;
import org.springframework.integration.mqtt.inbound.MqttPahoMessageDrivenChannelAdapter;
import org.springframework.integration.mqtt.outbound.MqttPahoMessageHandler;
import org.springframework.integration.mqtt.support.DefaultPahoMessageConverter;
import org.springframework.messaging.MessageChannel;
import org.springframework.messaging.MessageHandler;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;

/**
 * @author huangquanguang
 * @date 2020/1/3 16:33
 * @description mqtt配置類
 * outbound 是發送消息到mqtt broker   inbound是接收mqtt消息進行處理
 */
@Slf4j
@Configuration
public class MqttConfig {

    @Autowired
    private MongoTemplate mongoTemplate;
    @Value("${mqtt.server.url}")
    private String url = "192.168.137.1";
    @Value("${mqtt.server.port}")
    private String port = "1883";
    @Value("${mqtt.server.username}")
    private String username = "admin";
    @Value("${mqtt.server.password}")
    private String password = "admin";
    @Value("${mqtt.client.consumerId}")
    private String consumerId = "consumerClient";
    @Value("${mqtt.client.publishId}")
    private String publishId = "publishClient";
    @Value("${mqtt.default.topic}")
    private String topic = "topic";
    @Value("${mqtt.default.completionTimeout}")
    private Integer completionTimeout = 3000;

    /**
     * @author huangquanguang
     * @date 2020/1/3 16:34
     * @description 創建MqttPahoClientFactory，設置MQTT Broker連接屬性，如果使用SSL驗證，也在這裏設置。
     */
    @Bean
    public MqttPahoClientFactory mqttClientFactory() {
        DefaultMqttPahoClientFactory factory = new DefaultMqttPahoClientFactory();
        MqttConnectOptions options = new MqttConnectOptions();
        options.setUserName(username);
        options.setPassword(password.toCharArray());
        //多個服務器地址時處理
        String[] strings = url.split(",");
        String[] serverUrls = new String[strings.length];
        if (strings.length > 0) {
            for (int i = 0; i < strings.length; i++) {
                String serverUrl = "tcp://" + strings[i] + ":" + port;
                serverUrls[i] = serverUrl;
            }
        }
        options.setServerURIs(serverUrls);
        factory.setConnectionOptions(options);
        return factory;
    }


    //--------------------------------- 接收消息部分 ------------------------------------------

    /**
     * @param
     * @return adapter
     * 接收消息 配置及訂閱消息
     * @author huangquanguang
     * @date 2020/1/3 16:34
     */
    @Bean
    public MessageProducer inbound() {
        //這裏要設置接收的topic,要寫成動態添加的方式，後續新增設備時不用設置
        //獲取所有要訂閱的topic
//        List<TopicDO> topicDOS = topicService.listByMap(map);
        List<TopicDO> topicDOS = new ArrayList<>();
        //啓動項目執行：把所有當前主題保存到redis緩存
        RedisUtil redisUtil = new RedisUtil();
        redisUtil.addObject("topics", topicDOS);
        String[] topics = new String[topicDOS.size()];
        if (topicDOS != null && topicDOS.size() > 0) {
            for (int i = 0; i < topicDOS.size(); i++) {
                topics[i] = topicDOS.get(i).getTopic();
            }
        }
        MqttPahoMessageDrivenChannelAdapter adapter = new MqttPahoMessageDrivenChannelAdapter(consumerId,
                mqttClientFactory(), topics);
        adapter.setCompletionTimeout(completionTimeout);
        adapter.setConverter(new DefaultPahoMessageConverter());
        // 設置服務質量
        // 0 最多一次，數據可能丟失;
        // 1 至少一次，數據可能重複;
        // 2 只有一次，有且只有一次;最耗性能
        adapter.setQos(1);
        adapter.setOutputChannel(mqttInputChannel());
        return adapter;
    }

    @Bean
    public MessageChannel mqttInputChannel() {
        return new DirectChannel();
    }

    /**
     * @author huangquanguang
     * @date 2020/1/3 16:35
     * @description 消費消息
     * ServiceActivator註解表明當前方法用於處理MQTT消息，inputChannel參數指定了用於消費消息的channel。
     */
    @Bean
    @ServiceActivator(inputChannel = "mqttInputChannel")
    public MessageHandler handler() {
        return message -> {
            String payload = message.getPayload().toString();
            String topic = message.getHeaders().get("mqtt_receivedTopic").toString();
            //用線程池處理訂閱到的消息
            try {
                dealMessageByThreadPool(payload, topic);
            } catch (ExecutionException e) {
                e.printStackTrace();
                log.error("處理訂閱出錯：" + e.getMessage());
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        };
    }


    /**
     * @param
     * @return 多線程處理收到的傳感器推送, 線程處理參考：
     * https://www.bbsmax.com/A/mo5kxvPvdw/（這裏註釋掉線程的使用，保持數據的一致性
     * 在處理某一次上報的數據時可以用多線程，提高效率，不能每次訂閱數據都起一個線程讓它單獨處理）
     * 要根據廠商提交的設備id及主題先把所有設備的主題管理起來，一個設備對應一個或多個主題
     * @author huangquanguang
     * @date 2020/2/21 11:54
     */
    public void dealMessageByThreadPool(String payload, String topic) throws ExecutionException, InterruptedException {

        //保存mongodb部分
        log.info(": 處理消息 " + payload);
        //通用保存到mongodb的方法（集合存在則新增，不存在則先創建再新增）
        mongoTemplate.insert(payload, topic);

    }

//--------------------------------發送消息部分-------------------------------------------

    /**
     * @author huangquanguang
     * @date 2020/1/3 16:37
     * @description 發送消息
     */
    @Bean
    public MessageChannel mqttOutboundChannel() {
        return new DirectChannel();
    }

    /*****
     * 發送消息和消費消息Channel可以使用相同MqttPahoClientFactory
     * @return
     */
    @Bean
    @ServiceActivator(inputChannel = "mqttOutboundChannel")
    public MessageHandler outbound() {
        // 在這裏進行mqttOutboundChannel的相關設置
        MqttPahoMessageHandler messageHandler =
                new MqttPahoMessageHandler(publishId, mqttClientFactory());
        messageHandler.setAsync(true); //如果設置成true，發送消息時將不會阻塞。
        messageHandler.setDefaultTopic(topic);
        return messageHandler;
    }


}

四、發佈消息

1.MqttController ：

import com.cecjx.common.annotation.Log;
import com.cecjx.common.utils.R;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;

import java.util.Map;

/**
 * @author huangquanguang
 * @date 2020/1/3 16:30
 * @description mqtt發佈消息
 */
@Controller
@RequestMapping(value = "mqtt")
public class MqttController {

    @Autowired
    private MqttGateway mqttGateway;

    /**
     * @param topic：主題 message：內容
     * @return send message
     * @author huangquanguang
     * @date 2020/1/3 16:31
     */
    @RequestMapping("/send")
    @ResponseBody
    public R send(@RequestParam Map params) {
        // 發送消息到指定topic
        try {
            String topic = params.get("topic").toString();
            String message = params.get("message").toString();
            mqttGateway.sendToMqtt(topic, message);
            return R.ok().put("message","send message :"+message);
        } catch (Exception e) {
            e.printStackTrace();
            return R.error();
        }
    }
}

2.MqttGateway ：

import org.springframework.integration.annotation.MessagingGateway;
import org.springframework.integration.mqtt.support.MqttHeaders;
import org.springframework.messaging.handler.annotation.Header;
import org.springframework.stereotype.Component;

/**
 * @author huangquanguang
 * @date 2020/1/3 16:41
 * @description mqtt網關
 * 定義重載方法，用於消息發送
 */
@Component
@MessagingGateway(defaultRequestChannel = "mqttOutboundChannel")
public interface MqttGateway {

    // 定義重載方法，用於消息發送
    void sendToMqtt(String payload);

    // 指定topic進行消息發送
    void sendToMqtt(@Header(MqttHeaders.TOPIC) String topic, String payload);

    void sendToMqtt(@Header(MqttHeaders.TOPIC) String topic, @Header(MqttHeaders.QOS) int qos, String payload);
}

五、測試

1.請求MqttController 的send接口，發佈消息
2.在MqttConfig訂閱消息中可以打印訂閱到的消息並保存到mongodb