1. 程式人生 > >SpringBoot成長筆記(九)整合MQTT

SpringBoot成長筆記(九)整合MQTT

環境

使用Apollo搭建mqtt的服務進行測試

配置

pom.xml檔案

<!--MQTT -->
		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-integration</artifactId>
		</dependency>
		<dependency>
			<groupId>org.springframework.integration</groupId>
			<artifactId>spring-integration-stream</artifactId>
		</dependency>
		<dependency>
			<groupId>org.springframework.integration</groupId>
			<artifactId>spring-integration-mqtt</artifactId>
		</dependency>

配置檔案

mqtt.server=127.0.0.1
mqtt.port=8021
mqtt.clientid=zzclientid
mqtt.serverid=zzservertid
mqtt.username=admin
mqtt.password=password
mqtt.topic=v1/Gps

程式碼

生產者


/**
 * MQTT生產者
 * Created by MHm on 2018/3/9.
 */

public class MqttProducer {

    private static final MqttProducer mqttProducer = new MqttProducer();

    private MqttProducer(){

    }

    public static String HOST = "tcp://127.0.0.1:61613";
    public static String TOPIC = "v1/AlertResp";
    private static String CLIENTID = "zzservertid";

    private MqttClient client;
    private MqttTopic topic;

    private String userName = "admin";
    private String passWord = "password";

    //訊息體
    private MqttMessage message;

    private ScheduledExecutorService scheduler;

    public static MqttProducer getInstance(){
        return  mqttProducer;
    }

    public void init(Properties properties) {
        try {
            HOST = String.format("tcp://%s:%s", properties.getProperty("mqtt.server"),
                    properties.getProperty("mqtt.port"));
            CLIENTID = properties.getProperty("mqtt.serverid") + UUID.randomUUID().toString();
            this.userName = properties.getProperty("mqtt.username");
            this.passWord = properties.getProperty("mqtt.password");
            this.TOPIC = properties.getProperty("mqtt.topic");
        } catch (Exception e) {
            LogUtil.error(e.getMessage());
        }

        try {
            client = new MqttClient(HOST, CLIENTID, new MemoryPersistence());
        } catch (MqttException e) {
            e.printStackTrace();
        }
        connect();
    }

