cn.featherfly.common.mqtt.EasyMqttClientImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-mqtt Show documentation
featherfly mqtt
There is a newer version: 0.2.2

package cn.featherfly.common.mqtt;

import java.nio.charset.Charset;
import java.util.function.BiConsumer;
import java.util.function.Consumer;

import org.eclipse.paho.client.mqttv3.IMqttDeliveryToken;
import org.eclipse.paho.client.mqttv3.MqttException;
import org.eclipse.paho.client.mqttv3.MqttMessage;
import org.eclipse.paho.client.mqttv3.MqttPersistenceException;
import org.eclipse.paho.client.mqttv3.MqttTopic;

/**
 * EasyClient.
 *
 * @author zhongj
 */
public class EasyMqttClientImpl extends ReconnectableClient {

    /**
     * Instantiates a new Client.
     */
    EasyMqttClientImpl() {
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl connect() throws MqttException {
        return connect(new AutoDetectionMqttCallBack(this));
    }

    //    /**
    //     * Connect client.
    //     *
    //     * @return the client
    //     * @throws MqttException the mqtt exception
    //     */
    //    private EasyClient _connect(AutoDetectionMqttCallBack callback) throws MqttException {
    //        // host为主机名，clientid即连接MQTT的客户端ID，一般以唯一标识符表示，MemoryPersistence设置clientid的保存形式，默认为以内存保存
    //        if (!connected) {
    //            if (address == null) {
    //                address = protocol + "://" + host + ":" + port;
    //            }
    //            client = new MqttClient(address, clientId, persistence);
    //            this.callback = callback;
    //            client.setCallback(callback);
    //            try {
    //                client.connect(options);
    //                connected = true;
    //            } catch (Exception e) {
    //                // 开启线程自动重新连接
    //                connected = false;
    //                new Thread(() -> {
    //                    autoReconnect(1000);
    //                }).run();
    //            }
    //        }
    //        return this;
    //    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl subscribe(String topicFilter, Qos qos, Consumer consumer)
            throws MqttException {
        client.subscribe(topicFilter, qos.ordinal(), (topic, message) -> {
            if (logger.isDebugEnabled()) {
                logger.debug("receive topic -> {0}, msgId -> {1}, qos -> {2}, payload -> {3}", topic, message.getId(),
                        message.getQos(), new String(message.getPayload(), charset));
            }
            consumer.accept(message);
        });
        return this;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl subscribe(String topicFilter, Qos qos, BiConsumer consumer)
            throws MqttException {
        client.subscribe(topicFilter, qos.ordinal(), (topic, message) -> {
            if (logger.isDebugEnabled()) {
                logger.debug("receive topic -> {0}, msgId -> {1}, qos -> {2}, payload -> {3}", topic, message.getId(),
                        message.getQos(), new String(message.getPayload(), charset));
            }
            consumer.accept(topic, message);
        });
        return this;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl publish(String topicFilter, String msg) throws MqttPersistenceException, MqttException {
        return publish(topicFilter, msg, Qos.ONLY_ONCE);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl publish(String topic, String msg, Consumer consumer)
            throws MqttPersistenceException, MqttException {
        return publish(topic, msg, Qos.ONLY_ONCE, consumer);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl publish(String topic, String msg, Qos qos) throws MqttException {
        return publish(topic, msg, qos, null);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl publish(String topic, String msg, Qos qos, Consumer consumer)
            throws MqttPersistenceException, MqttException {
        return publish(topic, msg, qos, charset, consumer);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public EasyMqttClientImpl publish(String topic, String msg, Qos qos, Charset charset,
            Consumer consumer) throws MqttPersistenceException, MqttException {
        if (consumer != null) {
            ((AutoDetectionMqttCallBack) callback).publish(topic, consumer);
        }
        logger.debug("publish topic -> {0}, qos -> {1}, msg -> {2}", topic, qos.ordinal(), msg);
        MqttMessage message = new MqttMessage();
        message.setQos(qos.ordinal());
        message.setRetained(true);
        message.setPayload(msg.getBytes(charset));
        MqttTopic mqttTopic = client.getTopic(topic);
        publish(mqttTopic, message);
        return this;
    }
}