code updated for the new java driver

Author: miguelangelnieto

ycsb-mongodb/mongodb/src/main/java/com/yahoo/ycsb/db/MongoDbClient.java

@@ -145,19 +145,19 @@ private static String generateRemoteSchema(String keyId, int numFields) {
         return "{ $jsonSchema : " + generateSchema(keyId, numFields) + "}";
     }
 
-    private static synchronized String getDataKeyOrCreate(MongoCollection<Document> keyCollection, ClientEncryption clientEncryption ) {
-        Document findFilter = new Document();
-        Document keyDoc = keyCollection.find(findFilter).first();
-
+    private static synchronized String getDataKeyOrCreate(MongoCollection<BsonDocument> keyCollection, ClientEncryption clientEncryption ) {
+        BsonDocument findFilter = new BsonDocument();
+        BsonDocument keyDoc = keyCollection.find(findFilter).first();
+ 
         String base64DataKeyId;
-        if(keyDoc == null ) {
+        if (keyDoc == null ) {
             BsonBinary dataKeyId = clientEncryption.createDataKey("local", new DataKeyOptions());
             base64DataKeyId = Base64.getEncoder().encodeToString(dataKeyId.getData());
         } else {
-            UUID dataKeyId = (UUID) keyDoc.get("_id");
+            UUID dataKeyId = keyDoc.getBinary("_id").asUuid();
             base64DataKeyId = Base64.getEncoder().encodeToString(UuidUtils.asBytes(dataKeyId));
         }
-
+ 
         return base64DataKeyId;
     }
 
@@ -196,7 +196,7 @@ private static AutoEncryptionSettings generateEncryptionSettings(String url, Boo
 
         MongoClient vaultClient = MongoClients.create(keyVaultUrls);
 
-        final MongoCollection<Document> keyCollection = vaultClient.getDatabase(database).getCollection(keyVaultNamespace);
+        final MongoCollection<BsonDocument> keyCollection = vaultClient.getDatabase(database).getCollection(keyVaultNamespace, BsonDocument.class);
 
         String base64DataKeyId = getDataKeyOrCreate(keyCollection, clientEncryption);