use cluster

Author: tsctx

benchmarks/websocket/server/echo.mjs

@@ -1,10 +1,28 @@
+import cluster from 'node:cluster'
 import { WebSocketServer } from 'ws'
+import { cpus } from 'node:os'
 
-const server = new WebSocketServer({ port: 5001 })
+if (cluster.isPrimary) {
+  let cpu = cpus().length
+  while (cpu-- > 0) {
+    cluster.fork()
+  }
+} else {
+  const server = new WebSocketServer({
+    maxPayload: 600 * 1024 * 1024,
+    perMessageDeflate: false,
+    clientTracking: false,
+    port: 5001
+  })
+
+  server.on('connection', (socket) => {
+    socket.on('message', (data, isBinary) => {
+      socket.send(data, { binary: isBinary })
+      // socket.close();
+    })
+  })
 
-server.on('connection', (socket) => {
-  socket.on('message', (data, isBinary) => {
-    socket.send(data, { binary: isBinary })
-    // socket.close();
+  cluster.on('exit', () => {
+    server.close()
   })
-})
+}

benchmarks/websocket/server/simple.mjs

@@ -1,13 +1,31 @@
+import cluster from 'node:cluster'
 import { WebSocketServer } from 'ws'
+import { cpus } from 'node:os'
 
-const server = new WebSocketServer({ port: 5001 })
+if (cluster.isPrimary) {
+  let cpu = cpus().length
+  while (cpu-- > 0) {
+    cluster.fork()
+  }
+} else {
+  const server = new WebSocketServer({
+    maxPayload: 600 * 1024 * 1024,
+    perMessageDeflate: false,
+    clientTracking: false,
+    port: 5001
+  })
+
+  // Workaround for https://github.com/nodejs/undici/issues/3202
+  const emptyBuffer = Buffer.allocUnsafe(1)
 
-// 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();
+    })
+  })
 
-server.on('connection', (socket) => {
-  socket.on('message', (_data, _isBinary) => {
-    socket.send(emptyBuffer)
-    // socket.close();
+  cluster.on('exit', () => {
+    server.close()
   })
-})
+}

benchmarks/websocket/websocket-echo.mjs

@@ -1,3 +1,7 @@
+// --------------------SERVER--------------------
+// -> /server/echo.mjs
+// ----------------------------------------------
+
 import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { randomBytes } from 'node:crypto'

benchmarks/websocket/websocket-open-connection.mjs

@@ -1,3 +1,7 @@
+// --------------------SERVER--------------------
+// -> /server/simple.mjs
+// ----------------------------------------------
+
 import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { bench, run, group } from 'mitata'
@@ -15,7 +19,6 @@ const url = 'http://localhost:5001'
 group('open connection', () => {
   bench('ws', () => {
     const ws = new WsWebSocket(url)
-    ws.binaryType = 'fragments'
     return new Promise((resolve, reject) => {
       ws.addEventListener('open', () => {
         resolve()

benchmarks/websocket/websocket-send-buffer.mjs

@@ -1,3 +1,7 @@
+// --------------------SERVER--------------------
+// -> /server/simple.mjs
+// ----------------------------------------------
+
 import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { randomBytes } from 'node:crypto'

benchmarks/websocket/websocket-send-string.mjs

@@ -1,3 +1,7 @@
+// --------------------SERVER--------------------
+// -> /server/simple.mjs
+// ----------------------------------------------
+
 import { WebSocket as WsWebSocket } from 'ws'
 import { WebSocket as UndiciWebSocket } from '../../index.js'
 import { randomBytes } from 'node:crypto'