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(); }