Add clear() method into PrometheusRegistry to unregister all collectors

Signed-off-by: Petr Holubec <[email protected]>
prometheus · Jun 18, 2024 · cd28469 · cd28469
1 parent 5867b79
commit cd28469
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/...-metrics-model/src/main/java/io/prometheus/metrics/model/registry/PrometheusRegistry.java b/...-metrics-model/src/main/java/io/prometheus/metrics/model/registry/PrometheusRegistry.java
@@ -53,6 +53,12 @@ public void unregister(MultiCollector collector) {
 		}
 	}
 
+    public void clear() {
+        collectors.clear();
+        multiCollectors.clear();
+        prometheusNames.clear();
+    }
+
 	public MetricSnapshots scrape() {
 		return scrape((PrometheusScrapeRequest) null);
 	}

diff --git a/...rics-model/src/test/java/io/prometheus/metrics/model/registry/PrometheusRegistryTest.java b/...rics-model/src/test/java/io/prometheus/metrics/model/registry/PrometheusRegistryTest.java
@@ -101,4 +101,18 @@ public void registerOk() {
         snapshots = registry.scrape();
         Assert.assertEquals(3, snapshots.size());
     }
+
+    @Test
+    public void clearOk() {
+        PrometheusRegistry registry = new PrometheusRegistry();
+        registry.register(counterA1);
+        registry.register(counterB);
+        registry.register(gaugeA);
+        MetricSnapshots snapshots = registry.scrape();
+        Assert.assertEquals(3, snapshots.size());
+
+        registry.clear();
+        snapshots = registry.scrape();
+        Assert.assertEquals(0, snapshots.size());
+    }
 }