⬆️ WebSocket 出错回调处理

zatol · Nov 8, 2019 · 0112789 · 0112789
1 parent e1568d6
commit 0112789
Show file tree

Hide file tree

Showing 5 changed files with 50 additions and 0 deletions.
diff --git a/src/main/java/org/b3log/symphony/processor/channel/ArticleChannel.java b/src/main/java/org/b3log/symphony/processor/channel/ArticleChannel.java
@@ -248,6 +248,16 @@ public void onClose(final WebSocketSession session) {
     public void onMessage(final Message message) {
     }
 
+    /**
+     * Called when a error received.
+     *
+     * @param error error
+     */
+    @Override
+    public void onError(final Error error) {
+        removeSession(error.session);
+    }
+
     /**
      * Removes the specified session.
      *

diff --git a/src/main/java/org/b3log/symphony/processor/channel/ArticleListChannel.java b/src/main/java/org/b3log/symphony/processor/channel/ArticleListChannel.java
@@ -105,4 +105,14 @@ public void onClose(final WebSocketSession session) {
     @Override
     public void onMessage(final Message message) {
     }
+
+    /**
+     * Called when a error received.
+     *
+     * @param error error
+     */
+    @Override
+    public void onError(final Error error) {
+        SESSIONS.remove(error.session);
+    }
 }
diff --git a/src/main/java/org/b3log/symphony/processor/channel/ChatroomChannel.java b/src/main/java/org/b3log/symphony/processor/channel/ChatroomChannel.java
@@ -87,6 +87,16 @@ public void onClose(final WebSocketSession session) {
     public void onMessage(final Message message) {
     }
 
+    /**
+     * Called when a error received.
+     *
+     * @param error error
+     */
+    @Override
+    public void onError(final Error error) {
+        removeSession(error.session);
+    }
+
     /**
      * Notifies the specified chat message to browsers.
      *

diff --git a/src/main/java/org/b3log/symphony/processor/channel/GobangChannel.java b/src/main/java/org/b3log/symphony/processor/channel/GobangChannel.java
@@ -294,6 +294,16 @@ public void onMessage(final Message message) throws JSONException {
         }
     }
 
+    /**
+     * Called when a error received.
+     *
+     * @param error error
+     */
+    @Override
+    public void onError(final Error error) {
+        removeSession(error.session);
+    }
+
     /**
      * Removes the specified session.
      *

diff --git a/src/main/java/org/b3log/symphony/processor/channel/UserChannel.java b/src/main/java/org/b3log/symphony/processor/channel/UserChannel.java
@@ -113,6 +113,16 @@ public void onMessage(final Message message) {
         userMgmtService.updateOnlineStatus(userId, ip, true, true);
     }
 
+    /**
+     * Called when a error received.
+     *
+     * @param error error
+     */
+    @Override
+    public void onError(final Error error) {
+        removeSession(error.session);
+    }
+
     /**
      * Sends command to browsers.
      *