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