in remove_backend, call tick() then cleanup() on it

imandra-ai · Oct 16, 2024 · 1924ff7 · 1924ff7
1 parent 0ef97a0
commit 1924ff7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/core/opentelemetry.ml b/src/core/opentelemetry.ml
@@ -175,7 +175,12 @@ module Collector = struct
 
   (** Remove current backend, if any.
       @since NEXT_RELEASE *)
-  let remove_backend () : unit = Atomic.set backend None
+  let remove_backend () : unit =
+    match Atomic.exchange backend None with
+    | None -> ()
+    | Some (module B) ->
+      B.tick ();
+      B.cleanup ()
 
   (** Is there a configured backend? *)
   let[@inline] has_backend () : bool = Atomic.get backend != None