com.feingto.iot.server.handler.mqtt.PublishHandler Maven / Gradle / Ivy

Go to download
package com.feingto.iot.server.handler.mqtt;

import com.feingto.iot.common.cache.MessageCache;
import com.feingto.iot.common.cache.RetainedCache;
import com.feingto.iot.common.cache.SubscribeCache;
import com.feingto.iot.common.handler.BaseMessageHandler;
import com.feingto.iot.common.model.mqtt.SendMessage;
import com.feingto.iot.common.service.mqtt.MessageResponse;
import com.feingto.iot.server.service.PushService;
import io.netty.channel.Channel;
import io.netty.handler.codec.mqtt.*;
import lombok.extern.slf4j.Slf4j;

/**
 * 消息发布处理器
 *
 * @author longfei
 */
@Slf4j
public class PublishHandler extends BaseMessageHandler {
    private final PushService pushService;

    public PublishHandler(PushService pushService) {
        super(MqttMessageType.PUBLISH);
        this.pushService = pushService;
    }

    @Override
    public void handle(Channel channel, Object object) {
        MqttPublishMessage msg = (MqttPublishMessage) object;
        MqttFixedHeader fixedHeader = msg.fixedHeader();
        MqttPublishVariableHeader variableHeader = msg.variableHeader();
        SendMessage sendMessage = SendMessage.newInstance(msg);

        // 存储每个Topic的最后一条保留消息及其Qos
        if (sendMessage.retain()) {
            log.debug(">>> save retain message on {}", sendMessage.topic());
            RetainedCache.getInstance(igniteRetained).put(sendMessage.topic(), sendMessage);
        }

        sendMessage.retain(false);
        switch (fixedHeader.qosLevel()) {
            case AT_MOST_ONCE:
                // 发布消息到所有订阅者
                pushService.internalSend(sendMessage);
                break;
            case AT_LEAST_ONCE:
                // 持久化消息
                this.storeMessage(sendMessage, MqttMessageType.PUBLISH);

                // 发布消息到所有订阅者
                pushService.internalSend(sendMessage);

                // 返回puback消息
                MessageResponse.puback(channel, MqttMessageType.PUBACK, MqttQoS.AT_MOST_ONCE, variableHeader.packetId());
                break;
            case EXACTLY_ONCE:
                // 持久化消息
                this.storeMessage(sendMessage, MqttMessageType.PUBREC);

                // 返回pubrec消息
                MessageResponse.puback(channel, MqttMessageType.PUBREC, MqttQoS.AT_LEAST_ONCE, variableHeader.packetId());
                break;
            case FAILURE:
                break;
        }
    }

    /**
     * 持久化消息
     */
    private void storeMessage(SendMessage message, MqttMessageType type) {
        SubscribeCache.getInstance((igniteSubscribe)).get(message.topic())
                .forEach(subscribe -> MessageCache.getInstance(igniteMessage)
                        .put(subscribe.clientId(), message.type(type)));
    }
}