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

   1 import type { AddressInfo } from 'node:net'
   2 import { dirname, extname, join } from 'node:path'
   3 import { fileURLToPath } from 'node:url'
   4 import { ClusterWorker, availableParallelism } from 'poolifier'
   5 import Fastify from 'fastify'
   6 import type { ClusterWorkerData, ClusterWorkerResponse } from './types.js'
   7 import { fastifyPoolifier } from './fastify-poolifier.js'
   8
   9 const startFastify = async (
  10   workerData?: ClusterWorkerData
  11 ): Promise<ClusterWorkerResponse> => {
  12   const { port } = workerData as ClusterWorkerData
  13   const fastify = Fastify({
  14     logger: true
  15   })
  16
  17   const requestHandlerWorkerFile = join(
  18     dirname(fileURLToPath(import.meta.url)),
  19     `request-handler-worker${extname(fileURLToPath(import.meta.url))}`
  20   )
  21
  22   await fastify.register(fastifyPoolifier, {
  23     workerFile: requestHandlerWorkerFile,
  24     maxWorkers: Math.round(availableParallelism() / 2),
  25     enableTasksQueue: true,
  26     tasksQueueOptions: {
  27       concurrency: 8
  28     },
  29     errorHandler: (e: Error) => {
  30       fastify.log.error('Thread worker error', e)
  31     }
  32   })
  33
  34   fastify.all('/api/echo', async request => {
  35     return (await fastify.execute({ body: request.body }, 'echo')).body
  36   })
  37
  38   fastify.get<{
  39     Params: { number: number }
  40   }>('/api/factorial/:number', async request => {
  41     const { number } = request.params
  42     return (await fastify.execute({ body: { number } }, 'factorial')).body
  43   })
  44
  45   await fastify.listen({ port })
  46   return {
  47     status: true,
  48     port: (fastify.server.address() as AddressInfo).port
  49   }
  50 }
  51
  52 class FastifyWorker extends ClusterWorker<
  53 ClusterWorkerData,
  54 ClusterWorkerResponse
  55 > {
  56   public constructor () {
  57     super(startFastify)
  58   }
  59 }
  60
  61 export const fastifyWorker = new FastifyWorker()