From c08308a7d4209bf9da1a19d4b78170be46dc6058 Mon Sep 17 00:00:00 2001
From: liuhanqing <447067298@qq.com>
Date: Mon, 1 Feb 2021 22:33:56 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=A4=E6=96=ADwsurl=20=E4=B8=8D=E4=B8=BA?=
 =?UTF-8?q?=E7=A9=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../websocket/manager/WebSocketManager.java       | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/kernel-s-message/message-sdk-websocket/src/main/java/cn/stylefeng/roses/kernel/message/websocket/manager/WebSocketManager.java b/kernel-s-message/message-sdk-websocket/src/main/java/cn/stylefeng/roses/kernel/message/websocket/manager/WebSocketManager.java
index 2d7d93dc8..a5972e51b 100644
--- a/kernel-s-message/message-sdk-websocket/src/main/java/cn/stylefeng/roses/kernel/message/websocket/manager/WebSocketManager.java
+++ b/kernel-s-message/message-sdk-websocket/src/main/java/cn/stylefeng/roses/kernel/message/websocket/manager/WebSocketManager.java
@@ -41,7 +41,7 @@ public class WebSocketManager {
     /**
      * 移除失效的Session
      *
-     * @param userId   用户id
+     * @param userId  用户id
      * @param session 用户websocketSession
      * @author liuhanqing
      * @date 2021/1/24 22:11
@@ -75,11 +75,16 @@ public class WebSocketManager {
      * @author liuhanqing
      * @date 2021/1/24 22:11
      */
-    public static void sendMessage(Long userId, String message){
-        for(Session userSession: getSessionByUserId(userId)){
-            userSession.getAsyncRemote().sendText(message);
+    public static void sendMessage(Long userId, String message) {
+        List<Session> sessionList = getSessionByUserId(userId);
+        // 增加判断不为空
+        if (!CollectionUtils.isEmpty(sessionList)) {
+            for (Session userSession : sessionList) {
+                userSession.getAsyncRemote().sendText(message);
+            }
         }
     }
+
     /**
      * 发送消息
      *
@@ -87,7 +92,7 @@ public class WebSocketManager {
      * @author liuhanqing
      * @date 2021/1/24 22:11
      */
-    public static void sendMessageToAll(String message){
+    public static void sendMessageToAll(String message) {
         for (Long userId : WebSocketManager.getUserList()) {
             sendMessage(userId, message);
         }