in backends, call tick() before cleaning up

this helps flushing signals that are being batched. close #69
imandra-ai · Oct 16, 2024 · 276ba2e · 276ba2e
1 parent 1924ff7
commit 276ba2e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/client-cohttp-lwt/opentelemetry_client_cohttp_lwt.ml b/src/client-cohttp-lwt/opentelemetry_client_cohttp_lwt.ml
@@ -569,7 +569,11 @@ let setup_ ?stop ?config () =
   let backend = create_backend ?stop ?config () in
   let (module B : OT.Collector.BACKEND) = backend in
   OT.Collector.set_backend backend;
-  B.cleanup
+  let cleanup () =
+    B.tick ();
+    B.cleanup ()
+  in
+  cleanup
 
 let setup ?stop ?config ?(enable = true) () =
   if enable then (

diff --git a/src/client-ocurl/opentelemetry_client_ocurl.ml b/src/client-ocurl/opentelemetry_client_ocurl.ml
@@ -509,7 +509,11 @@ let setup_ ?(stop = Atomic.make false) ?(config : Config.t = Config.make ()) ()
     ignore (setup_ticker_thread ~stop ~sleep_ms backend () : Thread.t)
   );
 
-  B.cleanup
+  let cleanup () =
+    B.tick ();
+    B.cleanup ()
+  in
+  cleanup
 
 let setup ?stop ?config ?(enable = true) () =
   if enable then (