From 479f46b772ec17c4e60d948f0ee2de6c2cb8aedf Mon Sep 17 00:00:00 2001
From: liyawei <llyywwaoao@163.com>
Date: Thu, 21 Apr 2022 18:22:32 +0800
Subject: [PATCH] =?UTF-8?q?websocket=E5=BC=82=E5=B8=B8[TEXT=5FFULL=5FWRITI?=
 =?UTF-8?q?NG]=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../modules/message/websocket/WebSocket.java  | 33 +++++++++++--------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/message/websocket/WebSocket.java b/jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/message/websocket/WebSocket.java
index 910cc459b..3edaa541d 100644
--- a/jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/message/websocket/WebSocket.java
+++ b/jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/message/websocket/WebSocket.java
@@ -1,8 +1,11 @@
 package com.jero.modules.message.websocket;
 
-import java.util.HashMap;
-import java.util.Map;
-import java.util.concurrent.CopyOnWriteArraySet;
+import com.alibaba.fastjson.JSONObject;
+import com.jero.boot.starter.redis.client.JeroRedisClient;
+import com.jero.common.base.BaseMap;
+import com.jero.common.constant.WebsocketConst;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
 
 import javax.annotation.Resource;
 import javax.websocket.OnClose;
@@ -11,15 +14,10 @@ import javax.websocket.OnOpen;
 import javax.websocket.Session;
 import javax.websocket.server.PathParam;
 import javax.websocket.server.ServerEndpoint;
-
-import com.jero.boot.starter.redis.client.JeroRedisClient;
-import com.jero.common.base.BaseMap;
-import com.jero.common.constant.WebsocketConst;
-import org.springframework.stereotype.Component;
-
-import com.alibaba.fastjson.JSONObject;
-
-import lombok.extern.slf4j.Slf4j;
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.CopyOnWriteArraySet;
 
 /**
  * @Author scott
@@ -78,7 +76,7 @@ public class WebSocket {
         if (session != null && session.isOpen()) {
             try {
                 log.info("【websocket消息】 单点消息:" + message);
-                session.getAsyncRemote().sendText(message);
+                session.getBasicRemote().sendText(message);
             } catch (Exception e) {
                 e.printStackTrace();
             }
@@ -90,7 +88,14 @@ public class WebSocket {
      */
     public void pushMessage(String message) {
         try {
-            webSockets.forEach(ws -> ws.session.getAsyncRemote().sendText(message));
+            for (WebSocket ws : webSockets) {
+                try {
+                    ws.session.getBasicRemote().sendText(message);
+                } catch (IOException e) {
+                    e.printStackTrace();
+                    continue;
+                }
+            }
         } catch (Exception e) {
             e.printStackTrace();
         }