Spring Boot Spring Batch多数据源多Schema：如何分别指定默认Schema

针对你在Spring Boot + Spring Batch场景下的多数据源Schema配置问题，直接通过全局JPA配置只能覆盖主数据源，要给第二个数据源单独指定默认Schema，最佳方案是为每个数据源创建独立的EntityManagerFactory配置，在各自的配置中指定Hibernate的默认Schema属性，完全替代手动执行alter session的方式。

以下是具体实现步骤：

1. 先在配置文件中定义两个数据源的连接信息 #

假设你用application.yml，可以这样配置：

# 主数据源配置
spring.datasource.primary:
  url: jdbc:oracle:thin:@//localhost:1521/ORCLPDB1
  username: primary_user
  password: primary_pass
  driver-class-name: oracle.jdbc.OracleDriver

# 第二个数据源配置
spring.datasource.secondary:
  url: jdbc:oracle:thin:@//localhost:1521/ORCLPDB1
  username: secondary_user
  password: secondary_pass
  driver-class-name: oracle.jdbc.OracleDriver

2. 为主数据源创建专属的JPA配置类 #

这个类会定义主数据源的DataSource、EntityManagerFactory和TransactionManager，并指定它的默认Schema：

@Configuration
@EnableJpaRepositories(
    basePackages = "com.yourproject.primary.repositories", // 主数据源的Repository包路径
    entityManagerFactoryRef = "primaryEntityManagerFactory",
    transactionManagerRef = "primaryTransactionManager"
)
public class PrimaryDataSourceConfig {

    @Primary
    @Bean(name = "primaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.primary")
    public DataSource primaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Primary
    @Bean(name = "primaryEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean primaryEntityManagerFactory(
            EntityManagerFactoryBuilder builder,
            @Qualifier("primaryDataSource") DataSource dataSource) {
        Map<String, Object> hibernateProps = new HashMap<>();
        hibernateProps.put("hibernate.default_schema", "PRIMARY_SCHEMA"); // 主数据源的默认Schema
        // 可以添加其他Hibernate配置，比如show_sql、format_sql等
        hibernateProps.put("hibernate.show_sql", true);

        return builder
                .dataSource(dataSource)
                .packages("com.yourproject.primary.entities") // 主数据源的实体类包路径
                .persistenceUnit("primaryPersistenceUnit")
                .properties(hibernateProps)
                .build();
    }

    @Primary
    @Bean(name = "primaryTransactionManager")
    public PlatformTransactionManager primaryTransactionManager(
            @Qualifier("primaryEntityManagerFactory") EntityManagerFactory entityManagerFactory) {
        return new JpaTransactionManager(entityManagerFactory);
    }
}

3. 为第二个数据源创建对应的JPA配置类 #

同样的逻辑，为第二个数据源独立配置，指定它的默认Schema：

@Configuration
@EnableJpaRepositories(
    basePackages = "com.yourproject.secondary.repositories", // 第二个数据源的Repository包路径
    entityManagerFactoryRef = "secondaryEntityManagerFactory",
    transactionManagerRef = "secondaryTransactionManager"
)
public class SecondaryDataSourceConfig {

    @Bean(name = "secondaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondaryEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean secondaryEntityManagerFactory(
            EntityManagerFactoryBuilder builder,
            @Qualifier("secondaryDataSource") DataSource dataSource) {
        Map<String, Object> hibernateProps = new HashMap<>();
        hibernateProps.put("hibernate.default_schema", "SECONDARY_SCHEMA"); // 第二个数据源的默认Schema
        hibernateProps.put("hibernate.show_sql", true);

        return builder
                .dataSource(dataSource)
                .packages("com.yourproject.secondary.entities") // 第二个数据源的实体类包路径
                .persistenceUnit("secondaryPersistenceUnit")
                .properties(hibernateProps)
                .build();
    }

    @Bean(name = "secondaryTransactionManager")
    public PlatformTransactionManager secondaryTransactionManager(
            @Qualifier("secondaryEntityManagerFactory") EntityManagerFactory entityManagerFactory) {
        return new JpaTransactionManager(entityManagerFactory);
    }
}

4. Spring Batch作业中的适配 #

在你的Spring Batch Step中，使用对应数据源的JdbcTemplate、EntityManager或者ItemReader/Writer时，要注入对应的Bean：
比如，使用JdbcCursorItemReader读取第二个数据源的数据：

@Bean
public JdbcCursorItemReader<SecondaryEntity> secondaryItemReader(
        @Qualifier("secondaryDataSource") DataSource dataSource) {
    return new JdbcCursorItemReaderBuilder<SecondaryEntity>()
            .dataSource(dataSource)
            .sql("SELECT id, name FROM entity_table") // Hibernate会自动加上SECONDARY_SCHEMA前缀
            .rowMapper(new BeanPropertyRowMapper<>(SecondaryEntity.class))
            .build();
}

为什么这个方案比alter session更好？ #

• 不需要手动管理会话切换，避免了线程安全问题（尤其是在多线程的Batch作业中）
• Hibernate会自动在生成的SQL语句前加上指定的Schema前缀，代码更简洁
• 配置和业务逻辑分离，后续修改Schema只需要调整配置类中的属性，无需修改业务代码

内容的提问来源于stack exchange，提问作者poorna chandra