    /**
     * 連線伺服器
     */
    private void connect() {
        MqttConnectOptions options = new MqttConnectOptions();
        options.setCleanSession(false);
        options.setUserName(userName);
        options.setPassword(passWord.toCharArray());
        // 設定超時時間
        options.setConnectionTimeout(10);
        // 設定會話心跳時間
        options.setKeepAliveInterval(20);
        try {
            client.connect(options);
            topic = client.getTopic(TOPIC);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void setCallback(MqttCallback callback) {
        // 設定回撥
        client.setCallback(callback);
    }

    /**
     * 訊息釋出
     *
     * @param topic
     * @param message
     * @throws MqttPersistenceException
     * @throws MqttException
     */
    public void publish(MqttTopic topic, MqttMessage message) throws MqttPersistenceException, MqttException {
        MqttDeliveryToken token = topic.publish(message);
        token.waitForCompletion();
        LogUtil.info("message is published completely! " + token.isComplete() + " topic:" + topic.getName() +
                " Message:" + new String(message.getPayload()));
    }

    /**
     * 訊息釋出
     *
     * @param topic
     * @param message
     * @throws MqttPersistenceException
     * @throws MqttException
     */
    public void publish(String topic, String message) throws MqttPersistenceException, MqttException {
        if (!client.isConnected()){
            client.connect();
        }

        MqttTopic mqttTopic = client.getTopic(topic);
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(2);
        mqttMessage.setRetained(false);
        try {
            mqttMessage.setPayload(message.getBytes("UTF-8"));
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        MqttDeliveryToken token = mqttTopic.publish(mqttMessage);
        token.waitForCompletion();
        LogUtil.info("message is published completely! " + token.isComplete());
    }

    /**
     * 通過定時器檢查斷線重連重連
     */
    public void startReconnect() {
        scheduler = Executors.newSingleThreadScheduledExecutor();
        scheduler.scheduleAtFixedRate(new Runnable() {

            @Override
            public void run() {
                LogUtil.info("客戶端重連");
                if (!client.isConnected()) {
                    connect();
                }
            }
        }, 0 * 1000, 10 * 1000, TimeUnit.MILLISECONDS);
    }

    public static void main(String[] args) throws MqttException {
//        Properties properties = LoadProperties.getProperties("mqtt.properties");
//        MqttProducer server = MqttProducer.getInstance();
//        server.init(properties);
//
//        server.message = new MqttMessage();
//        server.message.setQos(2);
//        server.message.setRetained(true);
//        server.setCallback(new ConnectionCallback(server));
//
//        for (int i = 0; i < 10000; i++) {
//            server.message.setPayload(("給客戶端124推送的資訊" + i).getBytes());
//            server.publish(server.topic, server.message);
//            try {
//                Thread.sleep(1000);
//            } catch (InterruptedException e) {
//                e.printStackTrace();
//            }
//        }

//        LogUtil.info(server.message.isRetained() + "------ratained狀態");

//        Properties p = LoadProperties.getProperties("mqtt.properties");
//        MqttProducer server = MqttProducer.getInstance();
//        server.init(p);
//        for (int i = 0; i < 10000; i++) {
//            try {
//                server.publish(server.TOPIC, "給客戶端124推送的資訊" + i);
//            } catch (MqttException e) {
//                e.printStackTrace();
//            }
//            try {
//                Thread.sleep(1000);
//            } catch (InterruptedException e) {
//                e.printStackTrace();
//            }
//        }
    }
}

消費者


/**
 * MQTT消費者
 * Created by MHm on 2018/3/9.
 */
public class MqttConsumer {
	public String HOST = "tcp://127.0.0.1:61613";
	private String clientId = "server";
	private String topic = "v1/Gps";
    private MqttClient client;
    private MqttConnectOptions options;
    private String userName = "admin";
    private String passWord = "password";

    private ScheduledExecutorService scheduler;

	public MqttConsumer(String propertiesName) throws MqttException {
		this.initProperties(propertiesName);
        client = new MqttClient(HOST, clientId, new MemoryPersistence());
        init();
    }

	private void initProperties(String propertiesName) {
		try {
			Properties mqttProperties = LoadProperties
					.getProperties(propertiesName);
			HOST = String.format("tcp://%s:%s",
					mqttProperties.getProperty("mqtt.server"),
					mqttProperties.getProperty("mqtt.port"));
			this.userName = mqttProperties.getProperty("mqtt.username");
			this.passWord = mqttProperties.getProperty("mqtt.password");
			this.clientId = mqttProperties.getProperty("mqtt.clientid") + UUID.randomUUID().toString();
			this.topic = mqttProperties.getProperty("mqtt.topic");
		} catch (Exception e) {
			LogUtil.error(e.getMessage());
		}
	}
    private void init() {
        try {
            // MQTT的連線設定
            options = new MqttConnectOptions();
            // 設定是否清空session,這裡如果設定為false表示伺服器會保留客戶端的連線記錄,這裡設定為true表示每次連線到伺服器都以新的身份連線
            options.setCleanSession(true);
            // 設定連線的使用者名稱
            options.setUserName(userName);
            // 設定連線的密碼
            options.setPassword(passWord.toCharArray());
            // 設定超時時間 單位為秒
			options.setConnectionTimeout(100);
            // 設定會話心跳時間 單位為秒 伺服器會每隔1.5*20秒的時間向客戶端傳送個訊息判斷客戶端是否線上,但這個方法並沒有重連的機制
            options.setKeepAliveInterval(20);
			// MqttTopic topic = client.getTopic(TOPIC);
            // setWill方法,如果專案中需要知道客戶端是否掉線可以呼叫該方法。設定最終埠的通知訊息
			// options.setWill(topic, "close".getBytes(), 2, true);

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 訂閱訊息
     */
    public void connect() {
        try {
            client.connect(options);
            // 訂閱訊息
            int[] Qos = { 1 };
			String[] topic1 = { this.topic };
            client.subscribe(topic1, Qos);

        } catch (MqttSecurityException e) {
            e.printStackTrace();
        } catch (MqttException e) {
            e.printStackTrace();
        }

    }

    public void setCallback(MqttCallback callback) {
        // 設定回撥
        client.setCallback(callback);
    }

    /**
     * 通過定時器檢查斷線重連重連
     */
    public void startReconnect() {
		if (scheduler == null) {
        scheduler = Executors.newSingleThreadScheduledExecutor();
        scheduler.scheduleAtFixedRate(new Runnable() {
            @Override
            public void run() {
                if (!client.isConnected()) {
                    LogUtil.info("mqtt client start reconnect...");
                    connect();
                }
            }
        }, 0 * 1000, 10 * 1000, TimeUnit.MILLISECONDS);
		}
    }

    public String getClientId() {
        return clientId;
    }

    public void setClientId(String clientId) {
        this.clientId = clientId;
    }

    public static void main(String[] args) throws MqttException {
        MqttConsumer client = new MqttConsumer("mqtt.properties");
        client.setCallback(new PushCallback(client));
        client.connect();
    }

}

回撥類


/**
 * 客戶端訊息推送及重連
 * Created by MHm on 2018/3/9.
 */
public class PushCallback implements MqttCallback {
    private MqttConsumer consumer;

    public PushCallback(MqttConsumer consumer) {
        this.consumer = consumer;
    }

    @Override
    public void connectionLost(Throwable throwable) {
        LogUtil.warn("mqtt connection lost...");
        consumer.startReconnect();
    }

    /**
     * @param topic
     * @param mqttMessage
     * @throws Exception
     */
    @Override
    public void messageArrived(String topic, MqttMessage mqttMessage) throws Exception {
//        LogUtil.debug("接收訊息主題 : " + topic);
//        LogUtil.debug("接收訊息Qos : " + mqttMessage.getQos());
        LogUtil.debug("####gps : " + new String(mqttMessage.getPayload()));
        //接收到的訊息傳送到kafka
        KafkaServer.getIntance().sendMessage(KafkaConsts.KAFKA_TOPIC_DATA,new String(mqttMessage.getPayload()));
    }

    /**
     * 接收到已經發布的 QoS 1 或 QoS 2 訊息的傳遞令牌時呼叫。
     *
     * @param iMqttDeliveryToken
     */
    @Override
    public void deliveryComplete(IMqttDeliveryToken iMqttDeliveryToken) {
        LogUtil.info("mqtt deliveryComplete...");
    }
}