Upgrade MongoDB client version. (#1758)

Upgrades the MongoDB sync driver to a more up-to-date version.

Author: bvliu

mongodb/pom.xml

@@ -33,7 +33,7 @@ LICENSE file.
   <dependencies>
     <dependency>
       <groupId>org.mongodb</groupId>
-      <artifactId>mongo-java-driver</artifactId>
+      <artifactId>mongodb-driver-sync</artifactId>
       <version>${mongodb.version}</version>
     </dependency>
     <dependency>

mongodb/src/main/java/site/ycsb/db/MongoDbClient.java

@@ -24,17 +24,20 @@
  */
 package site.ycsb.db;
 
-import com.mongodb.MongoClient;
-import com.mongodb.MongoClientURI;
+import com.mongodb.MongoClientSettings;
+import com.mongodb.ConnectionString;
 import com.mongodb.ReadPreference;
 import com.mongodb.WriteConcern;
 import com.mongodb.client.FindIterable;
+import com.mongodb.client.MongoClient;
+import com.mongodb.client.MongoClients;
 import com.mongodb.client.MongoCollection;
 import com.mongodb.client.MongoCursor;
 import com.mongodb.client.MongoDatabase;
 import com.mongodb.client.model.InsertManyOptions;
 import com.mongodb.client.model.UpdateOneModel;
 import com.mongodb.client.model.UpdateOptions;
+import com.mongodb.client.model.ReplaceOptions;
 import com.mongodb.client.result.DeleteResult;
 import com.mongodb.client.result.UpdateResult;
 import site.ycsb.ByteArrayByteIterator;
@@ -78,6 +81,8 @@ public class MongoDbClient extends DB {
   /** The options to use for inserting a single document. */
   private static final UpdateOptions UPDATE_WITH_UPSERT = new UpdateOptions()
       .upsert(true);
+  private static final ReplaceOptions REPLACE_WITH_UPSERT = new ReplaceOptions()
+      .upsert(true);
 
   /**
    * The database name to access.
@@ -204,7 +209,8 @@ public void init() throws DBException {
       }
 
       try {
-        MongoClientURI uri = new MongoClientURI(url);
+        ConnectionString uri = new ConnectionString(url);
+        MongoClientSettings.Builder csb = MongoClientSettings.builder().applyConnectionString(uri);
 
         String uriDb = uri.getDatabase();
         if (!defaultedUrl && (uriDb != null) && !uriDb.isEmpty()
@@ -216,14 +222,8 @@ public void init() throws DBException {
 
         }
 
-        readPreference = uri.getOptions().getReadPreference();
-        writeConcern = uri.getOptions().getWriteConcern();
-
-        mongoClient = new MongoClient(uri);
-        database =
-            mongoClient.getDatabase(databaseName)
-                .withReadPreference(readPreference)
-                .withWriteConcern(writeConcern);
+        mongoClient = MongoClients.create(csb.build());
+        database = mongoClient.getDatabase(databaseName);
 
         System.out.println("mongo client connection created with " + url);
       } catch (Exception e1) {
@@ -266,7 +266,7 @@ public Status insert(String table, String key,
           // to current inability of the framework to clean up after itself
           // between test runs.
           collection.replaceOne(new Document("_id", toInsert.get("_id")),
-              toInsert, UPDATE_WITH_UPSERT);
+              toInsert, REPLACE_WITH_UPSERT);
         } else {
           collection.insertOne(toInsert);
         }
@@ -279,7 +279,7 @@ public Status insert(String table, String key,
             for (Document doc : bulkInserts) {
               updates.add(new UpdateOneModel<Document>(
                   new Document("_id", doc.get("_id")),
-                  doc, UPDATE_WITH_UPSERT));
+                  new Document("$set", doc), UPDATE_WITH_UPSERT));
             }
             collection.bulkWrite(updates);
           } else {

pom.xml

@@ -135,7 +135,7 @@ LICENSE file.
     <infinispan.version>7.2.2.Final</infinispan.version>
     <kudu.version>1.11.1</kudu.version>
     <maprhbase.version>1.1.8-mapr-1710</maprhbase.version>
-    <mongodb.version>3.11.0</mongodb.version>
+    <mongodb.version>5.1.0</mongodb.version>
     <mongodb.async.version>2.0.1</mongodb.async.version>
     <openjpa.jdbc.version>2.1.1</openjpa.jdbc.version>
     <orientdb.version>2.2.37</orientdb.version>