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 }
  13
  14 const pool = new DynamicPool(size)
  15
  16 async function run () {
  17   const promises = []
  18   for (let i = 0; i < iterations; i++) {
  19     promises.push(
  20       pool.exec({
  21         task: functionToBench,
  22         param: data
  23       })
  24     )
  25   }
  26   await Promise.all(promises)
  27   process.exit()
  28 }
  29
  30 run()