Merge dependabot/npm_and_yarn/examples/typescript/http-server-pool/fastify-hybrid...

[poolifier.git] / benchmarks / versus-external-pools / bench.sh

diff --git a/benchmarks/versus-external-pools/bench.sh b/benchmarks/versus-external-pools/bench.sh
index 05c156c92b27d774f56e742b19e1df7fe81e916e..7d7e9651469252304dec4145adcb42c858f88bd6 100755 (executable)
--- a/benchmarks/versus-external-pools/bench.sh
+++ b/benchmarks/versus-external-pools/bench.sh
@@ -1,37 +1,59 @@
-export NODE_ENV=production
-
-# Execute bench
-# export POOL_SIZE=8
-# export NUM_ITERATIONS=10000
-# hyperfine --export-markdown BENCH-10000.MD --min-runs 10 \
-#   'node dynamic-poolifier.js' \
-#   'node dynamic-suchmokuo-node-worker-threads-pool.js' \
-#   'node fixed-poolifier.js' \
-#   'node static-suchmokuo-node-worker-threads-pool.js' \
-#   'node piscina.js'
+#!/usr/bin/env bash
 
-# echo "Sleeping...."
-# sleep 60
+usage() {
+    echo "Usage: $0 [-s <pool size> -i <number of iterations> -t <CPU_INTENSIVE|IO_INTENSIVE>] [-n <number of tasks>]" 1>&2
+    exit 1
+}
 
-export POOL_SIZE=10
-export NUM_ITERATIONS=100000
-hyperfine --export-markdown BENCH-100000.MD --min-runs 10 \
-  'node static-suchmokuo-node-worker-threads-pool.js' \
-  'node dynamic-poolifier.js' \
-  'node piscina.js'
+while getopts "s:i:t:n:h" option
+do
+  case "${option}" in
+    s)
+      poolSize=${OPTARG}
+      ;;
+    i)
+      iterations=${OPTARG}
+      ;;
+    t)
+      taskType=${OPTARG}
+      ;;
+    n)
+      taskSize=${OPTARG}
+      ;;
+    :)
+      echo "Error: Missing option argument for '-${OPTARG}'" >&2;
+      usage
+      ;;
+    \?)
+      # Invalid option
+      echo "Error: Unknown or invalid option for '-${OPTARG}'" >&2;
+      usage
+      ;;
+    *|h)
+      usage
+      ;;
+  esac
+done
 
-# export POOL_SIZE=8
-# export NUM_ITERATIONS=50000
-# hyperfine --export-markdown BENCH-50000.MD --min-runs 10 \
-#   'node dynamic-poolifier.js' \
-#   'node dynamic-suchmokuo-node-worker-threads-pool.js' \
-#   'node fixed-poolifier.js' \
-#   'node static-suchmokuo-node-worker-threads-pool.js' \
-#   'node piscina.js'
-
-# export NUM_ITERATIONS=100000
-#   hyperfine --export-markdown BENCH-50000.MD --min-runs 20 \
-#     'node dynamic-poolifier.js' \
-#     'node static-suchmokuo-node-worker-threads-pool.js' \
-#     'node piscina.js'
+taskType=${taskType:-'CPU_INTENSIVE'}
+taskSize=${taskSize:-5000}
+poolSize=${poolSize:-$(nproc --all)}
+iterations=${iterations:-100000}
 
+echo 'Running benchmarks with pool size:' ${poolSize}', number of iterations:' ${iterations}', task type:' ${taskType} 'and task size:' ${taskSize}
+export NODE_ENV=production
+export TASK_TYPE=${taskType}
+export TASK_SIZE=${taskSize}
+export POOL_SIZE=${poolSize}
+export NUM_ITERATIONS=${iterations}
+case "$OSTYPE" in
+  darwin*)
+    caffeinate ./hyperfine_benchmarks.sh
+    ;;
+  linux*)
+    systemd-inhibit --what=idle ./hyperfine_benchmarks.sh
+    ;;
+  *)
+    echo "Unsupported $OSTYPE"
+    ;;
+esac