perf: add benchmark defaults to external pools benchmark
[poolifier.git] / benchmarks / versus-external-pools / static-node-worker-threads-pool.mjs
index db14b180e6633461542a96a1df9f6b270e32bf80..0868d26ce1bc0a3f80f65f1d0fc3ddd7b4259d64 100644 (file)
@@ -1,13 +1,14 @@
 import { StaticPool } from 'node-worker-threads-pool'
-import { executeAsyncFn } from './utils.mjs'
+import { BenchmarkDefaults, executeAsyncFn } from './utils.mjs'
 import functionToBench from './functions/function-to-bench.js'
 
-const size = parseInt(process.env.POOL_SIZE)
-const iterations = parseInt(process.env.NUM_ITERATIONS)
+const size = parseInt(process.env.POOL_SIZE) || BenchmarkDefaults.poolSize
+const numIterations =
+  parseInt(process.env.NUM_ITERATIONS) || BenchmarkDefaults.numIterations
 const data = {
   test: 'MYBENCH',
-  taskType: process.env.TASK_TYPE,
-  taskSize: parseInt(process.env.TASK_SIZE)
+  taskType: process.env.TASK_TYPE || BenchmarkDefaults.taskType,
+  taskSize: parseInt(process.env.TASK_SIZE) || BenchmarkDefaults.taskSize
 }
 
 const staticPool = new StaticPool({
@@ -17,7 +18,7 @@ const staticPool = new StaticPool({
 
 async function run () {
   const promises = new Set()
-  for (let i = 0; i < iterations; i++) {
+  for (let i = 0; i < numIterations; i++) {
     promises.add(staticPool.exec(data))
   }
   await Promise.all(promises)