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