fix: properly size the pools in benchmarking code

[poolifier.git] / benchmarks / versus-external-pools / dynamic-piscina.mjs

// IMPORT LIBRARIES
import Piscina from 'piscina'
// FINISH IMPORT LIBRARIES
const size = parseInt(process.env.POOL_SIZE)
const iterations = parseInt(process.env.NUM_ITERATIONS)
const data = {
  test: 'MYBENCH',
  taskType: process.env.TASK_TYPE,
  taskSize: parseInt(process.env.TASK_SIZE)
}

const piscina = new Piscina({
  filename: './workers/piscina/function-to-bench-worker.mjs',
  minThreads: Math.floor(size / 2),
  maxThreads: size,
  idleTimeout: 60000 // this is the same as poolifier default
})

async function run () {
  const promises = []
  for (let i = 0; i < iterations; i++) {
    promises.push(piscina.run(data))
  }
  await Promise.all(promises)
  // eslint-disable-next-line n/no-process-exit
  process.exit()
}

await run()