chore: migrate to eslint 9

[poolifier.git] / examples / typescript / websocket-server-pool / ws-hybrid / src / websocket-server-worker.ts

diff --git a/examples/typescript/websocket-server-pool/ws-hybrid/src/websocket-server-worker.ts b/examples/typescript/websocket-server-pool/ws-hybrid/src/websocket-server-worker.ts
index dfcba66789934c53b05c2f0ec10af81a24c93804..6aa993d29ebd9d725cfed426535107061dad46e5 100644 (file)
--- a/examples/typescript/websocket-server-pool/ws-hybrid/src/websocket-server-worker.ts
+++ b/examples/typescript/websocket-server-pool/ws-hybrid/src/websocket-server-worker.ts
@@ -1,79 +1,111 @@
-import { ClusterWorker } from 'poolifier'
+import {
+  availableParallelism,
+  ClusterWorker,
+  DynamicThreadPool,
+} from 'poolifier'
 import { type RawData, WebSocketServer } from 'ws'
+
 import {
   type ClusterWorkerData,
   type ClusterWorkerResponse,
   type DataPayload,
   type MessagePayload,
-  MessageType
+  MessageType,
+  type ThreadWorkerData,
+  type ThreadWorkerResponse,
 } from './types.js'
-import { requestHandlerPool } from './request-handler-pool.js'
 
 const emptyFunction = (): void => {
-  /** Intentional */
+  /* Intentional */
 }
 
-const startWebSocketServer = (
-  workerData?: ClusterWorkerData
-): ClusterWorkerResponse => {
-  const { port } = workerData as ClusterWorkerData
-  const wss = new WebSocketServer({ port }, () => {
-    console.info(
-      `⚡️[ws server]: WebSocket server is started on cluster worker at ws://localhost:${port}/`
+class WebSocketServerWorker extends ClusterWorker<
+  ClusterWorkerData,
+  ClusterWorkerResponse
+> {
+  private static wss: WebSocketServer
+  private static requestHandlerPool: DynamicThreadPool<
+    ThreadWorkerData<DataPayload>,
+    ThreadWorkerResponse<DataPayload>
+  >
+
+  private static readonly startWebSocketServer = (
+    workerData?: ClusterWorkerData
+  ): ClusterWorkerResponse => {
+    const { port, workerFile, minWorkers, maxWorkers, ...poolOptions } =
+      // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+      workerData!
+
+    WebSocketServerWorker.requestHandlerPool = new DynamicThreadPool<
+      ThreadWorkerData<DataPayload>,
+      ThreadWorkerResponse<DataPayload>
+    >(
+      minWorkers ?? 1,
+      maxWorkers ?? availableParallelism(),
+      workerFile,
+      poolOptions
     )
-  })
 
-  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:
-          requestHandlerPool
-            .execute({ data }, 'echo')
-            .then(response => {
-              ws.send(
-                JSON.stringify({
-                  type: MessageType.echo,
-                  data: response.data
-                })
-              )
-              return null
-            })
-            .catch(emptyFunction)
-          break
-        case MessageType.factorial:
-          requestHandlerPool
-            .execute({ data }, 'factorial')
-            .then(response => {
-              ws.send(
-                JSON.stringify({
-                  type: MessageType.factorial,
-                  data: response.data
-                })
-              )
-              return null
-            })
-            .catch(emptyFunction)
-          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(
+          message.toString()
+        ) as MessagePayload<DataPayload>
+        switch (type) {
+          case MessageType.echo:
+            WebSocketServerWorker.requestHandlerPool
+              .execute({ data }, 'echo')
+              .then(response => {
+                ws.send(
+                  JSON.stringify({
+                    type: MessageType.echo,
+                    data: response.data,
+                  })
+                )
+                return undefined
+              })
+              .catch(emptyFunction)
+            break
+          case MessageType.factorial:
+            WebSocketServerWorker.requestHandlerPool
+              .execute({ data }, 'factorial')
+              .then(response => {
+                ws.send(
+                  JSON.stringify(
+                    {
+                      type: MessageType.factorial,
+                      data: response.data,
+                    },
+                    (_, v) => (typeof v === 'bigint' ? v.toString() : v)
+                  )
+                )
+                return undefined
+              })
+              .catch(emptyFunction)
+            break
+        }
+      })
+    })
+    return {
+      status: true,
+      port: WebSocketServerWorker.wss.options.port,
+    }
   }
-}
 
-class WebSocketServerWorker extends ClusterWorker<
-ClusterWorkerData,
-ClusterWorkerResponse
-> {
   public constructor () {
-    super(startWebSocketServer)
+    super(WebSocketServerWorker.startWebSocketServer, {
+      killHandler: async () => {
+        await WebSocketServerWorker.requestHandlerPool.destroy()
+        WebSocketServerWorker.wss.close()
+      },
+    })
   }
 }