BATCH-2686: Make BatchIntegrationConfiguration implement ApplicationContextAware

Author: fmbenhassine

spring-batch-integration/src/main/java/org/springframework/batch/integration/config/annotation/BatchIntegrationConfiguration.java

@@ -18,8 +18,11 @@
 import org.springframework.batch.core.repository.JobRepository;
 import org.springframework.batch.integration.chunk.RemoteChunkingMasterStepBuilderFactory;
 import org.springframework.batch.integration.chunk.RemoteChunkingWorkerBuilder;
+import org.springframework.beans.BeansException;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.support.BeanDefinitionRegistry;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.transaction.PlatformTransactionManager;
@@ -31,7 +34,7 @@
  * @author Mahmoud Ben Hassine
  */
 @Configuration
-public class BatchIntegrationConfiguration {
+public class BatchIntegrationConfiguration implements ApplicationContextAware {
 
 	private JobRepository jobRepository;
 
@@ -42,11 +45,9 @@ public class BatchIntegrationConfiguration {
 	@Autowired
 	public BatchIntegrationConfiguration(
 			JobRepository jobRepository,
-			PlatformTransactionManager transactionManager,
-			BeanDefinitionRegistry beanDefinitionRegistry) {
+			PlatformTransactionManager transactionManager) {
 		this.jobRepository = jobRepository;
 		this.transactionManager = transactionManager;
-		this.beanDefinitionRegistry = beanDefinitionRegistry;
 	}
 
 	@Bean
@@ -59,4 +60,9 @@ public <I,O> RemoteChunkingWorkerBuilder<I, O> remoteChunkingWorkerBuilder() {
 		return new RemoteChunkingWorkerBuilder<>();
 	}
 
+	@Override
+	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
+		this.beanDefinitionRegistry = (BeanDefinitionRegistry) applicationContext;
+	}
+
 }