X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=examples%2Ftypescript%2Fwebsocket-server-pool%2Fws-cluster%2Fsrc%2Fworker.ts;h=5e758eb7d3f845e1d8508bf907bda9fa7fcbcc3d;hb=dc76cd7f73cb89e123971eb5f45dff81733a3c98;hp=56060b2d28b043d40eb3dad1cf4f7ed52d33db8b;hpb=8a199a0a5716a52b4f54f212c2d139b7fddee9bc;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 56060b2d..5e758eb7 100644
--- a/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
+++ b/examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts
@@ -15,17 +15,21 @@ const factorial: (n: number) => number = n => {
   return factorial(n - 1) * n
 }
 
-const startWebSocketServer = (workerData?: WorkerData): WorkerResponse => {
-  try {
-    const wss = new WebSocketServer({ port: workerData?.port }, () => {
+class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
+  private static wss: WebSocketServer
+
+  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 on worker at ws://localhost:${
-          workerData?.port as number
-        }/`
+        `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
       )
     })
 
-    wss.on('connection', ws => {
+    WebSocketServerWorker.wss.on('connection', ws => {
       ws.on('error', console.error)
       ws.on('message', (message: RawData) => {
         const { type, data } = JSON.parse(
@@ -54,19 +58,16 @@ const startWebSocketServer = (workerData?: WorkerData): WorkerResponse => {
     })
     return {
       status: true,
-      port: wss.options.port
-    }
-  } catch (err) {
-    return {
-      status: false,
-      error: err as Error
+      port: WebSocketServerWorker.wss.options.port
     }
   }
-}
 
-class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   public constructor () {
-    super(startWebSocketServer)
+    super(WebSocketServerWorker.startWebSocketServer, {
+      killHandler: () => {
+        WebSocketServerWorker.wss.close()
+      }
+    })
   }
 }