Support adding a set of suffixes for a given metric.

1 files changed, 10 insertions, 0 deletions

diff --git a/metrics/src/main/java/ai/vespa/metrics/set/MetricSet.java b/metrics/src/main/java/ai/vespa/metrics/set/MetricSet.java
index d502109e24e..f334690a7ca 100644
--- a/metrics/src/main/java/ai/vespa/metrics/set/MetricSet.java
+++ b/metrics/src/main/java/ai/vespa/metrics/set/MetricSet.java
@@ -1,7 +1,11 @@
 // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package ai.vespa.metrics.set;
 
+import ai.vespa.metrics.Suffix;
+import ai.vespa.metrics.VespaMetrics;
+
 import java.util.Collection;
+import java.util.EnumSet;
 import java.util.LinkedHashMap;
 import java.util.LinkedHashSet;
 import java.util.Map;
@@ -102,6 +106,12 @@ public class MetricSet {
             return metric(new Metric(metric));
         }
 
+        /** Adds all given suffixes of the given metric to this set. */
+        public Builder metric(VespaMetrics metric, EnumSet<Suffix> suffixes) {
+            suffixes.forEach(suffix -> metrics.add(new Metric(metric.baseName() + "." + suffix.suffix())));
+            return this;
+        }
+
         public Builder metric(Metric metric) {
             metrics.add(metric);
             return this;