From 8c5435101924d68588448a4c981417df4b4dab1d Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Fri, 21 Feb 2025 11:08:20 +0800
Subject: [PATCH] 整合websocket 统一使用按类型区分

---
 ard-work/src/main/java/com/ard/tianyu/websocket/utils/WebSocketUtil.java |   11 +++++++----
 1 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/tianyu/websocket/utils/WebSocketUtil.java b/ard-work/src/main/java/com/ard/tianyu/websocket/utils/WebSocketUtil.java
index 7090119..27c8e65 100644
--- a/ard-work/src/main/java/com/ard/tianyu/websocket/utils/WebSocketUtil.java
+++ b/ard-work/src/main/java/com/ard/tianyu/websocket/utils/WebSocketUtil.java
@@ -27,9 +27,6 @@
 
     // 存储 websocket session
     public static final ConcurrentMap<String, WebSocketSession> ONLINE_USER_SESSIONS = new ConcurrentHashMap<>();
-    // 存储 拉流 websocket session
-    public static final ConcurrentMap<String, WebSocketSession> STREAM_ONLINE_USER_SESSIONS = new ConcurrentHashMap<>();
-
     /**
      * @param session 用户 session
      * @param message 发送内容
@@ -51,10 +48,12 @@
             }
         }
     }
+
     public static void sendMessage(String userId, String message) {
         WebSocketSession session = ONLINE_USER_SESSIONS.get(userId);
         sendMessage(session, message);
     }
+
     /**
      * @param session 用户 session
      * @param message 发送内容
@@ -63,7 +62,7 @@
         if (session == null) {
             return;
         }
-       final InetSocketAddress remoteAddress = session.getRemoteAddress();
+        final InetSocketAddress remoteAddress = session.getRemoteAddress();
         if (remoteAddress == null) {
             return;
         }
@@ -101,6 +100,7 @@
     public static void sendMessageAll(String message) {
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, message));
     }
+
     /**
      * 推送消息到其他客户端
      *
@@ -110,6 +110,7 @@
         JSONObject jsonObject = new JSONObject(message);
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, jsonObject.toString()));
     }
+
     /**
      * 推送消息到其他客户端
      *
@@ -119,6 +120,7 @@
         String jsonMessage = JSONObject.toJSONString(message);  // 确保对象转换为 JSON 字符串
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, jsonMessage));
     }
+
     /**
      * 推送消息到其他客户端
      *
@@ -127,6 +129,7 @@
     public static void sendMessageAll(List message) {
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, JSON.toJSONString(message)));
     }
+
     /**
      * 发送消息给userId_前缀的人
      */

--
Gitblit v1.9.3