improve script config

devcryptoph · Oct 22, 2018 · 7e4b55b · 7e4b55b
1 parent 1a50093
commit 7e4b55b
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/load-gen/load-gen.sh b/load-gen/load-gen.sh
@@ -1,11 +1,19 @@
 #!/bin/sh
 
+# set -x
+
 # Changing the NUM_CLIENTS environment variable varies the load on the application
 # The bigger the number the more requests, the bigger the load
+NUM_CLIENTS=1
+
+# HOST where Stan's Robot Shop web UI is running
+HOST="http://localhost:8080"
 
-# Set SILENT to anything to have all output discarded. Useful when running load for
-# a long time to stop the disk filling up with copious log messages
-# -e 'SILENT=1'
+if [ ! -f ../.env ]
+then
+    echo "Please run this script from the load-gen directory"
+    exit 1
+fi
 
 # get the tag info
 eval $(egrep '[A-Z]+=' ../.env)
@@ -20,17 +28,17 @@ then
         -d \
         --rm \
         --network=host \
-        -e 'HOST=http://localhost:8080' \
-        -e 'NUM_CLIENTS=1' \
+        -e "HOST=$HOST" \
+        -e "NUM_CLIENTS=$NUM_CLIENTS" \
         -e 'SILENT=1' \
         ${REPO}/rs-load:${TAG}
 else
     docker run \
         -it \
         --rm \
         --network=host \
-        -e 'HOST=http://localhost:8080' \
-        -e 'NUM_CLIENTS=1' \
+        -e "HOST=$HOST" \
+        -e "NUM_CLIENTS=$NUM_CLIENTS" \
         ${REPO}/rs-load:${TAG}
 fi