From 0985280517bf2acb96f522c6ef0626c8ffe77d09 Mon Sep 17 00:00:00 2001 From: John Wagster Date: Thu, 31 Jul 2025 14:22:57 -0500 Subject: [PATCH] added java version and reset num docs reported to actual ingested --- .../main/java/org/elasticsearch/test/knn/KnnIndexTester.java | 3 ++- .../src/main/java/org/elasticsearch/test/knn/KnnIndexer.java | 3 +++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexTester.java b/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexTester.java index 29dbe5613469c..5d7f79f828d6d 100644 --- a/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexTester.java +++ b/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexTester.java @@ -205,6 +205,7 @@ public static void main(String[] args) throws Exception { cmdLineArgs.filterSelectivity() ); } + logger.info("Running with Java: " + Runtime.version()); logger.info("Running KNN index tester with arguments: " + cmdLineArgs); Codec codec = createCodec(cmdLineArgs); Path indexPath = PathUtils.get(formatIndexPath(cmdLineArgs)); @@ -382,7 +383,7 @@ private int[] calculateColumnWidths(String[] headers, String[]... data) { static class Results { final String indexType, indexName; - final int numDocs; + int numDocs; final float filterSelectivity; long indexTimeMS; long forceMergeTimeMS; diff --git a/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexer.java b/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexer.java index aa8792bb2c4a5..0f0174b16dd20 100644 --- a/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexer.java +++ b/qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexer.java @@ -206,6 +206,9 @@ public boolean isEnabled(String component) { long elapsed = System.nanoTime() - start; logger.debug("Indexing took {} ms for {} docs", TimeUnit.NANOSECONDS.toMillis(elapsed), numDocs); result.indexTimeMS = TimeUnit.NANOSECONDS.toMillis(elapsed); + + // report numDocsIndexed here in case we have less than the total numDocs + result.numDocs = numDocsIndexed.get(); } void forceMerge(KnnIndexTester.Results results) throws Exception {