From 6db5c7649be957e34a66d4fe6dc1abd99bccd965 Mon Sep 17 00:00:00 2001
From: Arne H Juul <arnej@yahoo-inc.com>
Date: Thu, 3 Nov 2016 13:10:08 +0100
Subject: make available map for a metric name (for unit testing)

---
 .../src/main/java/com/yahoo/metrics/simple/Bucket.java         | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'simplemetrics')

diff --git a/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java b/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java
index 4ae284c4304..45393ff5f77 100644
--- a/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java
+++ b/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java
@@ -146,6 +146,16 @@ public class Bucket {
         return singleMetric;
     }
 
+    public Map<Point, UntypedMetric> getMapForMetric(@NonNull String metricName) {
+        Map<Point, UntypedMetric> result = new HashMap<>();
+        for (Map.Entry<Identifier, UntypedMetric> entry : values.entrySet()) {
+            if (metricName.equals(entry.getKey().getName())) {
+                result.put(entry.getKey().getLocation(), entry.getValue());
+            }
+        }
+        return result;
+    }
+
     public Map<String, List<Map.Entry<Point, UntypedMetric>>> getValuesByMetricName() {
         Map<String, List<Map.Entry<Point, UntypedMetric>>> result = new HashMap<>();
         for (Map.Entry<Identifier, UntypedMetric> entry : values.entrySet()) {
-- 
cgit v1.2.3