examples/typescript/http-server-pool/fastify-worker_threads/src/fastify-poolifier.ts

   1 import type { TransferListItem } from 'node:worker_threads'
   2 import { DynamicThreadPool, availableParallelism } from 'poolifier'
   3 import { type FastifyPluginCallback } from 'fastify'
   4 import fp from 'fastify-plugin'
   5 import {
   6   type FastifyPoolifierOptions,
   7   type WorkerData,
   8   type WorkerResponse
   9 } from './types.js'
  10
  11 const fastifyPoolifierPlugin: FastifyPluginCallback<FastifyPoolifierOptions> = (
  12   fastify,
  13   options,
  14   done
  15 ) => {
  16   options = {
  17     ...{
  18       minWorkers: 1,
  19       maxWorkers: availableParallelism()
  20     },
  21     ...options
  22   }
  23   const { workerFile, minWorkers, maxWorkers, ...poolOptions } = options
  24   const pool = new DynamicThreadPool<WorkerData, WorkerResponse>(
  25     minWorkers as number,
  26     maxWorkers as number,
  27     workerFile,
  28     poolOptions
  29   )
  30   if (!fastify.hasDecorator('pool')) {
  31     fastify.decorate('pool', pool)
  32   }
  33   if (!fastify.hasDecorator('execute')) {
  34     fastify.decorate(
  35       'execute',
  36       async (
  37         data?: WorkerData,
  38         name?: string,
  39         transferList?: TransferListItem[]
  40       ): Promise<WorkerResponse> => await pool.execute(data, name, transferList)
  41     )
  42   }
  43   done()
  44 }
  45
  46 export const fastifyPoolifier = fp(fastifyPoolifierPlugin, {
  47   fastify: '4.x',
  48   name: 'fastify-poolifier'
  49 })