Add worker-threads-pool to external pools benchmarks (#281)
[poolifier.git] / benchmarks / versus-external-pools / fixed-worker-threads-pool.js
diff --git a/benchmarks/versus-external-pools/fixed-worker-threads-pool.js b/benchmarks/versus-external-pools/fixed-worker-threads-pool.js
new file mode 100644 (file)
index 0000000..ed7e2ae
--- /dev/null
@@ -0,0 +1,29 @@
+// IMPORT LIBRARIES
+const Pool = require('worker-threads-pool')
+// FINISH IMPORT LIBRARIES
+const size = Number(process.env.POOL_SIZE)
+const iterations = Number(process.env.NUM_ITERATIONS)
+const data = {
+  test: 'MYBENCH',
+  taskType: process.env['TASK_TYPE']
+}
+
+const pool = new Pool({ max: size })
+
+async function run () {
+  const promises = []
+  for (let i = 0; i < iterations; i++) {
+    promises.push(
+      pool.acquire(
+        './workers/worker-threads-pool/function-to-bench-worker.js',
+        {
+          workerData: data
+        }
+      )
+    )
+  }
+  await Promise.all(promises)
+  process.exit()
+}
+
+run()