split server

Author: tsctx

benchmarks/websocket/server/echo.mjs

@@ -0,0 +1,10 @@
+import { WebSocketServer } from 'ws'
+
+const server = new WebSocketServer({ port: 5001 })
+
+server.on('connection', (socket) => {
+  socket.on('message', (data, isBinary) => {
+    socket.send(data, { binary: isBinary })
+    // socket.close();
+  })
+})

benchmarks/websocket/server/simple.mjs

@@ -0,0 +1,13 @@
+import { WebSocketServer } from 'ws'
+
+const server = new WebSocketServer({ port: 5001 })
+
+// Workaround for https://github.com/nodejs/undici/issues/3202
+const emptyBuffer = Buffer.allocUnsafe(1)
+
+server.on('connection', (socket) => {
+  socket.on('message', (_data, _isBinary) => {
+    socket.send(emptyBuffer)
+    // socket.close();
+  })
+})

benchmarks/websocket/websocket-echo.mjs

@@ -1,4 +1,4 @@
-import { WebSocket as WsWebSocket, WebSocketServer } from 'ws'
+import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { randomBytes } from 'node:crypto'
 import { bench, run, group } from 'mitata'
@@ -12,21 +12,9 @@ if (__GLOBAL_WEBSOCKET__ && typeof globalThis.WebSocket === 'function') {
   GlobalWebSocket = globalThis.WebSocket
 }
 
-const server = new WebSocketServer({ port: 5001 })
 const binary = randomBytes(__BINARY_SIZE__)
 
-server.on('connection', (socket) => {
-  socket.on('message', (data, _isBinary) => {
-    socket.send(data)
-    // socket.close();
-  })
-})
-
-await new Promise((resolve, _reject) => {
-  server.on('listening', resolve)
-})
-
-const url = `http://localhost:${server.address().port}`
+const url = 'http://localhost:5001'
 
 const connections = []
 
@@ -90,8 +78,6 @@ for (const ws of connections) {
 
 await run()
 
-server.close()
-
 for (const ws of connections) {
   ws.close()
 }

benchmarks/websocket/websocket-send-buffer.mjs

@@ -1,4 +1,4 @@
-import { WebSocket as WsWebSocket, WebSocketServer } from 'ws'
+import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { randomBytes } from 'node:crypto'
 import { bench, run, group } from 'mitata'
@@ -12,24 +12,9 @@ if (__GLOBAL_WEBSOCKET__ && typeof globalThis.WebSocket === 'function') {
   GlobalWebSocket = globalThis.WebSocket
 }
 
-const server = new WebSocketServer({ port: 5001 })
 const binary = randomBytes(__BINARY_SIZE__)
 
-// Workaround for https://github.com/nodejs/undici/issues/3202
-const emptyBuffer = Buffer.allocUnsafe(1)
-
-server.on('connection', (socket) => {
-  socket.on('message', (_data, _isBinary) => {
-    socket.send(emptyBuffer)
-    // socket.close();
-  })
-})
-
-await new Promise((resolve, _reject) => {
-  server.on('listening', resolve)
-})
-
-const url = `http://localhost:${server.address().port}`
+const url = 'http://localhost:5001'
 
 const connections = []
 
@@ -93,8 +78,6 @@ for (const ws of connections) {
 
 await run()
 
-server.close()
-
 for (const ws of connections) {
   ws.close()
 }

benchmarks/websocket/websocket-send-string.mjs

@@ -1,4 +1,4 @@
-import { WebSocket as WsWebSocket, WebSocketServer } from 'ws'
+import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { randomBytes } from 'node:crypto'
 import { bench, run, group } from 'mitata'
@@ -12,24 +12,9 @@ if (__GLOBAL_WEBSOCKET__ && typeof globalThis.WebSocket === 'function') {
   GlobalWebSocket = globalThis.WebSocket
 }
 
-const server = new WebSocketServer({ port: 5001 })
 const binary = randomBytes(__BINARY_SIZE__).toString('utf-8')
 
-// Workaround for https://github.com/nodejs/undici/issues/3202
-const emptyBuffer = Buffer.allocUnsafe(1)
-
-server.on('connection', (socket) => {
-  socket.on('message', (_data, _isBinary) => {
-    socket.send(emptyBuffer)
-    // socket.close();
-  })
-})
-
-await new Promise((resolve, _reject) => {
-  server.on('listening', resolve)
-})
-
-const url = `http://localhost:${server.address().port}`
+const url = 'http://localhost:5001'
 
 const connections = []
 
@@ -93,8 +78,6 @@ for (const ws of connections) {
 
 await run()
 
-server.close()
-
 for (const ws of connections) {
   ws.close()
 }