xiaoyong931011
2021-05-26 a0da4723b75c2bb13d2b4bef8fed196c678ddcde
src/main/java/com/xcong/excoin/netty/common/ChannelManager.java
@@ -28,16 +28,31 @@
        CHANNEL_MAP.put(channel.id().asShortText(), channel.id());
    }
    public static void addWsChannel(Channel channel, Long memberId) {
        WEBSOCKET_GROUP.add(channel);
        CHANNEL_MAP.put(memberId.toString(), channel.id());
    }
    public static void removeWebSocketChannel(Channel channel) {
        WEBSOCKET_GROUP.remove(channel);
        CHANNEL_MAP.remove(channel.id().asShortText());
    }
    public static void removeWsChannel(Channel channel, Long memberId) {
        WEBSOCKET_GROUP.remove(channel);
        CHANNEL_MAP.remove(memberId.toString());
    }
    public static Channel findWebSocketChannel(String id){
        ChannelId channelId = CHANNEL_MAP.get(id);
        return WEBSOCKET_GROUP.find(channelId);
    }
    public static Channel findWsChannel(Long id){
        ChannelId channelId = CHANNEL_MAP.get(id.toString());
        return WEBSOCKET_GROUP.find(channelId);
    }
    public static ChannelGroup getWebSocketGroup() {
        return WEBSOCKET_GROUP;
    }