collect cpu/mem statistics

bench.sh

@@ -2,6 +2,8 @@
 
 RESULT_DIR=results/$(date "+%y%d%mT%H%M%S")
 
+docker pull infoblox/ghz:0.0.1
+
 # Rust Tonic benchmark
 ./run_single_bench.sh rust_tonic_test "${RESULT_DIR}"
 

collect_stats.sh

@@ -0,0 +1,16 @@
+#!/bin/sh
+
+NAME=$1
+REPORT_DIR=${2:-"results"}
+
+rm -f "${REPORT_DIR}"/"${NAME}".stats
+
+sleep 1
+
+while true; do
+    (docker stats \
+        --no-stream \
+        --format "table {{.CPUPerc}}\t{{.MemUsage}}" \
+        "${NAME}" | grep -v CPU) >> "${REPORT_DIR}"/"${NAME}".stats 2> /dev/null || break;
+    sleep 5 || break;
+done

run_single_bench.sh

@@ -7,21 +7,30 @@ GRPC_BENCHMARK_DURATION=${GRPC_BENCHMARK_DURATION:-"30s"}
 GRPC_SERVER_CPUS=${GRPC_SERVER_CPUS:-"1"}
 
 echo "==> Running benchmark for ${NAME}..."
+
 mkdir -p "${REPORT_DIR}"
+
 docker run --name "${NAME}" --rm --cpus "${GRPC_SERVER_CPUS}" --network=host -d -t ${NAME}
+
 sleep 5
+
+./collect_stats.sh "${NAME}" "${REPORT_DIR}" &
+
 docker run \
     --name ghz \
     --rm \
     --network=host \
     -v "${PWD}"/proto:/proto:ro \
     --entrypoint=ghz \
     infoblox/ghz:0.0.1 \
-    --proto=/proto/helloworld/helloworld.proto \
-    --call=helloworld.Greeter.SayHello \
-    --insecure \
-    --connections=5 \
-    --duration "${GRPC_BENCHMARK_DURATION}" \
-    -d "{\"name\":\"it's not as performant as we expected\"}" \
-    127.0.0.1:50051 > "${REPORT_DIR}"/"${NAME}".report
+        --proto=/proto/helloworld/helloworld.proto \
+        --call=helloworld.Greeter.SayHello \
+        --insecure \
+        --connections=5 \
+        --duration "${GRPC_BENCHMARK_DURATION}" \
+        -d "{\"name\":\"it's not as performant as we expected\"}" \
+        127.0.0.1:50051 > "${REPORT_DIR}"/"${NAME}".report
+
+kill -INT %1 2> /dev/null
+
 docker container stop "${NAME}"