examples/typescript/http-server-pool/fastify-cluster/src/worker.ts

   1 import type { AddressInfo } from 'node:net'
   2
   3 import Fastify, { type FastifyInstance } from 'fastify'
   4 import { ClusterWorker } from 'poolifier'
   5
   6 import type { WorkerData, WorkerResponse } from './types.js'
   7
   8 class FastifyWorker extends ClusterWorker<WorkerData, WorkerResponse> {
   9   private static fastify: FastifyInstance
  10
  11   private static readonly factorial = (n: number | bigint): bigint => {
  12     if (n === 0 || n === 1) {
  13       return 1n
  14     } else {
  15       n = BigInt(n)
  16       let factorial = 1n
  17       for (let i = 1n; i <= n; i++) {
  18         factorial *= i
  19       }
  20       return factorial
  21     }
  22   }
  23
  24   private static readonly startFastify = async (
  25     workerData?: WorkerData
  26   ): Promise<WorkerResponse> => {
  27     // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
  28     const { port } = workerData!
  29
  30     FastifyWorker.fastify = Fastify({
  31       logger: true,
  32     })
  33
  34     FastifyWorker.fastify.all('/api/echo', request => {
  35       return request.body
  36     })
  37
  38     FastifyWorker.fastify.get<{
  39       Params: { number: number }
  40     }>('/api/factorial/:number', request => {
  41       const { number } = request.params
  42       return { number: FastifyWorker.factorial(number).toString() }
  43     })
  44
  45     await FastifyWorker.fastify.listen({ port })
  46     return {
  47       status: true,
  48       port: (FastifyWorker.fastify.server.address() as AddressInfo).port,
  49     }
  50   }
  51
  52   public constructor () {
  53     super(FastifyWorker.startFastify, {
  54       killHandler: async () => {
  55         await FastifyWorker.fastify.close()
  56       },
  57     })
  58   }
  59 }
  60
  61 export const fastifyWorker = new FastifyWorker()