com.iteaj.iot.client.mqtt.impl.DefaultMqttComponent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of iot-mqtt Show documentation
基于iot框架实现的mqtt协议客户端
There is a newer version: 3.1.1
package com.iteaj.iot.client.mqtt.impl;

import com.iteaj.iot.AbstractProtocol;
import com.iteaj.iot.SocketMessage;
import com.iteaj.iot.client.ClientConnectProperties;
import com.iteaj.iot.client.IotClientBootstrap;
import com.iteaj.iot.client.MultiClientManager;
import com.iteaj.iot.client.mqtt.MessagePublishListener;
import com.iteaj.iot.client.mqtt.MqttClientComponent;
import com.iteaj.iot.client.mqtt.MqttClientException;
import com.iteaj.iot.client.mqtt.MqttConnectProperties;
import io.netty.handler.codec.mqtt.MqttPublishMessage;
import io.netty.handler.codec.mqtt.MqttTopicSubscription;

import java.util.ArrayList;
import java.util.List;

/**
 * Mqtt客户端默认实现
 * @see MqttSubscribeListener spring容器存在此对象将启用此组件
 */
public class DefaultMqttComponent extends MqttClientComponent {

    private static final String NAME = "MQTT<默认>";
    private static final String DESC = "MQTT默认客户端实现";

    public DefaultMqttComponent() { }

    public DefaultMqttComponent(MqttConnectProperties config) {
        super(config);
    }

    public DefaultMqttComponent(MqttConnectProperties config, MultiClientManager clientManager) {
        super(config, clientManager);
    }

    public DefaultMqttComponent(ClientConnectProperties config, MultiClientManager clientManager, MessagePublishListener publishListener) {
        super(config, clientManager, publishListener);
    }

    @Override
    public String getName() {
        return NAME;
    }

    @Override
    public String getDesc() {
        return DESC;
    }

    @Override
    public AbstractProtocol getProtocol(DefaultMqttMessage message) {
        // 使用监听的方式处理
        return new DefaultMqttSubscribeProtocol(message);
    }

    @Override
    protected List doSubscribe(MqttConnectProperties client) {
        List subscriptions = new ArrayList<>();
        IotClientBootstrap.applicationContext.getAutowireCapableBeanFactory()
                .getBeanProvider(MqttSubscribeListenerManager.class).ifAvailable((manager) -> {
            manager.getListeners().forEach(item -> {
                MqttTopicSubscription topic = item.topic();
                if(topic == null) {
                    throw new MqttClientException("mqtt监听器["+item.getClass().getSimpleName()+"]没有返回topic[null]");
                }

                subscriptions.add(topic);
            });
        });

        return subscriptions;
    }

    @Override
    public Class getMessageClass() {
        return DefaultMqttMessage.class;
    }

    @Override
    public DefaultMqttMessage createMessage(byte[] message) {
        return new DefaultMqttMessage(message);
    }
}