benchmarks/versus-external-pools/fixed-piscina.mjs

   1 import Piscina from 'piscina'
   2 import { executeAsyncFn } from './utils.mjs'
   3
   4 const size = parseInt(process.env.POOL_SIZE)
   5 const iterations = parseInt(process.env.NUM_ITERATIONS)
   6 const data = {
   7   test: 'MYBENCH',
   8   taskType: process.env.TASK_TYPE,
   9   taskSize: parseInt(process.env.TASK_SIZE)
  10 }
  11
  12 const piscina = new Piscina({
  13   filename: './workers/piscina/function-to-bench-worker.mjs',
  14   minThreads: size,
  15   maxThreads: size,
  16   idleTimeout: 60000 // this is the same as poolifier default
  17 })
  18
  19 async function run () {
  20   const promises = new Set()
  21   for (let i = 0; i < iterations; i++) {
  22     promises.add(piscina.run(data))
  23   }
  24   await Promise.all(promises)
  25   // eslint-disable-next-line n/no-process-exit
  26   process.exit()
  27 }
  28
  29 await executeAsyncFn(run)