From 0005c320c6ba8cb3bafd02b35fe388f9c7f178dc Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 26 May 2021 17:14:07 +0800
Subject: [PATCH] netty
---
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