Joburgess 4 anni fa
parent
commit
a91cf240c6

+ 5 - 1
mec-teacher/src/main/java/com/ym/mec/teacher/handler/WebSocketHandler.java

@@ -59,12 +59,16 @@ public class WebSocketHandler extends AbstractWebSocketHandler {
         exception.printStackTrace();
         LOGGER.info("发生了错误,移除客户端: {}", session.getPrincipal().getName().split(":")[1]);
         session.close();
+        String phone = session.getPrincipal().getName().split(":")[1];
+        WS_CLIENTS.remove(phone);
     }
 
     @Override
     public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
         super.afterConnectionClosed(session, status);
-        LOGGER.info("{}离线", session.getPrincipal().getName().split(":")[1]);
+        String phone = session.getPrincipal().getName().split(":")[1];
+        LOGGER.info("{}离线", phone);
+        WS_CLIENTS.remove(phone);
     }
 
     @Override