update 电单车协议

jsowell-common/src/main/java/com/jsowell/common/util/ChannelManagerUtil.java

@@ -0,0 +1,63 @@
+package com.jsowell.common.util;
+
+import io.netty.channel.ChannelHandlerContext;
+import lombok.extern.slf4j.Slf4j;
+
+import java.net.InetSocketAddress;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * 管理一个全局map，保存连接进服务端的通道数量
+ */
+@Slf4j
+public class ChannelManagerUtil {
+
+    // 使用 ConcurrentHashMap 来保证线程安全
+    private static final ConcurrentHashMap<String, ChannelHandlerContext> channelMap = new ConcurrentHashMap<>();
+
+    /**
+     * 添加通道到 map 中
+     *
+     * @param channelId 通道 ID
+     * @param ctx   连接的 Channel 对象
+     */
+    public static void addChannel(String channelId, ChannelHandlerContext ctx) {
+        if (channelMap.containsKey(channelId)) {
+            log.info("客户端【{}】是连接状态，连接通道数量: {}", channelId, channelMap.size());
+        } else {
+            channelMap.put(channelId, ctx);
+            InetSocketAddress socket = (InetSocketAddress) ctx.channel().remoteAddress();
+            String clientIp = socket.getAddress().getHostAddress();
+            int clientPort = socket.getPort();
+            log.info("客户端【{}】, 连接netty服务器【IP:{}, PORT:{}】, 连接通道数量: {}", channelId, clientIp, clientPort, channelMap.size());
+        }
+    }
+
+    /**
+     * 移除指定的通道
+     *
+     * @param channelId 通道 ID
+     */
+    public static void removeChannel(String channelId) {
+        channelMap.remove(channelId);
+    }
+
+    /**
+     * 获取当前连接的通道数量
+     *
+     * @return 通道数量
+     */
+    public static int getChannelCount() {
+        return channelMap.size();
+    }
+
+    /**
+     * 根据通道 ID 获取 Channel 对象
+     *
+     * @param channelId 通道 ID
+     * @return Channel 对象或 null 如果不存在
+     */
+    public static ChannelHandlerContext getChannel(String channelId) {
+        return channelMap.get(channelId);
+    }
+}