Commit | Line | Data |
---|---|---|
3b311539 JB |
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() |