From b99e4e26ce2750eafdb8e10c7c9e243328fc12c2 Mon Sep 17 00:00:00 2001 From: xiaoyong931011 <15274802129@163.com> Date: Wed, 26 May 2021 19:19:08 +0800 Subject: [PATCH] 20210526 聊天 --- src/main/java/com/xcong/excoin/netty/common/ChannelManager.java | 15 +++++++++++++++ 1 files changed, 15 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/xcong/excoin/netty/common/ChannelManager.java b/src/main/java/com/xcong/excoin/netty/common/ChannelManager.java index 5d158f6..0167807 100644 --- a/src/main/java/com/xcong/excoin/netty/common/ChannelManager.java +++ b/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; } -- Gitblit v1.9.1