Merge pull request #1127 from poolifier/combined-prs-branch
[poolifier.git] / examples / typescript / http-server-pool / fastify-cluster / src / worker.ts
index 4084bf71fb7b6461311e5a2490ed6ab6bbe9c7a3..283a223e3438aa07af36343368d6dd03de05017e 100644 (file)
@@ -3,7 +3,7 @@ import { ClusterWorker } from 'poolifier'
 import Fastify, { type FastifyInstance } from 'fastify'
 import type { WorkerData, WorkerResponse } from './types.js'
 
-const factorial: (n: number) => number = n => {
+const factorial: (n: number) => number = (n) => {
   if (n === 0) {
     return 1
   }
@@ -17,17 +17,18 @@ class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
     workerData?: WorkerData
   ): Promise<WorkerResponse> => {
     const { port } = workerData as 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) }
     })
@@ -35,12 +36,16 @@ class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
     await FastifyWorker.fastify.listen({ port })
     return {
       status: true,
-      port: (FastifyWorker.fastify.server.address() as AddressInfo).port
+      port: (FastifyWorker.fastify.server.address() as AddressInfo)?.port
     }
   }
 
   public constructor () {
-    super(FastifyWorker.startFastify)
+    super(FastifyWorker.startFastify, {
+      killHandler: async () => {
+        await FastifyWorker.fastify.close()
+      }
+    })
   }
 }