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 |   17 ++++++++++-------
 1 files changed, 10 insertions(+), 7 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 dc78204..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,7 +27,6 @@
 
     // 存储 websocket session
     public static final ConcurrentMap<String, WebSocketSession> ONLINE_USER_SESSIONS = new ConcurrentHashMap<>();
-
     /**
      * @param session 用户 session
      * @param message 发送内容
@@ -50,6 +49,11 @@
         }
     }
 
+    public static void sendMessage(String userId, String message) {
+        WebSocketSession session = ONLINE_USER_SESSIONS.get(userId);
+        sendMessage(session, message);
+    }
+
     /**
      * @param session 用户 session
      * @param message 发送内容
@@ -58,7 +62,7 @@
         if (session == null) {
             return;
         }
-       final InetSocketAddress remoteAddress = session.getRemoteAddress();
+        final InetSocketAddress remoteAddress = session.getRemoteAddress();
         if (remoteAddress == null) {
             return;
         }
@@ -106,6 +110,7 @@
         JSONObject jsonObject = new JSONObject(message);
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, jsonObject.toString()));
     }
+
     /**
      * 推送消息到其他客户端
      *
@@ -115,6 +120,7 @@
         String jsonMessage = JSONObject.toJSONString(message);  // 确保对象转换为 JSON 字符串
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, jsonMessage));
     }
+
     /**
      * 推送消息到其他客户端
      *
@@ -123,12 +129,9 @@
     public static void sendMessageAll(List message) {
         ONLINE_USER_SESSIONS.forEach((sessionId, session) -> sendMessage(session, JSON.toJSONString(message)));
     }
+
     /**
-     * @Author 刘苏义
-     * @Description  发送消息给userId_前缀的人
-     * @Date   2024/7/16 10:24
-     * @Param
-     * @return
+     * 发送消息给userId_前缀的人
      */
     public static void sendMessagePrefix(String targetId, String message) {
         String regex = "^" + Pattern.quote(targetId) + "_\\d+$";

--
Gitblit v1.9.3