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 {