From 4bdbf0f4939c2f0f74ccac6760a43fc7f7e8e7cd Mon Sep 17 00:00:00 2001
From: redkale <redkale@qq.com>
Date: Wed, 5 Apr 2023 18:43:38 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96WebSocket?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/org/redkale/net/http/HttpContext.java      | 8 +++++---
 src/main/java/org/redkale/net/http/WebSocketServlet.java | 2 +-
 2 files changed, 6 insertions(+), 4 deletions(-)

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<Serializable> userFuture = webSocket.createUserid();