wait for port

cloudwalk · Dec 2, 2024 · ee505f8 · ee505f8
1 parent 06c27b2
commit ee505f8
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/utils/benchmark_rocks_config/main.py b/utils/benchmark_rocks_config/main.py
@@ -35,7 +35,18 @@ def run_stratus():
     )
     print("Started stratus")
     # Give it some time to start up
-    time.sleep(10)
+    print("Waiting for stratus to start...")
+    port_open = False
+    while not port_open:
+        try:
+            result = subprocess.run(["lsof", "-i", ":3000"], capture_output=True, text=True)
+            if result.stdout:
+                port_open = True
+            else:
+                time.sleep(1)
+        except subprocess.CalledProcessError:
+            time.sleep(1)
+    print("Port 3000 is now in use, stratus has started")
     return process
 
 def start_benchmark():