Clear socket.waiting before invoking callbacks (#819)

SOHU-Co · Feb 20, 2018 · 850e2d6 · 850e2d6
1 parent b56571f
commit 850e2d6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/client.js b/lib/client.js
@@ -759,6 +759,10 @@ Client.prototype.handleReceivedData = function (socket) {
   var size = buffer.readUInt32BE(0) + 4;
 
   if (buffer.length >= size) {
+    if (socket.longpolling) {
+      socket.waiting = false;
+    }
+
     var resp = buffer.shallowSlice(0, size);
     var correlationId = resp.readUInt32BE(4);
     var handlers = this.unqueueCallback(socket, correlationId);
@@ -772,9 +776,6 @@ Client.prototype.handleReceivedData = function (socket) {
       logger.error(`missing handlers for Correlation ID: ${correlationId}`);
     }
     buffer.consume(size);
-    if (socket.longpolling) {
-      socket.waiting = false;
-    }
   } else {
     return;
   }