com.github.cm.heclouds.adapter.protocolhub.tcp.session.TcpDeviceSessionManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of protocol-adapter-protocolhub-tcp Show documentation
OneNET Studio泛协议接入SDK协议站-TCP
There is a newer version: 1.0.4
Show newest version
package com.github.cm.heclouds.adapter.protocolhub.tcp.session;

import com.github.cm.heclouds.adapter.core.entity.Device;
import com.github.cm.heclouds.adapter.core.logging.ILogger;
import com.github.cm.heclouds.adapter.core.utils.DeviceUtils;
import com.github.cm.heclouds.adapter.protocolhub.tcp.config.TcpProtocolHubConfigUtils;
import io.netty.channel.Channel;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import static com.github.cm.heclouds.adapter.core.logging.LoggerFormat.Action.DISCONNECT;
import static com.github.cm.heclouds.adapter.core.logging.LoggerFormat.Action.RUNTIME;

/**
 * 设备Session管理
 */
public final class TcpDeviceSessionManager {

    private static final ILogger LOGGER = TcpProtocolHubConfigUtils.getLogger();

    /**
     * 设备连接Session池
     * 设备名称与设备连接Session的映射
     */
    private static final ConcurrentMap DEVICE_SESSION_POOL = new ConcurrentHashMap<>();

    private TcpDeviceSessionManager() {
    }

    /**
     * 创建设备连接Session
     *
     * @param productId  产品ID
     * @param deviceName 设备名称
     * @param channel    连接channel
     * @return 设备连接Session
     */
    public static TcpDeviceSession createDeviceSession(String productId, String deviceName, Channel channel) {
        return TcpDeviceSession.newBuilder()
                .productId(productId)
                .deviceName(deviceName)
                .channel(channel)
                .build();
    }

    /**
     * 放入设备连接Session
     *
     * @param deviceSession 设备连接Session
     */
    public static void putDeviceSession(TcpDeviceSession deviceSession) {
        DEVICE_SESSION_POOL.put(genKey(deviceSession), deviceSession);
    }

    /**
     * 获取设备连接Session
     *
     * @param productId  产品ID
     * @param deviceName 设备名称
     * @return 设备连接Session
     */
    public static TcpDeviceSession getDeviceSession(String productId, String deviceName) {
        return DEVICE_SESSION_POOL.get(productId + "-" + deviceName);
    }

    /**
     * 连接断开情况处理
     */
    public static void handleConnectionLost(TcpDeviceSession deviceSession) {
        if (deviceSession != null) {
            Device device = Device.newBuilder().productId(deviceSession.getProductId()).deviceName(deviceSession.getDeviceName()).build();
            String reason = DeviceUtils.getDeviceCloseReason(device).getValue();
            DeviceUtils.removeDeviceCloseReason(device);
            // 移除DevSessionManager中的DeviceSession
            TcpDeviceSessionManager.removeDeviceSession(deviceSession);
            LOGGER.logDevInfo(TcpProtocolHubConfigUtils.getName(), DISCONNECT, deviceSession.getProductId(), deviceSession.getDeviceName(), reason);
        } else {
            LOGGER.logInnerWarn(TcpProtocolHubConfigUtils.getName(), RUNTIME, "device connection lost without logging in");
        }
    }

    /**
     * 移除设备连接Session
     *
     * @param deviceSession 连接Session
     */
    private static void removeDeviceSession(TcpDeviceSession deviceSession) {
        DEVICE_SESSION_POOL.remove(genKey(deviceSession));
    }

    /**
     * 生成DevSession池的key，规则为产品id+设备名称
     *
     * @param deviceSession 设备Session
     * @return key
     */
    private static String genKey(TcpDeviceSession deviceSession) {
        return deviceSession.getProductId() + "-" + deviceSession.getDeviceName();
    }

}