X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=examples%2Ftypescript%2Fwebsocket-server-pool%2Fws-cluster%2Fsrc%2Fworker.ts;h=0e8dc20a95fd8d2f3f161381d07f90a36e60b933;hb=240f52487f4ea0395633393dab4155ea4a71597f;hp=c82c7e636f8199495720eccfa71a44db925c899b;hpb=3d49c6d2f943532974cefa71f13af2c313555d95;p=poolifier.git

diff --git a/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts b/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
index c82c7e63..0e8dc20a 100644
--- a/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
+++ b/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
@@ -8,20 +8,21 @@ import {
   type WorkerResponse
 } from './types.js'
 
-const factorial: (n: number) => number = n => {
-  if (n === 0) {
-    return 1
-  }
-  return factorial(n - 1) * n
-}
-
 class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   private static wss: WebSocketServer
 
+  private static readonly factorial = (n: number): number => {
+    if (n === 0) {
+      return 1
+    }
+    return WebSocketServerWorker.factorial(n - 1) * n
+  }
+
   private static readonly startWebSocketServer = (
     workerData?: WorkerData
   ): WorkerResponse => {
     const { port } = workerData as WorkerData
+
     WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
       console.info(
         `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
@@ -48,7 +49,9 @@ class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
             ws.send(
               JSON.stringify({
                 type: MessageType.factorial,
-                data: { number: factorial(data.number as number) }
+                data: {
+                  number: WebSocketServerWorker.factorial(data.number as number)
+                }
               })
             )
             break
@@ -62,7 +65,11 @@ class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   }
 
   public constructor () {
-    super(WebSocketServerWorker.startWebSocketServer)
+    super(WebSocketServerWorker.startWebSocketServer, {
+      killHandler: () => {
+        WebSocketServerWorker.wss.close()
+      }
+    })
   }
 }