add websocket-stream

benchmarks/websocket-benchmark.mjs

@@ -1,18 +1,17 @@
 // @ts-check
 
 import { bench } from './_util/runner.js'
-import { WebSocket } from '../index.js'
+import * as undici from '../index.js'
 import { WebSocket as WsWebSocket } from 'ws'
 
-/**
- * @typedef {object} WebSocketImpl
- * @property {(buffer: Uint8Array | string) => void} send
- * @property {(code?: number, reason?: string) => void} close
- * @property {string} binaryType
- */
+const {
+  WebSocket,
+  // @ts-ignore https://github.com/nodejs/undici/pull/3560
+  WebSocketStream
+} = undici
 
 /**
- * @type {Record<string, { fn: (ws: WebSocketImpl, binary: string | Uint8Array) => import('./_util/runner.js').BenchMarkHandler; connect: (url: string) => Promise<WebSocketImpl>; binaries: (string | Uint8Array)[] }>}
+ * @type {Record<string, { fn: (ws: any, binary: string | Uint8Array) => import('./_util/runner.js').BenchMarkHandler; connect: (url: string) => Promise<any>; binaries: (string | Uint8Array)[] }>}
  */
 const experiments = {}
 /**
@@ -21,7 +20,7 @@ const experiments = {}
 const experimentsInfo = {}
 
 /**
- * @type {WebSocketImpl[]}
+ * @type {any[]}
  */
 const connections = []
 
@@ -77,6 +76,32 @@ experiments['undici'] = {
   binaries
 }
 
+if (typeof WebSocketStream === 'function') {
+  experiments['undici - stream'] = {
+    fn: (ws, binary) => {
+      const { readable, writable } = ws
+      /** @type {ReadableStreamDefaultReader<string | Uint8Array>} */
+      const reader = readable.getReader()
+      /** @type {WritableStreamDefaultWriter<string | BufferSource>} */
+      const writer = writable.getWriter()
+      return async (ev) => {
+        ev.start()
+        await writer.write(binary)
+        await reader.read()
+        ev.end()
+      }
+    },
+
+    connect: async (url) => {
+      const ws = new WebSocketStream(url)
+
+      return await ws.opened
+    },
+
+    binaries
+  }
+}
+
 experiments['ws'] = {
   fn: (ws, binary) => {
     if (!(ws instanceof WsWebSocket)) {