update

Author: tsctx

benchmarks/websocket-benchmark.mjs

@@ -2,11 +2,9 @@
 
 import { bench } from './_util/runner.js'
 import { formatBytes } from './_util/index.js'
-import * as undici from '../index.js'
+import { WebSocket, WebSocketStream } from '../index.js'
 import { WebSocket as WsWebSocket } from 'ws'
 
-const { WebSocket } = undici
-
 /**
  * @type {Record<string, { fn: (ws: any, binary: string | Uint8Array) => import('./_util/runner.js').BenchMarkHandler; connect: (url: string) => Promise<any>; binaries: (string | Uint8Array)[] }>}
  */
@@ -74,35 +72,35 @@ experiments['undici'] = {
 }
 
 // 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()
-//       }
-//     },
-
-//     connect: async (url) => {
-//       const ws = new WebSocketStream(url)
-
-//       const { readable, writable } = await ws.opened
-//       const reader = readable.getReader()
-//       const writer = writable.getWriter()
-
-//       return { reader, writer, close: () => ws.close() }
-//     },
-
-//     binaries
-//   }
-// }
+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()
+      }
+    },
+
+    connect: async (url) => {
+      const ws = new WebSocketStream(url)
+
+      const { readable, writable } = await ws.opened
+      const reader = readable.getReader()
+      const writer = writable.getWriter()
+
+      return { reader, writer, close: () => ws.close() }
+    },
+
+    binaries
+  }
+}
 
 experiments['ws'] = {
   fn: (ws, binary) => {