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

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