From a3f99d09838375dba13647d41b27632defada409 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Torma=20Krist=C3=B3f?= <tormakristof@tormakristof.eu>
Date: Tue, 29 Oct 2019 12:21:03 +0100
Subject: [PATCH] loadtest benchmark

---
 benchmark/classic/benchmark.sh | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/benchmark/classic/benchmark.sh b/benchmark/classic/benchmark.sh
index 7189663..ab37346 100644
--- a/benchmark/classic/benchmark.sh
+++ b/benchmark/classic/benchmark.sh
@@ -19,6 +19,8 @@ wave_loop_max=2
 climb_max=300
 climb_connection=30
 kubeless=true
+#Loadtest specific fonguration
+loadtest_rps=1000
 
 HEY_INSTALLED=$(command -v hey)
 if [[ $HEY_INSTALLED = "" ]]
@@ -29,6 +31,14 @@ then
         cp "$HOME"/go/bin/hey /usr/local/bin
 fi
 
+LOADTEST_INSTALLED=$(command -v loadtest)
+if [[ LOADTEST_INSTALLED = "" ]]
+then
+        apt update
+        apt install -y nodejs npm
+        npm i -g loadtest
+fi
+
 echo -e "Benchmarking functions\n"
 for function in "${functions[@]}"
 do
@@ -95,14 +105,24 @@ do
                 do
                     datetime=$(date '+%Y-%m-%d-%H-%M-%S')
                     echo -e "Time: $time\n"
-                    if [[ $* = *"--hey"* ]]
+                    if [[ $* = *"--fire"* ]]
                     then
                         echo -e "hey-summary $datetime\n"
                         if $kubeless;
                         then
-                            hey -c "$connection" -z "$time" -m POST -host "$function.kubeless" -D "$function_firendly".body  -T "application/json" http://$kuberhost/"$function" > ./data/"$function"."$connection"."$time"."$datetime".txt
+                            if [[ $* = *"--loadtest"* ]]
+                            then
+                                loadtest -H Host: $function.kubeless --rps $leadtest_rps -c $connection -t $time -p "$function_firendly".body http://$kuberhost/"$function" > ./data/"$function"."$connection"."$time"."$datetime".txt
+                            else
+                                hey -c "$connection" -z "$time" -m POST -host "$function.kubeless" -D "$function_firendly".body  -T "application/json" http://$kuberhost/"$function" > ./data/"$function"."$connection"."$time"."$datetime".txt
+                            fi
                         else
-                            hey -c "$connection" -z "$time" -m POST -host "$function.default.example.com" http://$kuberhost/ > ./data/"$function"."$connection"."$time"."$datetime".txt
+                        if [[ $* = *"--loadtest"* ]]
+                            then
+                                loadtest -H Host: $function.default.example.com --rps $leadtest_rps -c $connection -t $time http://$kuberhost/ > ./data/"$function"."$connection"."$time"."$datetime".txt
+                            else
+                                hey -c "$connection" -z "$time" -m POST -host "$function.default.example.com" http://$kuberhost/ > ./data/"$function"."$connection"."$time"."$datetime".txt
+                            fi
                         fi
                     fi
                     if [[ $* = *"--csv"* ]]