Skip to content

Commit 0985280

Browse files
committed
added java version and reset num docs reported to actual ingested
1 parent a857290 commit 0985280

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexTester.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,6 +205,7 @@ public static void main(String[] args) throws Exception {
205205
cmdLineArgs.filterSelectivity()
206206
);
207207
}
208+
logger.info("Running with Java: " + Runtime.version());
208209
logger.info("Running KNN index tester with arguments: " + cmdLineArgs);
209210
Codec codec = createCodec(cmdLineArgs);
210211
Path indexPath = PathUtils.get(formatIndexPath(cmdLineArgs));
@@ -382,7 +383,7 @@ private int[] calculateColumnWidths(String[] headers, String[]... data) {
382383

383384
static class Results {
384385
final String indexType, indexName;
385-
final int numDocs;
386+
int numDocs;
386387
final float filterSelectivity;
387388
long indexTimeMS;
388389
long forceMergeTimeMS;

qa/vector/src/main/java/org/elasticsearch/test/knn/KnnIndexer.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,9 @@ public boolean isEnabled(String component) {
206206
long elapsed = System.nanoTime() - start;
207207
logger.debug("Indexing took {} ms for {} docs", TimeUnit.NANOSECONDS.toMillis(elapsed), numDocs);
208208
result.indexTimeMS = TimeUnit.NANOSECONDS.toMillis(elapsed);
209+
210+
// report numDocsIndexed here in case we have less than the total numDocs
211+
result.numDocs = numDocsIndexed.get();
209212
}
210213

211214
void forceMerge(KnnIndexTester.Results results) throws Exception {

0 commit comments

Comments
 (0)