| 1 | // IMPORT LIBRARIES |
| 2 | const { job, start } = require('microjob') |
| 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 | async function run () { |
| 15 | await start({ maxWorkers: size }) |
| 16 | const promises = [] |
| 17 | for (let i = 0; i < iterations; i++) { |
| 18 | promises.push( |
| 19 | job( |
| 20 | data => { |
| 21 | functionToBench(data) |
| 22 | }, |
| 23 | { data: data, ctx: { functionToBench } } |
| 24 | ) |
| 25 | ) |
| 26 | } |
| 27 | await Promise.all(promises) |
| 28 | process.exit() |
| 29 | } |
| 30 | |
| 31 | run() |