Spring/spring-batch
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Implemented BulkOperations API in MongoItemWriter

Browse Source
This commit is contained in:
Parikshit Dutta
2020-05-28 21:55:26 +05:30
committed by Mahmoud Ben Hassine
parent 43c40b12ef
commit 629497d5bb
2 changed files with 83 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
spring-batch-infrastructure/src/main/java/org/springframework/batch/item/data/MongoItemWriter.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2012-2017 the original author or authors.
* Copyright 2012-2020 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -19,12 +19,22 @@ package org.springframework.batch.item.data;
import java.util.ArrayList;
import java.util.List;
import org.bson.Document;
import org.bson.types.ObjectId;
import org.springframework.batch.item.ItemWriter;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.data.mongodb.core.BulkOperations;
import org.springframework.data.mongodb.core.BulkOperations.BulkMode;
import org.springframework.data.mongodb.core.FindAndReplaceOptions;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.convert.MongoConverter;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.transaction.support.TransactionSynchronizationAdapter;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import org.springframework.util.Assert;
import org.springframework.util.ClassUtils;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
@@ -42,6 +52,7 @@ import org.springframework.util.StringUtils;
* </p>
*
* @author Michael Minella
* @author Parikshit Dutta
*
*/
public class MongoItemWriter<T> implements ItemWriter<T>, InitializingBean {
@@ -133,16 +144,28 @@ public class MongoItemWriter<T> implements ItemWriter<T>, InitializingBean {
}
}
else {
BulkOperations bulkOperations = null;
if(StringUtils.hasText(collection)) {
for (Object object : items) {
template.save(object, collection);
}
bulkOperations = template.bulkOps(BulkMode.ORDERED, collection);
}
else {
for (Object object : items) {
template.save(object);
}
bulkOperations = template.bulkOps(BulkMode.ORDERED, ClassUtils.getUserClass(items.get(0)));
}
for (Object object : items) {
Document document = new Document();
MongoConverter mongoConverter = template.getConverter();
mongoConverter.write(object, document);
Query query = new Query();
query.addCriteria(Criteria.where("_id").is((document.get("_id") != null)
? document.get("_id") : new ObjectId()));
bulkOperations.replaceOne(query, document, new FindAndReplaceOptions().upsert());
}
bulkOperations.execute();
}
}
}