perf: add benchmark defaults to external pools benchmark

[poolifier.git] / benchmarks / versus-external-pools / dynamic-worker-nodes.js

'use strict'
const WorkerNodes = require('worker-nodes')
const { BenchmarkDefaults } = require('./utils.mjs')

const size = parseInt(process.env.POOL_SIZE) || BenchmarkDefaults.poolSize
const numIterations =
  parseInt(process.env.NUM_ITERATIONS) || BenchmarkDefaults.numIterations
const data = {
  test: 'MYBENCH',
  taskType: process.env.TASK_TYPE || BenchmarkDefaults.taskType,
  taskSize: parseInt(process.env.TASK_SIZE) || BenchmarkDefaults.taskSize
}

const workerNodes = new WorkerNodes(
  require.resolve('./workers/worker-nodes/function-to-bench-worker'),
  {
    minWorkers: Math.floor(size / 2),
    maxWorkers: size,
    taskTimeout: 60000 // this is the same as poolifier default
  }
)

async function run () {
  const promises = new Set()
  for (let i = 0; i < numIterations; i++) {
    promises.add(workerNodes.call.functionToBench(data))
  }
  await Promise.all(promises)
  // eslint-disable-next-line n/no-process-exit
  process.exit()
}

(async () => {
  try {
    await run()
  } catch (e) {
    console.error(e)
    // eslint-disable-next-line n/no-process-exit
    process.exit(1)
  }
})()