Update wrk benchmarks: Add a wait_until_live

Author: bitterpanda63

benchmarks/starlette_benchmark.py

@@ -2,11 +2,27 @@
 import subprocess
 import sys
 import time
+import requests
 
 def generate_wrk_command_for_url(url):
     # Define the command with awk included
     return "wrk -t5 -c200 -d15s " + url
 
+def wait_until_live(url):
+    for i in range(30):
+        try:
+            res = requests.get(url, timeout=5)
+            if res.status_code == 200:
+                print("Server is live: " + url)
+                return True
+            else:
+                print("Status code " + str(res.status_code) + " for " + url)
+        except requests.RequestException as e:
+            print(f"Request failed: {e}")
+        time.sleep(5)
+    return False
+
+
 def extract_requests_and_latency_tuple(output):
     if output.returncode == 0:
         # Extracting requests/sec
@@ -23,6 +39,11 @@ def extract_requests_and_latency_tuple(output):
         sys.exit(1)
 
 def run_benchmark(route1, route2, descriptor, percentage_limit):
+    # Make sure both routes are accessible before running benchmark
+    if not wait_until_live(route1):
+        raise Exception("Unable to access: " + route1)
+    if not wait_until_live(route2):
+        raise Exception("Unable to access: " + route2)
 
     output_nofw = subprocess.run(
         generate_wrk_command_for_url(route2),