Merge branch 'master' of github.com:poolifier/poolifier

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

import { DynamicThreadPool, availableParallelism } from 'poolifier'
import { type FastifyPluginCallback } from 'fastify'
import fp from 'fastify-plugin'
import {
  type FastifyPoolifierOptions,
  type WorkerData,
  type WorkerResponse
} from './types.js'

const fastifyPoolifierPlugin: FastifyPluginCallback<FastifyPoolifierOptions> = (
  fastify,
  options,
  done
) => {
  options = {
    ...{
      minWorkers: 1,
      maxWorkers: availableParallelism()
    },
    ...options
  }
  const pool = new DynamicThreadPool<WorkerData, WorkerResponse>(
    options.minWorkers as number,
    options.maxWorkers as number,
    options.workerFile,
    options
  )
  if (!fastify.hasDecorator('pool')) {
    fastify.decorate('pool', pool)
  }
  if (!fastify.hasDecorator('execute')) {
    fastify.decorate(
      'execute',
      async (data?: WorkerData, name?: string): Promise<WorkerResponse> =>
        await pool.execute(data, name)
    )
  }
  done()
}

export const fastifyPoolifier = fp(fastifyPoolifierPlugin, {
  fastify: '4.x',
  name: 'fastify-poolifier'
})