Improve benchmarks: add IO intensive task workload, add task size option, integrate...

[poolifier.git] / benchmarks / versus-external-pools / dynamic-suchmokuo-node-worker-threads-pool.js

// IMPORT LIBRARIES
const { DynamicPool } = require('node-worker-threads-pool')
// FINISH IMPORT LIBRARIES
// IMPORT FUNCTION TO BENCH
const functionToBench = require('./functions/function-to-bench')
// FINISH IMPORT FUNCTION TO BENCH
const size = Number(process.env.POOL_SIZE)
const iterations = Number(process.env.NUM_ITERATIONS)
const data = {
  test: 'MYBENCH',
  taskType: process.env.TASK_TYPE,
  taskSize: process.env.TASK_SIZE
}

const pool = new DynamicPool(size)

async function run () {
  const promises = []
  for (let i = 0; i < iterations; i++) {
    promises.push(
      pool.exec({
        task: functionToBench,
        param: data
      })
    )
  }
  await Promise.all(promises)
  // eslint-disable-next-line no-process-exit
  process.exit()
}

run()