1. 概述

在这个教程中,我们将探讨如何在MongoDB中执行批量更新和插入操作。此外,MongoDB提供了API调用来在一个操作中插入或检索多个文档。MongoDB使用ArrayBatch接口,通过减少客户端和数据库之间的调用次数,极大地提高了数据库性能。

我们将通过MongoDB Shell和Java驱动代码两种方式来实现文档的批量更新。

现在,让我们深入实施MongoDB中的文档批量更新操作。

2. 数据库初始化

首先,我们需要连接到MongoDB shell:

mongo --host localhost --port 27017

接下来,设置一个名为baeldung的数据库和一个示例集合populations

use baeldung;
db.createCollection(populations);

现在,我们可以使用insertMany方法向populations集合添加一些示例数据:

db.populations.insertMany([
{
    "cityId":1124,
    "cityName":"New York",
    "countryName":"United States",
    "continentName":"North America",
    "population":22
},
{
    "cityId":1125,
    "cityName":"Mexico City",
    "countryName":"Mexico",
    "continentName":"North America",
    "population":25
},
{
    "cityId":1126,
    "cityName":"New Delhi",
    "countryName":"India",
    "continentName":"Asia",
    "population":45
},
{
    "cityId":1134,
    "cityName":"London",
    "countryName":"England",
    "continentName":"Europe",
    "population":32
}]);

上述insertMany查询将返回以下文档:

{
    "acknowledged" : true,
    "insertedIds" : [
        ObjectId("623575049d55d4e137e477f6"),
        ObjectId("623575049d55d4e137e477f7"),
        ObjectId("623575049d55d4e137e477f8"),
        ObjectId("623575049d55d4e137e477f9")
    ]
}

在这个查询中,我们插入了四条文档,以便在MongoDB中执行所有类型的写入批量操作。

数据库baeldung已成功创建,并且所有所需数据已插入到populations集合中,因此我们准备好执行批量更新了。

3. 使用MongoDB Shell查询

MongoDB的批量操作构建器用于为单个集合构造一个批量写入操作列表。我们可以以两种不同的方式初始化批量操作。initializeOrderedBulkOp方法用于按顺序执行批量操作。**initializeOrderedBulkOp的一个缺点是,如果在处理任何写入操作时发生错误,MongoDB将在不处理列表中剩余写入操作的情况下返回。**

我们可以使用insertupdatereplaceremove方法在一个DB调用中执行不同类型的操作。下面是一个使用MongoDB shell查询的批量写入操作示例:

db.populations.bulkWrite([
    { 
        insertOne :
            { 
                "document" :
                    {
                        "cityId":1128,
                        "cityName":"Kathmandu",
                        "countryName":"Nepal",
                        "continentName":"Asia",
                        "population":12
                    }
            }
    },
    { 
        insertOne :
            { 
                "document" :
                    {
                        "cityId":1130,
                        "cityName":"Mumbai",
                        "countryName":"India",
                        "continentName":"Asia",
                        "population":55
                    }
            }
    },
    { 
        updateOne :
            { 
                "filter" : 
                     { 
                         "cityName": "New Delhi"
                     },
                 "update" : 
                     { 
                         $set : 
                         { 
                             "status" : "High Population"
                         } 
                     }
            }
    },
    { 
        updateMany :
            { 
                "filter" : 
                     { 
                         "cityName": "London"
                     },
                 "update" : 
                     { 
                         $set : 
                         { 
                             "status" : "Low Population"
                         } 
                     }
            }
    },
    { 
        deleteOne :
            { 
                "filter" : 
                    { 
                        "cityName":"Mexico City"
                    } 
            }
    },
    { 
        replaceOne :
            {
                "filter" : 
                    { 
                        "cityName":"New York"
                    },
                 "replacement" : 
                    {
                        "cityId":1124,
                        "cityName":"New York",
                        "countryName":"United States",
                        "continentName":"North America",
                        "population":28
                    }
             }
    }
]);

上述bulkWrite查询将返回以下文档:

{
    "acknowledged" : true,
    "deletedCount" : 1,
    "insertedCount" : 2,
    "matchedCount" : 3,
    "upsertedCount" : 0,
    "insertedIds" : 
        {
            "0" : ObjectId("623575f89d55d4e137e477f9"),
            "1" : ObjectId("623575f89d55d4e137e477fa")
        },
    "upsertedIds" : {}
}

在这个查询中,我们执行了所有类型的写入操作,即insertOneupdateOnedeleteOnereplaceOne

首先,我们使用insertOne方法向集合中插入一个新的文档。其次,我们使用updateOne更新cityName为"New Delhi"的文档。然后,我们根据过滤器使用deleteOne方法从集合中删除文档。最后,我们使用replaceOne替换满足cityName为"New York"的完整文档。

4. 使用Java驱动

我们已经讨论了使用MongoDB shell查询执行批量写入操作。在创建批量写入操作之前,让我们先创建一个与baeldung数据库的populations集合的MongoClient连接:

MongoClient mongoClient = new MongoClient("localhost", 27017);
MongoDatabase database = mongoClient.getDatabase("baeldung");
MongoCollection<Document> collection = database.getCollection("populations");

这里,我们使用默认端口27017连接到运行中的MongoDB服务器。现在,让我们使用Java代码实现相同的操作:

List<WriteModel<Document>> writeOperations = new ArrayList<WriteModel<Document>>();
writeOperations.add(new InsertOneModel<Document>(new Document("cityId", 1128)
  .append("cityName", "Kathmandu")
  .append("countryName", "Nepal")
  .append("continentName", "Asia")
  .append("population", 12)));
writeOperations.add(new InsertOneModel<Document>(new Document("cityId", 1130)
  .append("cityName", "Mumbai")
  .append("countryName", "India")
  .append("continentName", "Asia")
  .append("population", 55)));
writeOperations.add(new UpdateOneModel<Document>(new Document("cityName", "New Delhi"),
  new Document("$set", new Document("status", "High Population"))
));
writeOperations.add(new UpdateManyModel<Document>(new Document("cityName", "London"),
  new Document("$set", new Document("status", "Low Population"))
));
writeOperations.add(new DeleteOneModel<Document>(new Document("cityName", "Mexico City")));
writeOperations.add(new ReplaceOneModel<Document>(new Document("cityId", 1124), 
  new Document("cityName", "New York").append("cityName", "United States")
    .append("continentName", "North America")
    .append("population", 28)));
BulkWriteResult bulkWriteResult = collection.bulkWrite(writeOperations);
System.out.println("bulkWriteResult:- " + bulkWriteResult);

首先,我们创建了一个包含所有不同写入操作的writeModel列表,将它们添加到单个更新列表中。在查询中,我们使用了InsertOneModelUpdateOneModelUpdateManyModelDeleteOneModelReplaceOneModel。最后,bulkWrite方法一次执行所有操作。

5. 总结

在这篇文章中,我们学习了如何使用MongoDB的不同写入操作执行批量操作。我们在单个DB查询中完成了插入、更新、删除和替换文档。此外,我们了解了initializeOrderedBulkOp在MongoDB批量更新中的用法。

首先,我们探讨了MongoDB shell查询中的批量操作用例,然后讨论了相应的Java驱动代码。

所有案例的示例可以在GitHub上找到。