prometheus: Add CollectorRegistry.unregister functions

Sources of metrics can sometimes disappear at runtime (a VM being
destroyed, disk being unplugged), and rather than continuing to present
the last value, it is useful to remove them altogether.

Other Prometheus client libraries do implement similar functions as
well, see https://docs.rs/prometheus/latest/prometheus/struct.Registry.html#method.unregister
for example.

Signed-off-by: Andrii Sultanov <andrii.sultanov@cloud.com>

Author: Andrii Sultanov

src/prometheus.ml

@@ -121,10 +121,16 @@ module CollectorRegistry = struct
     ensure_not_registered t info;
     t.metrics <- MetricFamilyMap.add info collector t.metrics
 
+  let unregister t info =
+    t.metrics <- MetricFamilyMap.remove info t.metrics
+
   let register_lwt t info collector =
     ensure_not_registered t info;
     t.metrics_lwt <- MetricFamilyMap.add info collector t.metrics_lwt
 
+  let unregister_lwt t info =
+    t.metrics_lwt <- MetricFamilyMap.remove info t.metrics_lwt
+
   open Lwt.Infix
 
   let map_p m =