init

bench.mjs

@@ -0,0 +1,79 @@
+// --------------------SERVER--------------------
+// -> /server/simple.mjs
+// ----------------------------------------------
+
+import { WebSocket as WsWebSocket } from 'ws'
+import { WebSocket as UndiciWebSocket } from './index.js'
+import { bench, run, group } from 'mitata'
+
+const __GLOBAL_WEBSOCKET__ = true
+
+const binary = new Uint8Array(0)
+
+const url = 'http://localhost:5001'
+
+const connections = []
+
+group('send', () => {
+ {
+ const ws = new WsWebSocket(url)
+ let _resolve
+ ws.addEventListener('message', () => {
+ _resolve()
+ })
+ bench('ws', () => {
+ return new Promise((resolve, reject) => {
+ ws.send(binary)
+ _resolve = resolve
+ })
+ })
+ connections.push(ws)
+ }
+ {
+ const ws = new UndiciWebSocket(url)
+ let _resolve
+ ws.addEventListener('message', () => {
+ _resolve()
+ })
+ bench('undici', () => {
+ return new Promise((resolve, reject) => {
+ ws.send(binary)
+ _resolve = resolve
+ })
+ })
+ connections.push(ws)
+ }
+ if (__GLOBAL_WEBSOCKET__ && typeof globalThis.WebSocket === 'function') {
+ const ws = new globalThis.WebSocket(url)
+ let _resolve
+ ws.addEventListener('message', () => {
+ _resolve()
+ })
+ bench('undici - global', () => {
+ return new Promise((resolve, reject) => {
+ ws.send(binary)
+ _resolve = resolve
+ })
+ })
+ connections.push(ws)
+ }
+})
+
+for (const ws of connections) {
+ // for fairness
+ ws.binaryType = 'arraybuffer'
+ await new Promise((resolve, reject) => {
+ ws.addEventListener('open', () => {
+ resolve()
+ })
+ ws.addEventListener('error', (err) => {
+ reject(err)
+ })
+ })
+}
+
+await run()
+
+for (const ws of connections) {
+ ws.close()
+}