X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;ds=sidebyside;f=examples%2Ftypescript%2Fwebsocket-server-pool%2Fws-cluster%2Fsrc%2Fworker.ts;h=9ec81eb5db8b8c8fc5b7b5d4ad94d522629df89c;hb=6bb2a64f78c7ba84e4d3690d18c2f8ca9e418069;hp=f261a6d74413f34ee7ab6e368e03a204db6bd8b9;hpb=80ccdab6c5ce5e7e6305c15dd29257d0acc9e302;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 f261a6d7..9ec81eb5 100644
--- a/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
+++ b/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
@@ -1,5 +1,6 @@
 import { ClusterWorker } from 'poolifier'
 import { type RawData, WebSocketServer } from 'ws'
+
 import {
   type DataPayload,
   type MessagePayload,
@@ -8,57 +9,68 @@ import {
   type WorkerResponse
 } from './types.js'
 
-const factorial: (n: number) => number = n => {
-  if (n === 0) {
-    return 1
+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
   }
-  return factorial(n - 1) * n
-}
 
-const startWebSocketServer = (workerData?: WorkerData): WorkerResponse => {
-  const { port } = workerData as WorkerData
-  const wss = new WebSocketServer({ port }, () => {
-    console.info(
-      `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
-    )
-  })
+  private static readonly startWebSocketServer = (
+    workerData?: WorkerData
+  ): WorkerResponse => {
+    const { port } = workerData!
 
-  wss.on('connection', ws => {
-    ws.on('error', console.error)
-    ws.on('message', (message: RawData) => {
-      const { type, data } = JSON.parse(
-        // eslint-disable-next-line @typescript-eslint/no-base-to-string
-        message.toString()
-      ) as MessagePayload<DataPayload>
-      switch (type) {
-        case MessageType.echo:
-          ws.send(
-            JSON.stringify({
-              type: MessageType.echo,
-              data
-            })
-          )
-          break
-        case MessageType.factorial:
-          ws.send(
-            JSON.stringify({
-              type: MessageType.factorial,
-              data: { number: factorial(data.number as number) }
-            })
-          )
-          break
-      }
+    WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
+      console.info(
+        `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
+      )
     })
-  })
-  return {
-    status: true,
-    port: wss.options.port
+
+    WebSocketServerWorker.wss.on('connection', ws => {
+      ws.on('error', console.error)
+      ws.on('message', (message: RawData) => {
+        const { type, data } = JSON.parse(
+          // eslint-disable-next-line @typescript-eslint/no-base-to-string
+          message.toString()
+        ) as MessagePayload<DataPayload>
+        switch (type) {
+          case MessageType.echo:
+            ws.send(
+              JSON.stringify({
+                type: MessageType.echo,
+                data
+              })
+            )
+            break
+          case MessageType.factorial:
+            ws.send(
+              JSON.stringify({
+                type: MessageType.factorial,
+                data: {
+                  number: WebSocketServerWorker.factorial(data.number!)
+                }
+              })
+            )
+            break
+        }
+      })
+    })
+    return {
+      status: true,
+      port: WebSocketServerWorker.wss.options.port
+    }
   }
-}
 
-class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   public constructor () {
-    super(startWebSocketServer)
+    super(WebSocketServerWorker.startWebSocketServer, {
+      killHandler: () => {
+        WebSocketServerWorker.wss.close()
+      }
+    })
   }
 }