refactor: move pools options handling in main ws-hybrid example

[poolifier.git] / examples / typescript / http-server-pool / fastify-hybrid / src / fastify-worker.ts

import type { AddressInfo } from 'node:net'
import { ClusterWorker } from 'poolifier'
import Fastify from 'fastify'
import type { ClusterWorkerData, ClusterWorkerResponse } from './types.js'
import { fastifyPoolifier } from './fastify-poolifier.js'

const startFastify = async (
  workerData?: ClusterWorkerData
): Promise<ClusterWorkerResponse> => {
  const { port } = workerData as ClusterWorkerData
  const fastify = Fastify({
    logger: true
  })

  await fastify.register(fastifyPoolifier, workerData)

  fastify.all('/api/echo', async request => {
    return (await fastify.execute({ body: request.body }, 'echo')).body
  })

  fastify.get<{
    Params: { number: number }
  }>('/api/factorial/:number', async request => {
    const { number } = request.params
    return (await fastify.execute({ body: { number } }, 'factorial')).body
  })

  await fastify.listen({ port })
  return {
    status: true,
    port: (fastify.server.address() as AddressInfo).port
  }
}

class FastifyWorker extends ClusterWorker<
ClusterWorkerData,
ClusterWorkerResponse
> {
  public constructor () {
    super(startFastify)
  }
}

export const fastifyWorker = new FastifyWorker()