benchmarks/versus-external-pools/dynamic-suchmokuo-node-worker-threads-pool.js

   1 // IMPORT LIBRARIES
   2 const { DynamicPool } = require('node-worker-threads-pool')
   3 // FINISH IMPORT LIBRARIES
   4 // IMPORT FUNCTION TO BENCH
   5 const functionToBench = require('./functions/function-to-bench')
   6 // FINISH IMPORT FUNCTION TO BENCH
   7 const size = Number(process.env.POOL_SIZE)
   8 const iterations = Number(process.env.NUM_ITERATIONS)
   9 const data = {
  10   test: 'MYBENCH',
  11   taskType: process.env.TASK_TYPE,
  12   taskSize: process.env.TASK_SIZE
  13 }
  14
  15 const pool = new DynamicPool(size)
  16
  17 async function run () {
  18   const promises = []
  19   for (let i = 0; i < iterations; i++) {
  20     promises.push(
  21       pool.exec({
  22         task: functionToBench,
  23         param: data
  24       })
  25     )
  26   }
  27   await Promise.all(promises)
  28   // eslint-disable-next-line no-process-exit
  29   process.exit()
  30 }
  31
  32 run()