lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Browse Source
This commit is contained in:
Redkale
2018-05-28 09:59:39 +08:00
parent 4a2ba4e914
commit d9ddd6d772
2 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/org/redkale/net/http/WebSocket.java
View File

@@ -754,11 +754,10 @@ public abstract class WebSocket<G extends Serializable, T> {
} }
/** /**
* 当Single模式下用户重复登陆时回调函数， 默认处理逻辑：关闭之前的WebSocket连接 * 当Single模式下用户重复登陆时回调函数，调用完之后关闭之前的WebSocket连接
* *
*/ */
public void onSingleRepeatConnect() { public void onSingleRepeatConnect() {
this._engine.node.forceCloseWebSocket(getUserid());
} }
/** /**

22
src/org/redkale/net/http/WebSocketServlet.java
View File

@@ -241,12 +241,22 @@ public abstract class WebSocketServlet extends HttpServlet implements Resourcabl
webSocket._userid = userid; webSocket._userid = userid;
if (single && !anyuser) { if (single && !anyuser) {
WebSocketServlet.this.node.existsWebSocket(userid).whenComplete((rs, nex) -> { WebSocketServlet.this.node.existsWebSocket(userid).whenComplete((rs, nex) -> {
if (rs) webSocket.onSingleRepeatConnect(); if (rs) {
WebSocketServlet.this.node.localEngine.add(webSocket); webSocket.onSingleRepeatConnect();
WebSocketRunner runner = new WebSocketRunner(context, webSocket, restMessageConsumer, response.removeChannel()); webSocket.forceCloseWebSocket(userid).whenComplete((fr, fex) -> {
webSocket._runner = runner; WebSocketServlet.this.node.localEngine.add(webSocket);
context.runAsync(runner); WebSocketRunner runner = new WebSocketRunner(context, webSocket, restMessageConsumer, response.removeChannel());
response.finish(true); 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);
}
}); });
} else { } else {
WebSocketServlet.this.node.localEngine.add(webSocket); WebSocketServlet.this.node.localEngine.add(webSocket);