From 5fe605c130a95c0927d8b5782a12fa174ec0ec97 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Mon, 11 Nov 2024 10:45:57 +0800
Subject: [PATCH] =?UTF-8?q?=E9=95=BF=E8=BF=9E=E6=8E=A5=20=E5=85=B3?=
 =?UTF-8?q?=E9=97=AD=E6=97=A5=E5=BF=97=E6=89=93=E5=8D=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/websocket/NotificationConnection.java  | 15 +++++++++++++--
 .../common/websocket/config/WebSocketConfig.java  |  2 --
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/ai/da/common/websocket/NotificationConnection.java b/src/main/java/com/ai/da/common/websocket/NotificationConnection.java
index 7dc7c7da..55e99ef3 100644
--- a/src/main/java/com/ai/da/common/websocket/NotificationConnection.java
+++ b/src/main/java/com/ai/da/common/websocket/NotificationConnection.java
@@ -38,12 +38,23 @@ public class NotificationConnection {
     }
 
     @OnClose
-    public void onClose() {
+    public void onClose(CloseReason reason) {
         websockets.remove(this); // 将当前的对象从集合中删除
-        log.info("【连接断开】 用户为：{}", this.session);
+        log.info("【连接断开】 用户为：{}, 原因为{}", this.session, reason);
 //        log.info("【连接断开】 总数为：{}", websockets.size());
     }
 
+    /**
+     * 错误时调用
+     * @param session 连接
+     * @param throwable 异常
+     */
+    @OnError
+    public void onError(Session session, Throwable throwable) {
+        log.info("【连接异常】[session({}) 发生异常]", session, throwable);
+        websockets.remove(this); // 将当前的对象从集合中删除
+    }
+
     //收到了客户端消息执行的操作
     @OnMessage
     public void onMessage(String text){
diff --git a/src/main/java/com/ai/da/common/websocket/config/WebSocketConfig.java b/src/main/java/com/ai/da/common/websocket/config/WebSocketConfig.java
index df5b252b..9feb56a1 100644
--- a/src/main/java/com/ai/da/common/websocket/config/WebSocketConfig.java
+++ b/src/main/java/com/ai/da/common/websocket/config/WebSocketConfig.java
@@ -6,8 +6,6 @@ import org.springframework.web.socket.server.standard.ServerEndpointExporter;
 
 /**
  * Configuration of WebSocket
- *
- * @author db1995
  */
 @Configuration
 public class WebSocketConfig {