Utilisation de plusieurs sources de données dans Spring Batch

Vous devez fournir votre propre BatchConfigurer. Spring ne veut pas prendre cette décision pour vous

@Configuration
@EnableBatchProcessing
public class BatchConfig {

     @Bean
      BatchConfigurer configurer(@Qualifier("batchDataSource") DataSource dataSource){
        return new DefaultBatchConfigurer(dataSource);
      }

...

Si je puis Ajouter à la question ci-dessus, les implications d'avoir 2 contextes de transaction un pour chaque DS. Comment intégrer la transaction XA avec Batch step comme nous aurions besoin pour assurer la gestion TXN au niveau de l'étape? L'exigence est comme dans une étape de lot, nous avons besoin de ce qui suit.

1. Lire À partir de DS 1 -- jpaItemReader
2. écrire dans DS2-JPAItemwriter
3. Lire À partir de DS2-JPAItemreader
4. écrire dans Ds1-JPAItemwriter
5. Commit tous les TXN Étape compléter.

Tout d'abord, créez un BatchConfigurer personnalisé

@Configuration
@Component
public class TwoDataSourcesBatchConfigurer implements BatchConfigurer {

    @Autowired
    @Qualifier("dataSource1")
    DataSource dataSource;

    @Override
    public JobExplorer getJobExplorer() throws Exception {
        ...
    }

    @Override
    public JobLauncher getJobLauncher() throws Exception {
        ...
    }

    @Override
    public JobRepository getJobRepository() throws Exception {
        JobRepositoryFactoryBean factory = new JobRepositoryFactoryBean();
        // use the autowired data source
        factory.setDataSource(dataSource);
        factory.setTransactionManager(getTransactionManager());
        factory.afterPropertiesSet();
        return factory.getObject();
    }

    @Override
    public PlatformTransactionManager getTransactionManager() throws Exception                      {
        ...
    }

}

Puis,

@Configuration
@EnableBatchProcessing
@ComponentScan("package")
public class JobConfig {
    // define job, step, ...
}

Je voudrais fournir une solution ici, qui est très similaire à celle répondue par @vanarchi, mais j'ai réussi à mettre toutes les configurations nécessaires dans une classe.

Par souci d'exhaustivité, la solution ici suppose que la source de données principale est hsql.

@Configuration
@EnableBatchProcessing
public class BatchConfiguration extends DefaultBatchConfigurer {

@Bean
@Primary
public DataSource batchDataSource() {

    // no need shutdown, EmbeddedDatabaseFactoryBean will take care of this
    EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder();
    EmbeddedDatabase embeddedDatabase = builder
            .addScript("classpath:org/springframework/batch/core/schema-drop-hsqldb.sql")
            .addScript("classpath:org/springframework/batch/core/schema-hsqldb.sql")
            .setType(EmbeddedDatabaseType.HSQL) //.H2 or .DERBY
            .build();
    return embeddedDatabase;
}

@Override
protected JobRepository createJobRepository() throws Exception {
    JobRepositoryFactoryBean factory = new JobRepositoryFactoryBean();
    factory.setDataSource(batchDataSource());
    factory.setTransactionManager(transactionManager());
    factory.afterPropertiesSet();

    return (JobRepository) factory.getObject();
}

private ResourcelessTransactionManager transactionManager() {
    return new ResourcelessTransactionManager();
}

//NOTE: the code below is just to provide developer an easy way to access the in-momery hsql datasource, as we configured it to the primary datasource to store batch job related data. Default username : sa, password : ''
@PostConstruct
public void getDbManager(){
    DatabaseManagerSwing.main(
            new String[] { "--url", "jdbc:hsqldb:mem:testdb", "--user", "sa", "--password", ""});
}

}

Trois points clés dans cette solution:

1. Cette classe est annotée avec @EnableBatchProcessing et @Configuration, ainsi qu'une extension à partir de DefaultBatchConfigurer. En faisant cela, nous demandons à spring-batch d'utiliser notre configurateur par lots personnalisé lorsque AbstractBatchConfiguration tente de rechercher BatchConfigurer;
2. annotez le bean batchDataSource en tant que @Primary, qui demande à spring-batch d'utiliser cette source de données comme source de données pour stocker les 9 tables liées au travail.
3. Override protected JobRepository createJobRepository() throws Exception méthode, qui permet au bean jobRepository d'utiliser la source de données principale, ainsi que d'utiliser une instance transactionManager différente de la ou des autres sources de données.

La solution la plus simple consiste à étendre le DefaultBatchConfigurer et autowire votre source de données via un qualificateur:

@Component
public class MyBatchConfigurer extends DefaultBatchConfigurer {

    /**
     * Initialize the BatchConfigurer to use the datasource of your choosing
     * @param firstDataSource
     */
    @Autowired
    public MyBatchConfigurer(@Qualifier("firstDataSource") DataSource firstDataSource) {
        super(firstDataSource);
    }
}

Note latérale (car cela concerne également l'utilisation de plusieurs sources de données): si vous utilisez autoconfig pour exécuter des scripts d'initialisation de données, vous remarquerez peut-être qu'il ne s'initialise pas sur la source de données que vous attendez. Pour ce problème, jetez un oeil à ceci: https://github.com/spring-projects/spring-boot/issues/9528