Add convenience method for retrieving stats for given code

Also add trailing new line to `BenchmarkingClusterTest`
vespa-engine · Jan 27, 2025 · 0a5dadf · 0a5dadf
1 parent 6a359ca
commit 0a5dadf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/vespa-feed-client-api/src/main/java/ai/vespa/feed/client/OperationStats.java b/vespa-feed-client-api/src/main/java/ai/vespa/feed/client/OperationStats.java
@@ -2,6 +2,7 @@
 package ai.vespa.feed.client;
 
 import java.util.Map;
+import java.util.Optional;
 
 /**
  * Statistics for feed operations over HTTP against a Vespa cluster.
@@ -57,6 +58,9 @@ public long successes() {
     /** Statistics per response code. */
     public Map<Integer, Response> statsByCode() { return statsByCode; }
 
+    /** Statistics for the given code. */
+    public Optional<Response> response(int code) { return Optional.ofNullable(statsByCode.get(code)); }
+
     /** Number of exceptions (instead of responses). */
     public long exceptions() {
         return exceptions;

diff --git a/vespa-feed-client/src/test/java/ai/vespa/feed/client/impl/BenchmarkingClusterTest.java b/vespa-feed-client/src/test/java/ai/vespa/feed/client/impl/BenchmarkingClusterTest.java
@@ -59,4 +59,4 @@ private static Void processRequest(AtomicLong nanoClock, InvocationOnMock invoca
         vessel.complete(HttpResponse.of(statusCode, "body".getBytes(StandardCharsets.UTF_8)));
         return null;
     }
-}
+}
-Original file line number
+Diff line change
@@ Expand Up @@
             vessel.complete(HttpResponse.of(statusCode, "body".getBytes(StandardCharsets.UTF_8)));
             return null;
         }
-    }
+    }