From cf528847ece9efcb9e1ad0e124dceac38f963db9 Mon Sep 17 00:00:00 2001 From: Jonathan Shook Date: Sat, 14 Oct 2023 22:33:40 -0500 Subject: [PATCH] update key extraction to fit returned server format --- .../java/io/nosqlbench/adapter/mongodb/MongoDbUtils.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/adapter-mongodb/src/main/java/io/nosqlbench/adapter/mongodb/MongoDbUtils.java b/adapter-mongodb/src/main/java/io/nosqlbench/adapter/mongodb/MongoDbUtils.java index 853b1dbfa..62f4d113a 100644 --- a/adapter-mongodb/src/main/java/io/nosqlbench/adapter/mongodb/MongoDbUtils.java +++ b/adapter-mongodb/src/main/java/io/nosqlbench/adapter/mongodb/MongoDbUtils.java @@ -1,5 +1,6 @@ package io.nosqlbench.adapter.mongodb; +import com.mongodb.client.MongoCursor; import org.bson.Document; import java.util.ArrayList; @@ -7,9 +8,10 @@ import java.util.List; public class MongoDbUtils { public static int[] getFieldFromResults(String field, Document document) { - List value = document.getList(field, Document.class); + Document _cursor = document.get("cursor", Document.class); + List _firstBatch = _cursor.getList("firstBatch", Document.class); List keyStrings = new ArrayList<>(); - for (Document matchingVector : value) { + for (Document matchingVector : _firstBatch) { keyStrings.add(matchingVector.get("key",String.class)); } return keyStrings.stream().mapToInt(Integer::parseInt).toArray();