diff --git a/src/org/redkale/net/http/WebSocket.java b/src/org/redkale/net/http/WebSocket.java index d4592f61d..78ece1cf1 100644 --- a/src/org/redkale/net/http/WebSocket.java +++ b/src/org/redkale/net/http/WebSocket.java @@ -596,6 +596,14 @@ public abstract class WebSocket { this.getLogger().log(Level.SEVERE, "WebSocket receive or send Message error", t); } + /** + * 当Single模式下用户重复登陆时回调函数, 默认处理逻辑:关闭之前的WebSocket连接 + * + */ + public void onSingleRepeatConnect() { + this._engine.node.forceCloseWebSocket(getUserid()); + } + /** * 获取Logger * diff --git a/src/org/redkale/net/http/WebSocketServlet.java b/src/org/redkale/net/http/WebSocketServlet.java index 20cfeb767..91475eaca 100644 --- a/src/org/redkale/net/http/WebSocketServlet.java +++ b/src/org/redkale/net/http/WebSocketServlet.java @@ -186,11 +186,22 @@ public abstract class WebSocketServlet extends HttpServlet implements Resourcabl return; } webSocket._userid = userid; - WebSocketServlet.this.node.localEngine.add(webSocket); - WebSocketRunner runner = new WebSocketRunner(context, webSocket, restMessageConsumer, response.removeChannel()); - webSocket._runner = runner; - context.runAsync(runner); - response.finish(true); + if (single) { + WebSocketServlet.this.node.existsWebSocket(userid).whenComplete((rs, ex) -> { + if (rs) webSocket.onSingleRepeatConnect(); + WebSocketServlet.this.node.localEngine.add(webSocket); + WebSocketRunner runner = new WebSocketRunner(context, webSocket, restMessageConsumer, response.removeChannel()); + webSocket._runner = runner; + context.runAsync(runner); + response.finish(true); + }); + } else { + WebSocketServlet.this.node.localEngine.add(webSocket); + WebSocketRunner runner = new WebSocketRunner(context, webSocket, restMessageConsumer, response.removeChannel()); + webSocket._runner = runner; + context.runAsync(runner); + response.finish(true); + } }); }