build(deps-dev): apply updates
[poolifier.git] / examples / typescript / websocket-server-pool / ws-cluster / src / worker.ts
index 18ece026c891efb3106499667027a1a2aba0b73a..d9091e7f0eb1c27ea0ecf208d9495a4e8ca4578f 100644 (file)
@@ -1,31 +1,39 @@
 import { ClusterWorker } from 'poolifier'
 import { type RawData, WebSocketServer } from 'ws'
+
 import {
   type DataPayload,
   type MessagePayload,
   MessageType,
   type WorkerData,
-  type WorkerResponse
+  type WorkerResponse,
 } from './types.js'
 
 class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   private static wss: WebSocketServer
 
-  private static readonly factorial = (n: number): number => {
-    if (n === 0) {
-      return 1
+  private static readonly factorial = (n: number | bigint): bigint => {
+    if (n === 0 || n === 1) {
+      return 1n
+    } else {
+      n = BigInt(n)
+      let factorial = 1n
+      for (let i = 1n; i <= n; i++) {
+        factorial *= i
+      }
+      return factorial
     }
-    return WebSocketServerWorker.factorial(n - 1) * n
   }
 
   private static readonly startWebSocketServer = (
     workerData?: WorkerData
   ): WorkerResponse => {
+    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
     const { port } = workerData!
 
     WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
       console.info(
-        `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
+        `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port.toString()}/`
       )
     })
 
@@ -41,18 +49,22 @@ class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
             ws.send(
               JSON.stringify({
                 type: MessageType.echo,
-                data
+                data,
               })
             )
             break
           case MessageType.factorial:
             ws.send(
-              JSON.stringify({
-                type: MessageType.factorial,
-                data: {
-                  number: WebSocketServerWorker.factorial(data.number!)
-                }
-              })
+              JSON.stringify(
+                {
+                  type: MessageType.factorial,
+                  data: {
+                    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+                    number: WebSocketServerWorker.factorial(data.number!),
+                  },
+                },
+                (_, v: unknown) => (typeof v === 'bigint' ? v.toString() : v)
+              )
             )
             break
         }
@@ -60,7 +72,7 @@ class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
     })
     return {
       status: true,
-      port: WebSocketServerWorker.wss.options.port
+      port: WebSocketServerWorker.wss.options.port,
     }
   }
 
@@ -68,7 +80,7 @@ class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
     super(WebSocketServerWorker.startWebSocketServer, {
       killHandler: () => {
         WebSocketServerWorker.wss.close()
-      }
+      },
     })
   }
 }