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

优化WebSocket

Browse Source
This commit is contained in:
redkale
2023-04-05 18:43:38 +08:00
parent 68b9d955f6
commit 4bdbf0f493
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/org/redkale/net/http/HttpContext.java
View File

@@ -95,9 +95,11 @@ public class HttpContext extends Context {
} }
protected void updateWebSocketWriteIOThread(WebSocket webSocket) { protected void updateWebSocketWriteIOThread(WebSocket webSocket) {
WebSocketWriteIOThread writeIOThread = webSocketWriterIOThreadFunc.apply(webSocket); if (webSocketWriterIOThreadFunc != null) {
updateWriteIOThread(webSocket._channel, writeIOThread); WebSocketWriteIOThread writeIOThread = webSocketWriterIOThreadFunc.apply(webSocket);
webSocket._writeIOThread = writeIOThread; updateWriteIOThread(webSocket._channel, writeIOThread);
webSocket._writeIOThread = writeIOThread;
}
} }
protected String createSessionid() { protected String createSessionid() {

2
src/main/java/org/redkale/net/http/WebSocketServlet.java
View File

@@ -286,7 +286,7 @@ public abstract class WebSocketServlet extends HttpServlet implements Resourcabl
public void completed(Integer result, Void attachment) { public void completed(Integer result, Void attachment) {
webSocket._readHandler = new WebSocketReadHandler(response.getContext(), webSocket, restMessageConsumer); webSocket._readHandler = new WebSocketReadHandler(response.getContext(), webSocket, restMessageConsumer);
webSocket._writeHandler = new WebSocketWriteHandler(response.getContext(), webSocket); webSocket._writeHandler = new WebSocketWriteHandler(response.getContext(), webSocket);
response.getContext().updateWebSocketWriteIOThread(webSocket); //response.getContext().updateWebSocketWriteIOThread(webSocket);
Runnable createUseridHandler = () -> { Runnable createUseridHandler = () -> {
CompletableFuture<Serializable> userFuture = webSocket.createUserid(); CompletableFuture<Serializable> userFuture = webSocket.createUserid();