Don't wait forever

NeuraLegion · Aug 8, 2023 · 448319d · 448319d
1 parent ff6cafc
commit 448319d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/sec_tester/repeater.cr b/src/sec_tester/repeater.cr
@@ -48,6 +48,7 @@ module SecTester
       # handle undeployed events
       @socket.on("undeployed") do |_|
         @id = ""
+        @socket.off("undeployed")
       end
 
       Log.debug { "Repeater started" }

diff --git a/src/sec_tester/scan.cr b/src/sec_tester/scan.cr
@@ -23,9 +23,13 @@ module SecTester
       spawn do
         @repeater.run
       end
-      while @repeater.id.empty?
+      # Wait for repeater to get ID
+      # Don't wait forever, if repeater is not running, we will get an error
+      # after 30 seconds
+
+      30.times do
+        break unless @repeater.id.empty?
         sleep 1.second
-        Fiber.yield
         Log.debug { "Waiting for repeater to get ID" }
       end
     end