1. 概述
在这个教程中,我们将探讨如何在MongoDB中执行批量更新和插入操作。此外,MongoDB提供了API调用来在一个操作中插入或检索多个文档。MongoDB使用Array
或Batch
接口,通过减少客户端和数据库之间的调用次数,极大地提高了数据库性能。
我们将通过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将在不处理列表中剩余写入操作的情况下返回。**
我们可以使用insert
、update
、replace
和remove
方法在一个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" : {}
}
在这个查询中,我们执行了所有类型的写入操作,即insertOne
、updateOne
、deleteOne
和replaceOne
。
首先,我们使用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
列表,将它们添加到单个更新列表中。在查询中,我们使用了InsertOneModel
、UpdateOneModel
、UpdateManyModel
、DeleteOneModel
和ReplaceOneModel
。最后,bulkWrite
方法一次执行所有操作。
5. 总结
在这篇文章中,我们学习了如何使用MongoDB的不同写入操作执行批量操作。我们在单个DB查询中完成了插入、更新、删除和替换文档。此外,我们了解了initializeOrderedBulkOp
在MongoDB批量更新中的用法。
首先,我们探讨了MongoDB shell查询中的批量操作用例,然后讨论了相应的Java驱动代码。
所有案例的示例可以在GitHub上找到。