From f7da12e09482c2d6a6c64ea208d0cc2ef0f1e0bd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Torma=20Krist=C3=B3f?= <tormakristof@tormakristof.eu>
Date: Tue, 2 Apr 2019 23:01:26 +0200
Subject: [PATCH] Update benchmark.sh

---
 functions/benchmark.sh | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/functions/benchmark.sh b/functions/benchmark.sh
index 4216458..568a48a 100644
--- a/functions/benchmark.sh
+++ b/functions/benchmark.sh
@@ -5,6 +5,22 @@ functions=(helloget matrix)
 connections=(2 5 10 20 21 50 100 200 400 500 1000)
 
 kuberhost="node1:32764"
+WRK_INSTALLED=$(which wrk)
+if [[ "$WRK_INSTALLED" -eq "" ]]
+then
+	apt update
+    apt install -y wrk
+fi
+
+HEY_INSTALLED=$(which hey)
+if [[ "$HEY_INSTALLED" -eq "" ]]
+then
+	apt update
+    apt install -y golang
+    go get -u github.com/rakyll/hey
+    export GOPATH=$HOME/go
+    export PATH=$PATH:$GOPATH/bin
+fi
 
 for function in "${functions[@]}"
 do