import type { AddressInfo } from 'node:net'
+
+import Fastify, { type FastifyInstance } from 'fastify'
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
- }
-}
+import { fastifyPoolifier } from './fastify-poolifier.js'
+import type { ClusterWorkerData, ClusterWorkerResponse } from './types.js'
class FastifyWorker extends ClusterWorker<
ClusterWorkerData,
ClusterWorkerResponse
> {
+ private static fastify: FastifyInstance
+
+ private static readonly startFastify = async (
+ workerData?: ClusterWorkerData
+ ): Promise<ClusterWorkerResponse> => {
+ const { port, ...fastifyPoolifierOptions } = workerData!
+
+ FastifyWorker.fastify = Fastify({
+ logger: true
+ })
+
+ await FastifyWorker.fastify.register(
+ fastifyPoolifier,
+ fastifyPoolifierOptions
+ )
+
+ FastifyWorker.fastify.all('/api/echo', async request => {
+ return (
+ await FastifyWorker.fastify.execute({ data: request.body }, 'echo')
+ ).data
+ })
+
+ FastifyWorker.fastify.get<{
+ Params: { number: number }
+ }>('/api/factorial/:number', async request => {
+ const { number } = request.params
+ return (
+ await FastifyWorker.fastify.execute({ data: { number } }, 'factorial')
+ ).data
+ })
+
+ await FastifyWorker.fastify.listen({ port })
+ return {
+ status: true,
+ port: (FastifyWorker.fastify.server.address() as AddressInfo).port
+ }
+ }
+
public constructor () {
- super(startFastify)
+ super(FastifyWorker.startFastify, {
+ killHandler: async () => {
+ await FastifyWorker.fastify.pool.destroy()
+ await FastifyWorker.fastify.close()
+ }
+ })
}
}