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

BATCH-2462 Allow JpaItemWriter to support persist rather than merge for improved performance.

Browse Source
This commit is contained in:
Chris Cranford
2015-12-23 09:06:27 -06:00
committed by Mahmoud Ben Hassine
parent 4d3090ac02
commit 229fed4e24
2 changed files with 100 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
spring-batch-infrastructure/src/main/java/org/springframework/batch/item/database/JpaItemWriter.java
View File

@@ -50,6 +50,7 @@ public class JpaItemWriter<T> implements ItemWriter<T>, InitializingBean {
protected static final Log logger = LogFactory.getLog(JpaItemWriter.class);
private EntityManagerFactory entityManagerFactory;
private boolean usePersist = false;
/**
* Set the EntityManager to be used internally.
@@ -59,6 +60,15 @@ public class JpaItemWriter<T> implements ItemWriter<T>, InitializingBean {
public void setEntityManagerFactory(EntityManagerFactory entityManagerFactory) {
this.entityManagerFactory = entityManagerFactory;
}
/**
* Set whether the EntityManager should perform a persist instead of a merge.
*
* @param usePersist whether to use persist instead of merge.
*/
public void setUsePersist(boolean usePersist) {
this.usePersist = usePersist;
}
/**
* Check mandatory properties - there must be an entityManagerFactory.
@@ -98,16 +108,21 @@ public class JpaItemWriter<T> implements ItemWriter<T>, InitializingBean {
}
if (!items.isEmpty()) {
long mergeCount = 0;
long addedToContextCount = 0;
for (T item : items) {
if (!entityManager.contains(item)) {
entityManager.merge(item);
mergeCount++;
if(usePersist) {
entityManager.persist(item);
}
else {
entityManager.merge(item);
}
addedToContextCount++;
}
}
if (logger.isDebugEnabled()) {
logger.debug(mergeCount + " entities merged.");
logger.debug((items.size() - mergeCount) + " entities found in persistence context.");
logger.debug(addedToContextCount + " entities " + (usePersist ? " persisted." : "merged."));
logger.debug((items.size() - addedToContextCount) + " entities found in persistence context.");
}
}