IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

update key extraction to fit returned server format

Browse Source
This commit is contained in:
Jonathan Shook
2023-10-14 22:33:40 -05:00
parent 588b6d6ba1
commit cf528847ec
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
adapter-mongodb/src/main/java/io/nosqlbench/adapter/mongodb/MongoDbUtils.java
View File

@@ -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<Document> value = document.getList(field, Document.class);
Document _cursor = document.get("cursor", Document.class);
List<Document> _firstBatch = _cursor.getList("firstBatch", Document.class);
List<String> 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();