diff --git a/src/main/java/org/redkale/net/http/HttpContext.java b/src/main/java/org/redkale/net/http/HttpContext.java index 6d38bbcf1..3bc11f16f 100644 --- a/src/main/java/org/redkale/net/http/HttpContext.java +++ b/src/main/java/org/redkale/net/http/HttpContext.java @@ -95,9 +95,11 @@ public class HttpContext extends Context { } protected void updateWebSocketWriteIOThread(WebSocket webSocket) { - WebSocketWriteIOThread writeIOThread = webSocketWriterIOThreadFunc.apply(webSocket); - updateWriteIOThread(webSocket._channel, writeIOThread); - webSocket._writeIOThread = writeIOThread; + if (webSocketWriterIOThreadFunc != null) { + WebSocketWriteIOThread writeIOThread = webSocketWriterIOThreadFunc.apply(webSocket); + updateWriteIOThread(webSocket._channel, writeIOThread); + webSocket._writeIOThread = writeIOThread; + } } protected String createSessionid() { diff --git a/src/main/java/org/redkale/net/http/WebSocketServlet.java b/src/main/java/org/redkale/net/http/WebSocketServlet.java index ecbe55fd9..b42a40a23 100644 --- a/src/main/java/org/redkale/net/http/WebSocketServlet.java +++ b/src/main/java/org/redkale/net/http/WebSocketServlet.java @@ -286,7 +286,7 @@ public abstract class WebSocketServlet extends HttpServlet implements Resourcabl public void completed(Integer result, Void attachment) { webSocket._readHandler = new WebSocketReadHandler(response.getContext(), webSocket, restMessageConsumer); webSocket._writeHandler = new WebSocketWriteHandler(response.getContext(), webSocket); - response.getContext().updateWebSocketWriteIOThread(webSocket); + //response.getContext().updateWebSocketWriteIOThread(webSocket); Runnable createUseridHandler = () -> { CompletableFuture userFuture = webSocket.createUserid();