fix(benchmark): add missing export

[poolifier.git] / benchmarks / internal / bench.mjs

import { exit } from 'node:process'
import { parseArgs } from 'node:util'

import { run } from 'mitata'

import {
  availableParallelism,
  PoolTypes,
  WorkerTypes
} from '../../lib/index.mjs'
import { TaskFunctions } from '../benchmarks-types.cjs'
import {
  buildPoolifierBenchmarkMitata,
  runPoolifierBenchmarkBenchmarkJs
} from '../benchmarks-utils.mjs'

const poolSize = availableParallelism()
const taskExecutions = 1
const workerData = {
  function: TaskFunctions.factorial,
  taskSize: 50000
}

let fixedThreadPool
let dynamicThreadPool
let fixedClusterPool
let dynamicClusterPool
switch (
  parseArgs({
    args: process.argv,
    options: {
      type: {
        type: 'string',
        short: 't'
      }
    },
    strict: true,
    allowPositionals: true
  }).values.type
) {
  case 'mitata':
    fixedThreadPool = buildPoolifierBenchmarkMitata(
      'FixedThreadPool',
      WorkerTypes.thread,
      PoolTypes.fixed,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    dynamicThreadPool = buildPoolifierBenchmarkMitata(
      'DynamicThreadPool',
      WorkerTypes.thread,
      PoolTypes.dynamic,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    fixedClusterPool = buildPoolifierBenchmarkMitata(
      'FixedClusterPool',
      WorkerTypes.cluster,
      PoolTypes.fixed,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    dynamicClusterPool = buildPoolifierBenchmarkMitata(
      'DynamicClusterPool',
      WorkerTypes.cluster,
      PoolTypes.dynamic,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    await run()
    await fixedThreadPool.destroy()
    await dynamicThreadPool.destroy()
    await fixedClusterPool.destroy()
    await dynamicClusterPool.destroy()
    break
  case 'benchmark.js':
  default:
    await runPoolifierBenchmarkBenchmarkJs(
      'FixedThreadPool',
      WorkerTypes.thread,
      PoolTypes.fixed,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    await runPoolifierBenchmarkBenchmarkJs(
      'DynamicThreadPool',
      WorkerTypes.thread,
      PoolTypes.dynamic,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    await runPoolifierBenchmarkBenchmarkJs(
      'FixedClusterPool',
      WorkerTypes.cluster,
      PoolTypes.fixed,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    await runPoolifierBenchmarkBenchmarkJs(
      'DynamicClusterPool',
      WorkerTypes.cluster,
      PoolTypes.dynamic,
      poolSize,
      {
        taskExecutions,
        workerData
      }
    )
    break
}

exit()