-
Notifications
You must be signed in to change notification settings - Fork 3
/
bench-wrk.sh
executable file
·51 lines (41 loc) · 1.23 KB
/
bench-wrk.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#!/bin/bash
usage() {
echo "$0 <BINARY-PATH> <HOST-IPV4> <START-PORT> <SERVER-THREADS> <WRK-CONNS> <WRK-DURATION> <WRK-THREADS> <WARMUP-ROUNDS> <NORMAL-ROUNDS>"
exit 1
}
if [[ $# -ne 9 ]]; then
usage
fi
readonly BINARY_PATH="$1"
readonly HOST_IPV4="$2"
readonly START_PORT="$3"
readonly SERVER_THREADS="$4"
readonly WRK_CONNS="$5"
readonly WRK_DURATION="$6"
readonly WRK_THREADS="$7"
readonly WARMUP_ROUNDS="$8"
readonly NORMAL_ROUNDS="$9"
total=0
for ((i = 1; i <= $((WARMUP_ROUNDS + NORMAL_ROUNDS)); i++)); do
# Workaround for io_uring bug
PORT=$((START_PORT + i))
"$BINARY_PATH" "$HOST_IPV4" "$PORT" "$SERVER_THREADS" &>/dev/null &
pid=$!
echo "Created server with pid $pid"
sleep 1
reqs=$(wrk -c "$WRK_CONNS" -d "$WRK_DURATION" -t "$WRK_THREADS" "http://$HOST_IPV4:$PORT" | grep 'Requests/sec' | grep -Eo '[+-]?[0-9]+([.][0-9]+)?')
if [[ $i -le $WARMUP_ROUNDS ]]; then
echo "$i/$WARMUP_ROUNDS warm up"
else
n=$((i - WARMUP_ROUNDS))
total=$(echo "$total" "$reqs" | awk '{ printf "%f", $1 + $2 }')
avg=$(echo "$total" $n | awk '{ printf "%.02f", $1 / $2 }')
echo "$n/$NORMAL_ROUNDS cur=$reqs avg=$avg"
fi
echo "Killing $pid"
if ! kill $pid; then
echo "Server failed"
exit 1
fi
wait
done