bug fix

src/main/java/com/huiju/estateagents/websocket/ChatServer.java

@@ -60,8 +60,7 @@ public class ChatServer {
      */
     @OnClose
     public void onClose() {
-        webSocketSet.remove(this);
-        subOnlineCount();
+        removeFromSets();
         log.info("有一连接关闭！当前在线人数为" + getOnlineCount());
     }
 
@@ -83,7 +82,7 @@ public class ChatServer {
 
         JSONObject jsnMsg = JSONObject.parseObject(message);
 
-        if ( null == ChatServer.taChatService) {
+        if (null == ChatServer.taChatService) {
             ChatServer.taChatService = applicationContext.getBean(ITaChatService.class);
         }
         System.out.println("收到的聊天消息的信息=========" + jsnMsg.toJSONString());
@@ -106,7 +105,7 @@ public class ChatServer {
     public void onError(Session session, Throwable error) {
         log.error("发生错误");
         error.printStackTrace();
-        webSocketSet.remove(this);
+        removeFromSets();
     }
 
     /**
@@ -116,6 +115,15 @@ public class ChatServer {
         this.session.getBasicRemote().sendText(message);
     }
 
+    private void removeFromSets() {
+        for(ChatServer c: webSocketSet) {
+            if (sid.equals(c.sid)) {
+                webSocketSet.remove(c);
+                subOnlineCount();
+            }
+        }
+    }
+
     /**
      * 发自定义消息
      * */