[英]Bulk Upsert with MongoDB Java 3.0 Driver
在早期版本的MongoDB Java驅動程序中,要運行查詢並對結果執行無序批量upsert,我們所做的就是:
BulkWriteOperation bulk = dbCollection.initializeUnorderedBulkOperation();
bulk.find(searchQuery).upsert().update(new BasicDBObject("$set", getDbObjectModel()));
但是在版本3中,隨着Bson Document支持和MongoCollection.bulkWrite()方法的引入,如何才能做到這一點?
我試過這個:
List<WriteModel<Document>> documentList = new ArrayList<>();
collection.bulkWrite(documentList, new BulkWriteOptions().ordered(false));
但是,我需要upsert功能。
謝謝。
您仍然可以使用所有功能,只是BulkWrites現在有不同的語法:
MongoCollection<Document> collection = db.getCollection("sample");
List<WriteModel<Document>> updates = Arrays.<WriteModel<Document>>asList(
new UpdateOneModel<Document>(
new Document(), // find part
new Document("$set",1), // update part
new UpdateOptions().upsert(true) // options like upsert
)
);
BulkWriteResult bulkWriteResult = collection.bulkWrite(updates);
因此,您可以使用UpdateOneModel
(如果需要,可以使用UpdateOneModel
),並將UpdateOptions
設置為構造函數的第三個參數。
需要一些習慣,但它基本上只是構建“列表”,其語法與其他地方相同。 我猜這是改變的主要原因。
以下是使用最新API的示例..
for (Long entityId : entityIDs) {
//Finder doc
Document filterDocument = new Document();
filterDocument.append("_id", entityId);
//Update doc
Document updateDocument = new Document();
Document setDocument = new Document();
setDocument.append("name", "xyz");
setDocument.append("role", "abc");
updateDocument.append("$set", setDocument);
//Update option
UpdateOptions updateOptions = new UpdateOptions();
updateOptions.upsert(true); //if true, will create a new doc in case of unmatched find
updateOptions.bypassDocumentValidation(true); //set true/false
//Prepare list of Updates
updateDocuments.add(
new UpdateOneModel<Document>(
filterDocument,
updateDocument,
updateOptions));
}
//Bulk write options
BulkWriteOptions bulkWriteOptions = new BulkWriteOptions();
bulkWriteOptions.ordered(false);
bulkWriteOptions.bypassDocumentValidation(true);
MongoCollection<Document> mongoCollection = mongoDB.getCollection("myCollection");
BulkWriteResult bulkWriteResult = null;
try {
//Perform bulk update
bulkWriteResult = mongoCollection.bulkWrite(updateDocuments,
bulkWriteOptions);
} catch (BulkWriteException e) {
//Handle bulkwrite exception
List<BulkWriteError> bulkWriteErrors = e.getWriteErrors();
for (BulkWriteError bulkWriteError : bulkWriteErrors) {
int failedIndex = bulkWriteError.getIndex();
Long failedEntityId = entityIDs.get(failedIndex);
System.out.println("Failed record: " + failedEntityId);
//handle rollback
}
}
int rowsUpdated = bulkWriteResult.getModifiedCount();
詳細信息: http : //ashutosh-srivastav-mongodb.blogspot.in/2017/09/mongodb-bulkwrite-java-api.html
如果你想要一些findAndModifyElseCreate(); 這意味着如果文檔存在然后更新它,則創建它並插入數據然后請關注此事。
BasicDBObject insertInCaseDocumentNotFound = new BasicDBObject();
insertInCaseDocumentNotFound.put("field1", "value1");
insertInCaseDocumentNotFound.put("date", new Date());
MongoCollection<BasicDBObject> table = db.getCollection("collectionName",BasicDBObject.class);
BasicDBObject updateObject = new BasicDBObject();
updateObject.append("$setOnInsert", new BasicDBObject());
updateObject.append("$set", new BasicDBObject("date",new Date());
List<WriteModel<BasicDBObject>> updates = Arrays.<WriteModel<BasicDBObject>> asList(
new UpdateOneModel<BasicDBObject>(new BasicDBObject("search_name", alert.getString("search_name")), // query for which we need to apply
updateObject, // update the document in case it is found
new UpdateOptions().upsert(true) // upsert to insert new data in case document is not found
));
table.bulkWrite(updates);
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.