k6 v0.42.0 updates - helper scripts (3/5)

k6bench.gnuplot

@@ -0,0 +1,28 @@
+set datafile separator ','
+set terminal pngcairo background rgb 'white' linewidth 4 size 1200,600 enhanced font 'Arial,16'
+set bmargin at screen 130.0/600
+set rmargin at screen 1080.0/1200
+set output sprintf('%s-%s.png', ec2instance, script)
+
+set title sprintf('k6 %s / EC2 %s / %s', k6version, ec2instance, script) font 'Arial Bold,20'
+set key at graph 0.6, 0.3 autotitle columnhead
+
+set border 31 lw 0.5
+set style data lines
+set style line 100 lt 1 lc rgb "grey" lw 0.5 # linestyle for the grid
+set grid ls 100
+
+set xtics 30
+set xlabel 'Time (M:S)'
+set xdata time
+set format x '%M:%S'
+set xtics rotate
+set y2tics           # enable second axis
+set ytics nomirror   # dont show the tics on that side
+set y2range [0:]     # start from 0
+set y2label "CPU (%)"
+
+plot sprintf('%s-%s.csv', ec2instance, script) using ($1):2 axis x1y2 lc rgb '#00d8bfd8', \
+     '' using ($1):($3 / 1000) title 'RAM (MB)', \
+     '' using ($1):4, \
+     '' using ($1):5

k6bench.sh

@@ -0,0 +1,50 @@
+#!/bin/bash
+# External dependencies:
+# - https://www.selenic.com/smem/
+# - https://stedolan.github.io/jq/
+set -eEuo pipefail
+
+trap 'cleanup; exit 0' EXIT
+trap 'trap - INT; cleanup; kill -INT $$' INT
+trap 'trap - TERM; cleanup; kill -TERM $$' TERM
+
+# Create temp files to store the output of each collection command.
+cpuf=$(mktemp)
+memf=$(mktemp)
+vusf=$(mktemp)
+rpsf=$(mktemp)
+
+cleanup() {
+  rm -f "$cpuf" "$memf" "$vusf" "$rpsf"
+}
+
+# 5s is the default interval between samples.
+# Note that this might be greater if either smem or the k6 API takes more time
+# than this to return a response.
+sint="${K6_BENCH_SAMPLE_INTERVAL:-5}"
+
+k6 run "$@" >k6out.txt 2>&1 &
+k6pid="$!"
+
+# Run the collection processes in parallel to avoid blocking.
+# For details see https://stackoverflow.com/a/68316571
+
+echo '"Time (s)","CPU (%)","RAM (kB)","VUs","RPS"'
+while true; do
+  etimes=$(ps -p "$k6pid" --no-headers -o etimes | awk '{ print $1 }')
+  pids=()
+  { exec >"$cpuf"; top -b -n 2 -d "$sint" -p "$k6pid" | {
+      grep "$k6pid" || echo; } | tail -1 | awk '{print (NF>0 ? $9 : "0")}'; } &
+  pids+=($!)
+  { exec >"$memf"; smem -H -U "$USER" -c 'pid pss' -P 'k6 run' | {
+      grep "$k6pid" || echo 0; } | awk '{ print $NF }'; } &
+  pids+=($!)
+  { exec >"$vusf"; { curl -fsSL http://localhost:6565/v1/metrics/vus 2>/dev/null || echo '{}'
+    } | jq '.data.attributes.sample.value // 0'; } &
+  pids+=($!)
+  { exec >"$rpsf"; { curl -fsSL http://localhost:6565/v1/metrics/http_reqs 2>/dev/null || echo '{}'
+    } | jq '.data.attributes.sample.rate // 0'; } &
+  pids+=($!)
+  wait "${pids[@]}"
+  echo "${etimes},$(cat "$cpuf"),$(cat "$memf"),$(cat "$vusf"),$(cat "$rpsf")"
+done