add wait for archive node bootstrap

Author: dkijania

src/test/archive/archive_node_tests/archive_precomputed_blocks_test.ml

@@ -124,6 +124,7 @@ let test_case (test_data : t) =
   in
   let log_file = output ^ "/precomputed_blocks_test.log" in
   Archive.Process.start_logging test_data.archive ~log_file ;
+  let%bind () = Archive.wait_for ~log:log_file in
   let%bind () =
     Daemon.archive_blocks_from_files daemon.executor
       ~archive_address:test_data.archive.config.server_port ~format:`Precomputed

src/test/mina_automation/archive.ml

@@ -3,6 +3,7 @@ Module to run archive process over archive PostgreSQL database.
 *)
 open Core
 
+open Logger
 open Async
 
 module Config = struct
@@ -118,3 +119,44 @@ let start t =
 
   let archive_process : Process.t = { process; config = t.config } in
   Deferred.return archive_process
+
+let wait_for ~log =
+  let open Deferred.Let_syntax in
+  let timeout = 30.0 in
+  let start_time = Time.now () in
+  let rec loop () =
+    Core.Unix.sleep 1 ;
+    let%bind log_exists =
+      Sys.file_exists log
+      >>| fun exists -> match exists with `Yes -> true | _ -> false
+    in
+    if not log_exists then
+      if
+        int_of_float (Time.Span.to_sec (Time.diff (Time.now ()) start_time))
+        > int_of_float timeout
+      then (
+        eprintf "Timeout waiting for log file to be created\n" ;
+        Deferred.return () )
+      else loop ()
+    else
+      let%bind log_contents = Reader.file_contents log in
+      let lines = String.split_lines log_contents in
+      let found =
+        List.exists lines ~f:(fun line ->
+            match Yojson.Safe.from_string line |> Logger.Message.of_yojson with
+            | Ok msg ->
+                String.equal msg.message
+                  "Archive process ready. Clients can now connect"
+            | Error _ ->
+                false )
+      in
+      if found then Deferred.return ()
+      else if
+        int_of_float (Time.Span.to_sec (Time.diff (Time.now ()) start_time))
+        > int_of_float timeout
+      then (
+        eprintf "Timeout waiting for archive process to be ready\n" ;
+        Deferred.return () )
+      else loop ()
+  in
+  loop ()