From 20a2fd03d02ae672257bfa22cbddcb2a3b7e4696 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Torma=20Krist=C3=B3f?= <tormakristof@tormakristof.eu>
Date: Fri, 26 Apr 2019 15:12:05 +0200
Subject: [PATCH] Update benchmark.sh

---
 benchmark/benchmark.sh | 107 ++++++++++++++++++++---------------------
 1 file changed, 53 insertions(+), 54 deletions(-)

diff --git a/benchmark/benchmark.sh b/benchmark/benchmark.sh
index 066aff2..ccc27d4 100644
--- a/benchmark/benchmark.sh
+++ b/benchmark/benchmark.sh
@@ -6,7 +6,6 @@ times=(1m)
 kuberhost="node1:30765"
 maxthreads=40
 
-wave_function=isprime
 wave_dir_up=true
 wave_connection=40
 wave_max_conn=1600
@@ -33,61 +32,61 @@ then
         cp "$HOME"/go/bin/hey /usr/local/bin
 fi
 
-if [[ $* = *"--wave"* ]]
-then
-while true; do
-        now=$(date '+%Y-%m-%d-%H-%M')
-        echo -e "Running"
-        hey -c $wave_connection -z $wave_time -m POST -o csv -host "$wave_function.kubeless" -D "$wave_function".body  -T "application/json" http://$kuberhost/"$wave_function" > ./"$wave_function"."$now".wave.txt
-        echo -e "Sleeping"
-        sleep $wave_time
-        if [[ $wave_dir_up ]]
-        then
-                if [[ $wave_connection -lt $wave_max_conn ]]
-                then
-                        wave_connection=$((wave_connection * 2))
-                else
-                        wave_dir_up=false
-                fi
-        else
-                if [[ $wave_connection -gt $wave_min_conn ]]
-                then
-                        wave_connection=$((wave_connection / 2))
-                else
-                        wave_dir_up=true
-                fi
-        fi
-done
-else
 echo -e "Benchmarking functions\n"
 for function in "${functions[@]}"
 do
-    echo -e "Benchmarking $function\n"
-    echo -e "Output of $function is:\n"
-    perl -pi -e 'chomp if eof' "$function".body
-    curl --data-binary @"$function".body --header "Host: $function.kubeless" --header "Content-Type:application/json" http://$kuberhost/"$function"
-    echo -e "\n"
-    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
-                datetime=$(date '+%Y-%m-%d-%H-%M-%S')
-                echo -e "Time: $time\n"
-                echo -e "wrk\n"
-                wrk -t$threads -c"$connection" -d"$time" -s"$function".wrk -H"Host: $function.kubeless" -H"Content-Type:application/json" --latency  http://$kuberhost/"$function" > ./"$function"."$connection"."$time"."$datetime".wrk.txt 2>&1
-                echo -e "hey-summary\n"
-                hey -c "$connection" -z "$time" -m POST -host "$function.kubeless" -D "$function".body  -T "application/json" http://$kuberhost/"$function" > ./"$function"."$connection"."$time"."$datetime".hey.txt
-                echo -e "hey-csv\n"
-                hey -c "$connection" -z "$time" -m POST -o csv -host "$function.kubeless" -D "$function".body  -T "application/json" http://$kuberhost/"$function" > ./"$function"."$connection"."$time"."$datetime".csv
-                echo -e "$datetime"
+    if [[ $* = *"--wave"* ]]
+    then
+        while true; do
+                now=$(date '+%Y-%m-%d-%H-%M')
+                echo -e "Running"
+                hey -c $wave_connection -z $wave_time -m POST -o csv -host "$function.kubeless" -D "$function".body  -T "application/json" http://$kuberhost/"$function" > ./"$function"."$now".wave.txt
+                echo -e "Sleeping"
+                sleep $wave_time
+                if [[ $wave_dir_up ]]
+                then
+                        if [[ $wave_connection -lt $wave_max_conn ]]
+                        then
+                            wave_connection=$((wave_connection * 2))
+                        else
+                            wave_dir_up=false
+                        fi
+                else
+                    if [[ $wave_connection -gt $wave_min_conn ]]
+                    then
+                        wave_connection=$((wave_connection / 2))
+                    else
+                        wave_dir_up=true
+                    fi
+                fi
         done
-    done
+    else
+        echo -e "Benchmarking $function\n"
+        echo -e "Output of $function is:\n"
+        perl -pi -e 'chomp if eof' "$function".body
+        curl --data-binary @"$function".body --header "Host: $function.kubeless" --header "Content-Type:application/json" http://$kuberhost/"$function"
+       echo -e "\n"
+        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
+                    datetime=$(date '+%Y-%m-%d-%H-%M-%S')
+                    echo -e "Time: $time\n"
+                    echo -e "wrk\n"
+                    wrk -t$threads -c"$connection" -d"$time" -s"$function".wrk -H"Host: $function.kubeless" -H"Content-Type:application/json" --latency  http://$kuberhost/"$function" > ./"$function"."$connection"."$time"."$datetime".wrk.txt 2>&1
+                    echo -e "hey-summary\n"
+                    hey -c "$connection" -z "$time" -m POST -host "$function.kubeless" -D "$function".body  -T "application/json" http://$kuberhost/"$function" > ./"$function"."$connection"."$time"."$datetime".hey.txt
+                    echo -e "hey-csv\n"
+                    hey -c "$connection" -z "$time" -m POST -o csv -host "$function.kubeless" -D "$function".body  -T "application/json" http://$kuberhost/"$function" > ./"$function"."$connection"."$time"."$datetime".csv
+                    echo -e "$datetime"
+            done
+        done
+    fi
 done
-fi