Spring Boot多数据源配置是否存在更简便的实现方式？

问题描述 #

我正在学习Spring Boot多数据源的使用，但目前看到的示例配置都十分复杂。原本Spring Boot的自动配置特性不再生效，需要手动完成所有配置：定义存放Entity和Repository的包路径，手动创建TransactionManager与EntityManagerFactoryBuilder。

现在我必须持续更新以下配置中的包路径：

• @EnableJpaRepositories中的basePackages
• entityManagerFactoryBuilder返回方法中的packages()

且当我调整类结构时，这些路径无法自动重构。

是否存在更简便的实现方式？比如定义不同的DataSource后，通过注解指定Entity的存储位置；或是手动为每个DataSource配置EntityManager，再通过注解指定Entity使用对应的EntityManager。

另外，能否避免单独指定Entity与Repository的包路径？当前配置的冗余开销过大。

当前使用的配置代码 #

Schema1Config.java #

@Configuration
@EnableJpaRepositories(
  basePackages      = "com.ivoronline.springboot_db_datasource_diffentitiesdiffschemas.schema1.repository",
  entityManagerFactoryRef = "schema1EntityManagerFactoryBean",
  transactionManagerRef   = "schema1TransactionManager"
)
public class Schema1Config {

  //=========================================================================================================
  // DATA SOURCE PROPERTIES
  //=========================================================================================================
  @Bean
  @ConfigurationProperties("schema1.spring.datasource")
  public DataSourceProperties schema1DataSourceProperties() {
    return new DataSourceProperties();
  }
  
  //=========================================================================================================
  // DATA SOURCE
  //=========================================================================================================
  @Bean
  @Primary
  public DataSource schema1DataSource() {
    return schema1DataSourceProperties().initializeDataSourceBuilder().build();
  }
  
  //=========================================================================================================
  // ENTITY MANAGER FACTORY BEAN
  //=========================================================================================================
  @Bean
  LocalContainerEntityManagerFactoryBean schema1EntityManagerFactoryBean (
    EntityManagerFactoryBuilder entityManagerFactoryBuilder,
    DataSource                  dataSource
  ) {
    return entityManagerFactoryBuilder
          .dataSource(dataSource)
          .packages("com.ivoronline.springboot_db_datasource_diffentitiesdiffschemas.schema1.entity")
          .build();
  }

  //=========================================================================================================
  // SCHEMA 1 TRANSACTION MANAGER
  //=========================================================================================================
  @Bean
  PlatformTransactionManager schema1TransactionManager(@Qualifier("schema1EntityManagerFactoryBean") LocalContainerEntityManagerFactoryBean emfb) {
    return new JpaTransactionManager(emfb.getObject());
  }
}

简化实现方案 #

1. 用类引用替代硬编码包路径 #

解决包路径无法自动重构的问题，最直接的方式是用类对象代替字符串包路径，IDE会在重构时自动更新关联：

修改@EnableJpaRepositories配置

将basePackages替换为basePackageClasses，直接指定对应包下的任意类（比如Repository接口）：

@EnableJpaRepositories(
  basePackageClasses = Schema1Repository.class, // 替换为你实际的Repository类
  entityManagerFactoryRef = "schema1EntityManagerFactoryBean",
  transactionManagerRef   = "schema1TransactionManager"
)

修改EntityManagerFactory配置

将packages()的字符串参数换成Entity类对象：

return entityManagerFactoryBuilder
      .dataSource(dataSource)
      .packages(Schema1Entity.class) // 替换为你实际的Entity类
      .build();

这样调整类结构或包名时，IDE会自动同步更新配置，无需手动修改字符串路径。

2. 基于自定义注解的数据源绑定（摆脱包路径限制） #

如果不想按包划分Entity和Repository，可以通过自定义注解来标记不同数据源对应的组件，实现更灵活的绑定：

步骤1：创建自定义数据源注解

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface Schema1DataSource {}

步骤2：给Entity和Repository标记注解

• Entity类上添加注解：

@Schema1DataSource
@Entity
public class Schema1Entity {
    // 实体字段
}

• Repository接口上添加注解：

@Schema1DataSource
@Repository
public interface Schema1Repository extends JpaRepository<Schema1Entity, Long> {}

步骤3：修改配置类过滤注解组件

• 调整@EnableJpaRepositories，通过includeFilters过滤带注解的Repository：

@EnableJpaRepositories(
  includeFilters = @ComponentScan.Filter(type = FilterType.ANNOTATION, classes = Schema1DataSource.class),
  entityManagerFactoryRef = "schema1EntityManagerFactoryBean",
  transactionManagerRef   = "schema1TransactionManager"
)

• 调整EntityManagerFactory配置，过滤带注解的Entity：

return entityManagerFactoryBuilder
      .dataSource(dataSource)
      .persistenceUnitPostProcessors(pup -> {
          pup.addManagedClassFilter(clazz -> clazz.isAnnotationPresent(Schema1DataSource.class));
      })
      .build();

这种方式完全摆脱包路径限制，只要给组件标记对应注解即可，重构时无需修改配置。

3. 复用Spring Boot自动配置简化代码 #

可以复用Spring Boot的JPA自动配置逻辑，减少手动配置的冗余：

通过JpaProperties复用自动配置的JPA通用属性，无需手动构建EntityManagerFactory的所有参数：

@Autowired
private JpaProperties jpaProperties;

@Bean
LocalContainerEntityManagerFactoryBean schema1EntityManagerFactoryBean (
  EntityManagerFactoryBuilder entityManagerFactoryBuilder,
  DataSource                  dataSource
) {
    Map<String, Object> properties = new HashMap<>(jpaProperties.getProperties());
    return entityManagerFactoryBuilder
          .dataSource(dataSource)
          .packages(Schema1Entity.class)
          .properties(properties)
          .build();
}

这样可以复用application.yml中配置的JPA通用属性，避免重复配置。

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