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...");
    }
}