diff --git a/src/main/java/com/zhangshu/chat/demo/websocket/ChatRoom.java b/src/main/java/com/zhangshu/chat/demo/websocket/ChatRoom.java index 3285974..6937706 100644 --- a/src/main/java/com/zhangshu/chat/demo/websocket/ChatRoom.java +++ b/src/main/java/com/zhangshu/chat/demo/websocket/ChatRoom.java @@ -39,10 +39,10 @@ public class ChatRoom { */ @OnOpen public void onOpen(Session session, @PathParam("roomId") String roomId) { -// if (getRoomCache().createSuccess(roomId)) { -// this.sendMessage("无此房间,拒绝连接", session); -// return; -// } + if (getRoomCache().createSuccess(roomId)) { + this.sendMessage("无此房间,拒绝连接", session); + return; + } Principal userPrincipal = session.getUserPrincipal(); if (Objects.isNull(userPrincipal)) { this.sendMessage("用户验证失败,拒绝连接", session);