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

Add usePersist chain method to JpaItemWriterBuilder

Browse Source
This commit is contained in:
Ruslan Mustaev
2020-02-02 20:56:52 +03:00
committed by Mahmoud Ben Hassine
parent e5cafe4625
commit bd016a9444
2 changed files with 35 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
spring-batch-infrastructure/src/main/java/org/springframework/batch/item/database/builder/JpaItemWriterBuilder.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2018 the original author or authors.
* Copyright 2018-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.
@@ -30,6 +30,7 @@ import org.springframework.util.Assert;
public class JpaItemWriterBuilder<T> {
private EntityManagerFactory entityManagerFactory;
private boolean usePersist = false;
/**
* The JPA {@link EntityManagerFactory} to obtain an entity manager from. Required.
@@ -44,6 +45,19 @@ public class JpaItemWriterBuilder<T> {
return this;
}
/**
* Set whether the entity manager should perform a persist instead of a merge.
*
* @param usePersist defaults to false
* @return this instance for method chaining
* @see JpaItemWriter#setUsePersist(boolean)
*/
public JpaItemWriterBuilder<T> usePersist(boolean usePersist) {
this.usePersist = usePersist;
return this;
}
/**
* Returns a fully built {@link JpaItemWriter}.
*
@@ -55,6 +69,7 @@ public class JpaItemWriterBuilder<T> {
JpaItemWriter<T> writer = new JpaItemWriter<>();
writer.setEntityManagerFactory(this.entityManagerFactory);
writer.setUsePersist(this.usePersist);
return writer;
}