kubeless/benchmark/benchmark.sh

75 lines
2.2 KiB
Bash

#!/bin/bash
functions=(hello)
connections=(2 5 10 20 50 100 200 400 500 1000)
times=(1m)
data=(isprime)
kuberhost="node1:32764"
maxthreads=160
wrk_options="-t$threads -d$time -c$connection -H\"Host: $function.kubeless\" -H\"Content-Type:application/json\" --latency http://$kuberhost/$function"
wrk_output=$function.$connection.$time.txt
hey_options="-c $connection -z $time -o csv -m POST -host \"$function.kubeless\" -T \"application/json\" http://$kuberhost/$function"
hey_output=$function.$connection.$time.csv
array_contains () {
local array="$1[@]"
local seeking=$2
local in=1
for element in "${!array}"; do
if [[ $element == $seeking ]]; then
in=0
break
fi
done
return $in
}
WRK_INSTALLED=$(which wrk)
if [ "$WRK_INSTALLED" = "" ]
then
apt update
apt install build-essential libssl-dev git -y
git clone https://github.com/wg/wrk.git wrk
cd wrk
make
cp wrk /usr/local/bin
fi
HEY_INSTALLED=$(which hey)
if [ "$HEY_INSTALLED" = "" ]
then
apt update
apt install -y golang
go get -u github.com/rakyll/hey
cp $HOME/go/bin/hey /usr/local/bin
fi
echo -e "Benchmarking GET functions\n"
for function in "${functions[@]}"
do
echo -e "Benchmarking $function\n"
echo -e "Output of $function is:\n"
array_contains data function && perl -pi -e 'chomp if eof' $function.body
curl_additional_options=$(array_contains data function && "--data-binary \"@$function.body\"")
curl $curl_additional_options --header "Host: $function.kubeless" --header "Content-Type:application/json" http://$kuberhost/$function
echo -e "\n"
wrk_additional_options=$(array_contains data function && "-s$function.wrk")
hey_additional_options=$(array_contains data function && "-D $function.body")
for connection in "${connections[@]}"
do
if [[ $connection -lt $(($maxthreads + 1)) ]]
then
threads=$(($connection-1))
else
threads=$maxthreads
fi
echo -e "Threads: $threads Connections $connection\n"
for time in "${times[@]}"
do
echo -e "Time: $time\n"
wrk $wrk_options $wrk_additional_options > $wrk_output 2>&1
hey $hey_options $hey_additional_options > $hey_output
done
done
done