Handles GRPC::NotFound exceptions

Also adds support for reporting the number of connections attempts have been made.
corgibytes · Sep 1, 2023 · 1c6d997 · 1c6d997
1 parent dd539e6
commit 1c6d997
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/corgibytes/freshli/commons/grpc_client.rb b/lib/corgibytes/freshli/commons/grpc_client.rb
@@ -104,11 +104,15 @@ def is_running!
         # rubocop:disable Metrics/MethodLength
         def wait_until_running!
           Timeout.timeout(5) do
+            attempts = 0
             loop do
+              attempts += 1
+              yield(attempts) if block_given?
+
               status = nil
               begin
                 status = health_check
-              rescue GRPC::Unavailable
+              rescue GRPC::Unavailable, GRPC::NotFound
                 status = nil
               end