convert RateLimitingSearcher to use simplemetrics

2 files changed, 20 insertions, 9 deletions

diff --git a/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingBenchmark.java b/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingBenchmark.java
index 9381cf2ab7e..b58abce7f5f 100644
--- a/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingBenchmark.java
+++ b/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingBenchmark.java
@@ -3,7 +3,10 @@ package com.yahoo.search.searchers.test;
 
 import com.yahoo.cloud.config.ClusterInfoConfig;
 import com.yahoo.component.chain.Chain;
-import com.yahoo.jdisc.Metric;
+import com.yahoo.metrics.simple.Bucket;
+import com.yahoo.metrics.simple.MetricReceiver;
+import com.yahoo.metrics.simple.Point;
+import com.yahoo.metrics.simple.UntypedMetric;
 import com.yahoo.search.Query;
 import com.yahoo.search.Result;
 import com.yahoo.search.Searcher;
@@ -41,7 +44,8 @@ public class RateLimitingBenchmark {
     private final int timeBetweenPeaksMs = 2000;
 
     private final Chain<Searcher> chain;
-    private final MockMetric metric;
+    private final MetricReceiver metric;
+    private Bucket metricSnapshot;
 
     private final Map<String, RequestCounts> requestCounters = new HashMap<>();
 
@@ -61,7 +65,7 @@ public class RateLimitingBenchmark {
         clusterInfoConfig.clusterId("testCluster");
         clusterInfoConfig.nodeCount(1);
 
-        this.metric = new MockMetric();
+        this.metric = new MetricReceiver.MockReceiver();
 
         chain = new Chain<>("test", new RateLimitingSearcher(new RateLimitingConfig(rateLimitingConfig),
                                     new ClusterInfoConfig(clusterInfoConfig), metric));
@@ -75,6 +79,7 @@ public class RateLimitingBenchmark {
         runWorkers();
         long totalTime = Math.max(1, System.currentTimeMillis() - startTime);
 
+        metricSnapshot = metric.getSnapshot();
         double totalAttemptedRate = 0;
         for (int i=0; i < clientCount; i++) {
             double attemptedRate = requestCounters.get(toClientId(i)).attempted.get() * 1000d / totalTime;
@@ -108,10 +113,10 @@ public class RateLimitingBenchmark {
     }
 
     private int rejectedRequests(int id) {
-        Metric.Context context = metric.createContext("id", toClientId(id));
-        Number rejectedRequestsMetric = metric.values(context).get("requestsOverQuota");
+        Point context = metric.pointBuilder().set("id", toClientId(id)).build();
+        UntypedMetric rejectedRequestsMetric = metricSnapshot.getMapForMetric("requestsOverQuota").get(context);
         if (rejectedRequestsMetric == null) return 0;
-        return rejectedRequestsMetric.intValue();
+        return (int)rejectedRequestsMetric.getCount();
     }
 
     private class Worker implements Runnable {
diff --git a/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingSearcherTestCase.java b/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingSearcherTestCase.java
index 02d6620df2e..e210213c515 100755
--- a/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingSearcherTestCase.java
+++ b/container-search/src/test/java/com/yahoo/search/searchers/test/RateLimitingSearcherTestCase.java
@@ -3,6 +3,10 @@ package com.yahoo.search.searchers.test;
 
 import com.yahoo.cloud.config.ClusterInfoConfig;
 import com.yahoo.component.chain.Chain;
+import com.yahoo.metrics.simple.Bucket;
+import com.yahoo.metrics.simple.MetricReceiver;
+import com.yahoo.metrics.simple.Point;
+import com.yahoo.metrics.simple.UntypedMetric;
 import com.yahoo.search.Query;
 import com.yahoo.search.Result;
 import com.yahoo.search.Searcher;
@@ -14,6 +18,7 @@ import org.junit.Test;
 import com.yahoo.test.ManualClock;
 
 import java.time.Duration;
+import java.util.Map;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
@@ -38,7 +43,7 @@ public class RateLimitingSearcherTestCase {
         clusterInfoConfig.nodeCount(4);
 
         ManualClock clock = new ManualClock();
-        MockMetric metric = new MockMetric();
+        MetricReceiver.MockReceiver metric = new MetricReceiver.MockReceiver();
 
         Chain<Searcher> chain = new Chain<Searcher>("test", new RateLimitingSearcher(new RateLimitingConfig(rateLimitingConfig),
                                                                               new ClusterInfoConfig(clusterInfoConfig),
@@ -68,8 +73,9 @@ public class RateLimitingSearcherTestCase {
         assertTrue(executeWasAllowed(chain, "id1"));
 
         // check metrics
-        assertEquals((double)requestsToTry-2 + 1 + requestsToTry-2 + 3, metric.values(metric.createContext("id", "id1")).get("requestsOverQuota"));
-        assertEquals((double)requestsToTry-2 + requestsToTry-4, metric.values(metric.createContext("id", "id2")).get("requestsOverQuota"));
+        Map<Point,UntypedMetric> map = metric.getSnapshot().getMapForMetric("requestsOverQuota");
+        assertEquals(requestsToTry-2 + 1 + requestsToTry-2 + 3, map.get(metric.point("id", "id1")).getCount());
+        assertEquals(requestsToTry-2 + requestsToTry-4,         map.get(metric.point("id", "id2")).getCount());
     }
 
     private int requestsToTry = 50;