add partial failure tests for batchget

m-nagarajan · Jul 25, 2023 · acc42fe · acc42fe
1 parent dbd8ccf
commit acc42fe
Show file tree

Hide file tree

Showing 5 changed files with 310 additions and 215 deletions.
diff --git a/...nice-client/src/main/java/com/linkedin/venice/fastclient/StatsAvroGenericStoreClient.java b/...nice-client/src/main/java/com/linkedin/venice/fastclient/StatsAvroGenericStoreClient.java
@@ -226,6 +226,9 @@ private <R> CompletableFuture<R> recordRequestMetrics(
           clientStats.recordRetryRequestKeyCount(batchGetRequestContext.numberOfKeysSentInRetryRequest);
           clientStats
               .recordRetryRequestSuccessKeyCount(batchGetRequestContext.numberOfKeysCompletedInRetryRequest.get());
+          if (batchGetRequestContext.numberOfKeysCompletedInRetryRequest.get() > 0) {
+            clientStats.recordRetryRequestWin();
+          }
         }
       }