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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of iot-hub-server Show documentation

There is a newer version: 1.2.5.RELEASE

package com.feingto.iot.server.handler.mqtt;

import com.feingto.iot.common.Constants;
import com.feingto.iot.common.model.mqtt.SubscribeMessage;
import com.feingto.iot.common.service.mqtt.MessageResponse;
import com.feingto.iot.server.cache.RetainedCache;
import com.feingto.iot.server.cache.SubscribeCache;
import com.feingto.iot.server.handler.BaseMessageHandler;
import com.feingto.iot.server.handler.MqttHandlerChain;
import io.netty.channel.Channel;
import io.netty.handler.codec.mqtt.MqttMessageType;
import io.netty.handler.codec.mqtt.MqttSubscribeMessage;
import lombok.extern.slf4j.Slf4j;

import java.util.Optional;
import java.util.stream.Collectors;

/**
 * Mqtt 消息订阅处理器
 *
 * @author longfei
 */
@Slf4j
public class SubscribeHandler extends BaseMessageHandler {
    public SubscribeHandler() {
        super(MqttMessageType.SUBSCRIBE);
    }

    @Override
    public void handle(Channel channel, Object object) {
        String clientId = channel.attr(Constants.KEY_CLIENT_ID).get();
        MqttSubscribeMessage msg = (MqttSubscribeMessage) object;
        msg.payload().topicSubscriptions().forEach(subscription ->
                SubscribeCache.getInstance(igniteSubscribe)
                        .put(subscription.topicName(), new SubscribeMessage()
                                .clientId(clientId)
                                .topicName(subscription.topicName())
                                .mqttQoS(subscription.qualityOfService().value())));

        // 返回suback消息
        MessageResponse.suback(channel, msg.payload().topicSubscriptions().stream()
                .map(subscription -> subscription.qualityOfService().value())
                .collect(Collectors.toList()), msg.variableHeader().messageId());

        // 返回保留消息
        SubscribeCache.getInstance(igniteSubscribe).findByClientId(clientId).forEach(subscribe ->
                Optional.ofNullable(RetainedCache.getInstance(igniteRetained).get(subscribe.topicName()))
                        .ifPresent(sendMessage -> {
                            log.debug(">>> publish retained message to {}", clientId);
                            MqttHandlerChain.getInstance().pushService.internalSend(sendMessage);
                        }));
    }
}