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();