spring-boot 使用 spring-data-jpa多数据源配置

application.yml：

server:
  port: 8888
spring:
  application:
    name: test
  aop:
    auto: true
    proxy-target-class: false
  jpa:
    generate-ddl: false
    show-sql: true
    hibernate:
      ddl-auto: none    
  datasource:
    primary:
      url: jdbc:oracle:thin:@192.168.xx.xxx:1521:orcl
      username: xxx
      password: xxx
      driver-class-name: oracle.jdbc.OracleDriver
    test:
      url: jdbc:mysql://192.168.xx.xxx:3306/security
      username: xxx
      password: xxx
      driver-class-name: com.mysql.jdbc.Driver

• 创建configuration包,创建以下文件：

• DataSourceConfig.java

 @Configuration
public class DataSourceConfig {

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

    @Bean(name = "testDataSource")
    @Qualifier("testDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.test")
    public DataSource kakou4DataSource() {
        return DataSourceBuilder.create().build();
    }
}

• PrimaryConfig.java

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(entityManagerFactoryRef = "entityManagerFactoryPrimary", transactionManagerRef = "transactionManagerPrimary", basePackages = {
        "com.test.data.openapi.repository.primary" }) // Repository位置
public class PrimaryConfig {

    @Autowired
    @Qualifier("primaryDataSource")
    private DataSource primaryDataSource;

    @Primary
    @Bean(name = "entityManagerPrimary")
    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
        return entityManagerFactoryPrimary(builder).getObject().createEntityManager();
    }

    @Primary
    @Bean(name = "entityManagerFactoryPrimary")
    public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary(EntityManagerFactoryBuilder builder) {
        return builder.dataSource(primaryDataSource).properties(getVendorProperties(primaryDataSource))
                .packages("com.test.data.openapi.domain.primary")//实体类位置
                .persistenceUnit("primaryPersistenceUnit").build();
    }

    @Autowired
    private JpaProperties jpaProperties;

    private Map<String, String> getVendorProperties(DataSource dataSource) {
        return jpaProperties.getHibernateProperties(dataSource);
    }

    @Primary
    @Bean(name = "transactionManagerPrimary")
    public PlatformTransactionManager transactionManagerPrimary(EntityManagerFactoryBuilder builder) {
        return new JpaTransactionManager(entityManagerFactoryPrimary(builder).getObject());
    }

}

• TestConfig.java

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(entityManagerFactoryRef = "testEntityManagerFactory", transactionManagerRef = "testTransactionManager", basePackages = {
        "com.test.data.openapi.repository.test" }) // Repository位置
public class TestConfig {

    @Autowired
    @Qualifier("testDataSource")
    private DataSource testDataSource;

    @Bean(name = "testEntityManager")
    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
        return testEntityManagerFactory(builder).getObject().createEntityManager();
    }

    @Bean(name = "testEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean testEntityManagerFactory(EntityManagerFactoryBuilder builder) {
        return builder.dataSource(testDataSource).properties(getVendorProperties(testDataSource))
                .packages("com.test.data.openapi.domain.test") // 实体类位置
                .persistenceUnit("testPersistenceUnit").build();
    }

    @Autowired
    private JpaProperties jpaProperties;

    private Map<String, String> getVendorProperties(DataSource dataSource) {
        return jpaProperties.getHibernateProperties(dataSource);
    }

    @Bean(name = "testTransactionManager")
    public PlatformTransactionManager testTransactionManager(EntityManagerFactoryBuilder builder) {
        return new JpaTransactionManager(testEntityManagerFactory(builder).getObject());
    }

}

• 测试

@RestController
@RequestMapping("/api/test")
public class GjfxController {

    @Resource
    @Qualifier("testClientRepository")
    KakouClientRepository kakouClientRepository;
    @Resource
    @Qualifier("primaryClientRepository")
    PrimaryClientRepository primaryClientRepository;

    @GetMapping("/test")
    public String getP() {
        Map<String, Object> map = new HashMap<>();
        map.put("testClientRepository", testClientRepository.findAll());
        map.put("primaryClientRepository", primaryClientRepository.findAll());
        return JSONObject.toJSONString(map);
    }
}

其中testRepository会查出mysql中数据，primaryRepository会查出oracle中数据