Merge dependabot/npm_and_yarn/examples/typescript/websocket-server-pool/ws-hybrid...
[poolifier.git] / examples / typescript / http-server-pool / fastify-cluster / src / worker.ts
index f679b4a6a9e3883db25909ec52f780259d319fa7..5fca89058695c7a45614b15d2eeb10bd690d9491 100644 (file)
@@ -1,44 +1,51 @@
 import type { AddressInfo } from 'node:net'
 import { ClusterWorker } from 'poolifier'
-import Fastify from 'fastify'
+import Fastify, { type FastifyInstance } from 'fastify'
 import type { WorkerData, WorkerResponse } from './types.js'
 
-const factorial: (n: number) => number = n => {
-  if (n === 0) {
-    return 1
+class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
+  private static fastify: FastifyInstance
+
+  private static readonly factorial = (n: number): number => {
+    if (n === 0) {
+      return 1
+    }
+    return FastifyWorker.factorial(n - 1) * n
   }
-  return factorial(n - 1) * n
-}
 
-const startFastify = async (
-  workerData?: WorkerData
-): Promise<WorkerResponse> => {
-  const { port } = workerData as WorkerData
-  const fastify = Fastify({
-    logger: true
-  })
-
-  fastify.all('/api/echo', request => {
-    return request.body
-  })
-
-  fastify.get<{
-    Params: { number: number }
-  }>('/api/factorial/:number', request => {
-    const { number } = request.params
-    return { number: factorial(number) }
-  })
-
-  await fastify.listen({ port })
-  return {
-    status: true,
-    port: (fastify.server.address() as AddressInfo).port
+  private static readonly startFastify = async (
+    workerData?: WorkerData
+  ): Promise<WorkerResponse> => {
+    const { port } = workerData!
+
+    FastifyWorker.fastify = Fastify({
+      logger: true
+    })
+
+    FastifyWorker.fastify.all('/api/echo', request => {
+      return request.body
+    })
+
+    FastifyWorker.fastify.get<{
+      Params: { number: number }
+    }>('/api/factorial/:number', request => {
+      const { number } = request.params
+      return { number: FastifyWorker.factorial(number) }
+    })
+
+    await FastifyWorker.fastify.listen({ port })
+    return {
+      status: true,
+      port: (FastifyWorker.fastify.server.address() as AddressInfo).port
+    }
   }
-}
 
-class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   public constructor () {
-    super(startFastify)
+    super(FastifyWorker.startFastify, {
+      killHandler: async () => {
+        await FastifyWorker.fastify.close()
+      }
+    })
   }
 }