perf: cork socket for a micro task

websockets · Apr 1, 2024 · 8f35bed · 8f35bed
1 parent 2405c17
commit 8f35bed
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/lib/sender.js b/lib/sender.js
@@ -34,6 +34,12 @@ class Sender {
     }
 
     this._socket = socket;
+    this._corked = false;
+    this._uncork = () => {
+      if (this._corked) {
+        this._socket.uncork();
+      }
+    }
 
     this._firstFragment = true;
     this._compress = false;
@@ -195,6 +201,10 @@ class Sender {
     } else {
       this.sendFrame(Sender.frame(buf, options), cb);
     }
+
+    if (this._corked) {
+      this._socket.uncork();
+    }
   }
 
   /**
@@ -463,11 +473,15 @@ class Sender {
    * @private
    */
   sendFrame(list, cb) {
-    if (list.length === 2) {
+    if (!this._corked) {
+      this._corked = true;
       this._socket.cork();
+      queueMicrotask(this._uncork);
+    }
+
+    if (list.length === 2) {
       this._socket.write(list[0]);
       this._socket.write(list[1], cb);
-      this._socket.uncork();
     } else {
       this._socket.write(list[0], cb);
     }