Fix client cant be disconnected

sundb · Nov 19, 2024 · bfdc208 · bfdc208
1 parent 55c2b51
commit bfdc208
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/networking.c b/src/networking.c
@@ -2766,14 +2766,10 @@ int processInputBuffer(client *c) {
 
         if (c->reqtype == PROTO_REQ_INLINE) {
             if (processInlineBuffer(c) != C_OK) {
-                if (c->running_tid != IOTHREAD_MAIN_THREAD_ID && c->read_error)
-                    putInPendingClienstForMainThread(c);
                 break;
             }
         } else if (c->reqtype == PROTO_REQ_MULTIBULK) {
             if (processMultibulkBuffer(c) != C_OK) {
-                if (c->running_tid != IOTHREAD_MAIN_THREAD_ID && c->read_error)
-                    putInPendingClienstForMainThread(c);
                 break;
             }
         } else {
@@ -2961,8 +2957,13 @@ void readQueryFromClient(connection *conn) {
          c = NULL;
 
 done:
-    if (c && c->running_tid == IOTHREAD_MAIN_THREAD_ID && c->read_error)
-        handleClientReadError(c);
+    if (c && c->read_error) {
+        if (c->running_tid == IOTHREAD_MAIN_THREAD_ID) {
+            handleClientReadError(c);
+        } else {
+            putInPendingClienstForMainThread(c);
+        }
+    }
 
     if (c && c->in_reusable_querybuf) {
         serverAssert(c->qb_pos == 0); /* Ensure the client's query buffer is trimmed in processInputBuffer */