diff --git a/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfiguration.java b/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfiguration.java index 4700d2008f48..be2161ad14c0 100644 --- a/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfiguration.java +++ b/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfiguration.java @@ -66,6 +66,7 @@ * @author Mahmoud Ben Hassine * @author Lars Uffmann * @author Lasse Wulff + * @author Yanming Zhou * @since 1.0.0 */ @AutoConfiguration(after = { HibernateJpaAutoConfiguration.class, TransactionAutoConfiguration.class }) @@ -140,6 +141,12 @@ protected String getTablePrefix() { return (tablePrefix != null) ? tablePrefix : super.getTablePrefix(); } + @Override + protected boolean getValidateTransactionState() { + Boolean validateTransactionState = this.properties.getJdbc().getValidateTransactionState(); + return (validateTransactionState != null) ? validateTransactionState : super.getValidateTransactionState(); + } + @Override protected Isolation getIsolationLevelForCreate() { Isolation isolation = this.properties.getJdbc().getIsolationLevelForCreate(); diff --git a/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchProperties.java b/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchProperties.java index 4f9161ac7bc8..8df4cf2055eb 100644 --- a/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchProperties.java +++ b/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/batch/BatchProperties.java @@ -27,6 +27,7 @@ * @author EddĂș MelĂ©ndez * @author Vedran Pavic * @author Mukul Kumar Chaundhyan + * @author Yanming Zhou * @since 1.2.0 */ @ConfigurationProperties("spring.batch") @@ -67,6 +68,11 @@ public static class Jdbc { private static final String DEFAULT_SCHEMA_LOCATION = "classpath:org/springframework/" + "batch/core/schema-@@platform@@.sql"; + /** + * Whether to validate the transaction state. + */ + private Boolean validateTransactionState; + /** * Transaction isolation level to use when creating job meta-data for new jobs. */ @@ -93,6 +99,14 @@ public static class Jdbc { */ private DatabaseInitializationMode initializeSchema = DatabaseInitializationMode.EMBEDDED; + public Boolean getValidateTransactionState() { + return this.validateTransactionState; + } + + public void setValidateTransactionState(Boolean validateTransactionState) { + this.validateTransactionState = validateTransactionState; + } + public Isolation getIsolationLevelForCreate() { return this.isolationLevelForCreate; } diff --git a/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfigurationTests.java b/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfigurationTests.java index 1bf6c66d2d29..9500379ed2ac 100644 --- a/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfigurationTests.java +++ b/spring-boot-project/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/batch/BatchAutoConfigurationTests.java @@ -90,6 +90,7 @@ import org.springframework.jdbc.datasource.init.DatabasePopulator; import org.springframework.orm.jpa.JpaTransactionManager; import org.springframework.transaction.PlatformTransactionManager; +import org.springframework.transaction.annotation.Isolation; import static org.assertj.core.api.Assertions.assertThat; import static org.assertj.core.api.Assertions.assertThatExceptionOfType; @@ -107,6 +108,7 @@ * @author Mahmoud Ben Hassine * @author Lars Uffmann * @author Lasse Wulff + * @author Yanming Zhou */ @ExtendWith(OutputCaptureExtension.class) class BatchAutoConfigurationTests { @@ -520,6 +522,18 @@ void defaultExecutionContextSerializerIsUsed() { }); } + @Test + void customJdbcPropertiesIsUsed() { + this.contextRunner.withUserConfiguration(EmbeddedDataSourceConfiguration.class) + .withPropertyValues("spring.batch.jdbc.validate-transaction-state:false", + "spring.batch.jdbc.isolation-level-for-create:READ_COMMITTED") + .run((context) -> { + SpringBootBatchConfiguration configuration = context.getBean(SpringBootBatchConfiguration.class); + assertThat(configuration.getValidateTransactionState()).isEqualTo(false); + assertThat(configuration.getIsolationLevelForCreate()).isEqualTo(Isolation.READ_COMMITTED); + }); + } + private JobLauncherApplicationRunner createInstance(String... registeredJobNames) { JobLauncherApplicationRunner runner = new JobLauncherApplicationRunner(mock(JobLauncher.class), mock(JobExplorer.class), mock(JobRepository.class));