Merge dependabot/npm_and_yarn/examples/typescript/http-server-pool/express-hybrid...
[poolifier.git] / examples / typescript / http-server-pool / fastify-cluster / src / worker.ts
index 3f7b3661f99cd2f51082366eadfc8cbe17ba804a..e491fa8f4b05c3ee4712740e1e669a102775108e 100644 (file)
@@ -1,35 +1,44 @@
 import type { AddressInfo } from 'node:net'
-import { ClusterWorker } from 'poolifier'
+
 import Fastify, { type FastifyInstance } from 'fastify'
-import type { WorkerData, WorkerResponse } from './types.js'
+import { ClusterWorker } from 'poolifier'
 
-const factorial: (n: number) => number = (n) => {
-  if (n === 0) {
-    return 1
-  }
-  return factorial(n - 1) * n
-}
+import type { WorkerData, WorkerResponse } from './types.js'
 
 class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   private static fastify: FastifyInstance
 
+  private static readonly factorial = (n: number | bigint): bigint => {
+    if (n === 0 || n === 1) {
+      return 1n
+    } else {
+      n = BigInt(n)
+      let factorial = 1n
+      for (let i = 1n; i <= n; i++) {
+        factorial *= i
+      }
+      return factorial
+    }
+  }
+
   private static readonly startFastify = async (
     workerData?: WorkerData
   ): Promise<WorkerResponse> => {
-    const { port } = workerData as WorkerData
+    const { port } = workerData!
+
     FastifyWorker.fastify = Fastify({
       logger: true
     })
 
-    FastifyWorker.fastify.all('/api/echo', (request) => {
+    FastifyWorker.fastify.all('/api/echo', request => {
       return request.body
     })
 
     FastifyWorker.fastify.get<{
       Params: { number: number }
-    }>('/api/factorial/:number', (request) => {
+    }>('/api/factorial/:number', request => {
       const { number } = request.params
-      return { number: factorial(number) }
+      return { number: FastifyWorker.factorial(number).toString() }
     })
 
     await FastifyWorker.fastify.listen({ port })
@@ -40,7 +49,11 @@ class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   }
 
   public constructor () {
-    super(FastifyWorker.startFastify)
+    super(FastifyWorker.startFastify, {
+      killHandler: async () => {
+        await FastifyWorker.fastify.close()
+      }
+    })
   }
 }