update

Author: tsctx

benchmarks/websocket-benchmark.mjs

@@ -71,35 +71,33 @@ experiments['undici'] = {
   binaries
 }
 
-// https://github.com/nodejs/undici/pull/3560
-if (typeof WebSocketStream === 'function') {
-  experiments['undici - stream'] = {
-    fn: (ws, binary) => {
-      /** @type {ReadableStreamDefaultReader<string | Uint8Array>} */
-      const reader = ws.reader
-      /** @type {WritableStreamDefaultWriter<string | BufferSource>} */
-      const writer = ws.writer
-
-      return async (ev) => {
-        ev.start()
-        await writer.write(binary)
-        await reader.read()
-        ev.end()
-      }
-    },
+experiments['undici - stream'] = {
+  fn: (ws, binary) => {
+    /** @type {ReadableStreamDefaultReader<string | Uint8Array>} */
+    const reader = ws.reader
+    /** @type {WritableStreamDefaultWriter<string | BufferSource>} */
+    const writer = ws.writer
+
+    return async (ev) => {
+      ev.start()
+      await writer.write(binary)
+      await reader.read()
+      ev.end()
+    }
+  },
 
-    connect: async (url) => {
-      const ws = new WebSocketStream(url)
+  connect: async (url) => {
+    const ws = new WebSocketStream(url)
 
-      const { readable, writable } = await ws.opened
-      const reader = readable.getReader()
-      const writer = writable.getWriter()
+    const { readable, writable } = await ws.opened
+    const reader = readable.getReader()
+    const writer = writable.getWriter()
 
-      return { reader, writer, close: () => ws.close() }
-    },
+    // @ts-ignore
+    return { reader, writer, close: () => ws.close() }
+  },
 
-    binaries
-  }
+  binaries
 }
 
 experiments['ws'] = {
@@ -200,7 +198,7 @@ function print (results) {
     const { bytes } = experimentsInfo[name]
 
     console.log(
-      `${name} -> ${formatBytes((bytes / average) * 1e9)}/s`
+      `${name}: transferred ${formatBytes((bytes / average) * 1e9)}/s`
     )
   }
 }