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

@@ -2,7 +2,7 @@
 Module to run archive process over archive PostgreSQL database.
 *)
 open Core
-
+open Logger
 open Async
 
 module Config = struct
@@ -118,3 +118,40 @@ 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 ()