refactor: adapt to latest uWebSockets.js changes

Reference: https://github.com/uNetworking/uWebSockets.js/releases

Author: darrachequesne

lib/transports-uws/polling.ts

@@ -162,7 +162,9 @@ export class Polling extends Transport {
     const onEnd = (buffer) => {
       this.onData(buffer.toString());
       this.onDataRequestCleanup();
-      res.end("ok");
+      res.cork(() => {
+        res.end("ok");
+      });
     };
 
     res.onAborted(() => {
@@ -312,10 +314,12 @@ export class Polling extends Transport {
 
     const respond = (data) => {
       this.headers(this.req, headers);
-      Object.keys(headers).forEach((key) => {
-        this.res.writeHeader(key, String(headers[key]));
+      this.res.cork(() => {
+        Object.keys(headers).forEach((key) => {
+          this.res.writeHeader(key, String(headers[key]));
+        });
+        this.res.end(data);
       });
-      this.res.end(data);
       callback();
     };
 

lib/userver.ts

@@ -70,24 +70,25 @@ export class uServer extends BaseServer {
     (app as TemplatedApp)
       .any(path, this.handleRequest.bind(this))
       //
-      .ws(path, {
+      .ws<{ transport: any }>(path, {
         compression: options.compression,
         idleTimeout: options.idleTimeout,
         maxBackpressure: options.maxBackpressure,
         maxPayloadLength: this.opts.maxHttpBufferSize,
         upgrade: this.handleUpgrade.bind(this),
         open: (ws) => {
-          ws.transport.socket = ws;
-          ws.transport.writable = true;
-          ws.transport.emit("drain");
+          const transport = ws.getUserData().transport;
+          transport.socket = ws;
+          transport.writable = true;
+          transport.emit("drain");
         },
         message: (ws, message, isBinary) => {
-          ws.transport.onData(
+          ws.getUserData().transport.onData(
             isBinary ? message : Buffer.from(message).toString()
           );
         },
         close: (ws, code, message) => {
-          ws.transport.onClose(code, message);
+          ws.getUserData().transport.onClose(code, message);
         },
       });
   }
@@ -323,11 +324,13 @@ class ResponseWrapper {
   public end(data) {
     if (this.isAborted) return;
 
-    if (!this.statusWritten) {
-      // status will be inferred as "200 OK"
-      this.writeBufferedHeaders();
-    }
-    this.res.end(data);
+    this.res.cork(() => {
+      if (!this.statusWritten) {
+        // status will be inferred as "200 OK"
+        this.writeBufferedHeaders();
+      }
+      this.res.end(data);
+    });
   }
 
   public onData(fn) {
@@ -345,4 +348,10 @@ class ResponseWrapper {
       fn();
     });
   }
+
+  public cork(fn) {
+    if (this.isAborted) return;
+
+    this.res.cork(fn);
+  }
 }

package-lock.json

@@ -57,7 +57,7 @@
     "rimraf": "^3.0.2",
     "superagent": "^3.8.1",
     "typescript": "^4.4.3",
-    "uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.15.0"
+    "uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.30.0"
   },
   "scripts": {
     "compile": "rimraf ./build && tsc",