diff --git a/im-server/src/main/java/com/bx/imserver/websocket/WebsocketChannelCtxHolder.java b/im-server/src/main/java/com/bx/imserver/websocket/WebsocketChannelCtxHolder.java index 2a87103..69f9a1d 100644 --- a/im-server/src/main/java/com/bx/imserver/websocket/WebsocketChannelCtxHolder.java +++ b/im-server/src/main/java/com/bx/imserver/websocket/WebsocketChannelCtxHolder.java @@ -18,16 +18,16 @@ public class WebsocketChannelCtxHolder { } public static void removeChannelCtx(Long userId){ - channelMap.remove(userId); + if(userId != null){ + channelMap.remove(userId); + } } public static ChannelHandlerContext getChannelCtx(Long userId){ + if(userId == null){ + return null; + } return channelMap.get(userId); } - public static Set getAllChannelIds(){ - return channelMap.keySet(); - } - - }