benchmarks/versus-external-pools/dynamic-workerpool.mjs

   1 import workerpool from 'workerpool'
   2 import { BenchmarkDefaults, executeAsyncFn } from './utils.mjs'
   3
   4 const size = parseInt(process.env.POOL_SIZE) || BenchmarkDefaults.poolSize
   5 const numIterations =
   6   parseInt(process.env.NUM_ITERATIONS) || BenchmarkDefaults.numIterations
   7 const dataArray = [
   8   'MYBENCH',
   9   process.env.TASK_TYPE || BenchmarkDefaults.taskType,
  10   parseInt(process.env.TASK_SIZE) || BenchmarkDefaults.taskSize
  11 ]
  12
  13 const workerPool = workerpool.pool(
  14   './workers/workerpool/function-to-bench-worker.mjs',
  15   {
  16     minWorkers: Math.floor(size / 2),
  17     maxWorkers: size,
  18     workerType: 'thread'
  19   }
  20 )
  21
  22 async function run () {
  23   const promises = new Set()
  24   for (let i = 0; i < numIterations; i++) {
  25     promises.add(workerPool.exec('functionToBench', dataArray))
  26   }
  27   await Promise.all(promises)
  28   // eslint-disable-next-line n/no-process-exit
  29   process.exit()
  30 }
  31
  32 await executeAsyncFn(run)