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