Spring Boot,Spring Data JPA多数据源支持

1 配置文件

 

wisely.primary.datasource.driverClassName=oracle.jdbc.OracleDriver
wisely.primary.datasource.url=jdbc\:oracle\:thin\:@192.168.1.103\:1521\:xe
wisely.primary.datasource.username=gis
wisely.primary.datasource.password=gis

wisely.secondary.datasource.driverClassName=oracle.jdbc.OracleDriver
wisely.secondary.datasource.url=jdbc\:oracle\:thin\:@192.168.1.103\:1522\:xe
wisely.secondary.datasource.username=gis
wisely.secondary.datasource.password=gis

 

 

 

2 datasource配置

 

第一个数据源

 

@Configuration
public class DataSourcePrimaryConfig {

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

}

 

 

 

第二个数据源

 

@Configuration
public class DataSourceSecondaryConfig {
    @Bean(name = "secondaryDS") @Qualifier("secondaryDS")
    @ConfigurationProperties(prefix="wisely.secondary.datasource")
    public DataSource secondaryDataSource(){
        return DataSourceBuilder.create().build();
    }
}

 

 

 

3 实体管理器及事务管理器配置

 

第一个数据源

 

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(entityManagerFactoryRef="entityManagerFactoryPrimary",transactionManagerRef="transactionManagerPrimary",basePackages= { "com.wisely.demo.dao.one" })//设置dao（repo）所在位置
public class RepositoryPrimaryConfig {
      @Autowired
      private JpaProperties jpaProperties;

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

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

      @Bean(name = "entityManagerFactoryPrimary")
      @Primary
      public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary (EntityManagerFactoryBuilder builder) {
          return builder
                  .dataSource(primaryDS)
                  .properties(getVendorProperties(primaryDS))
                  .packages("com.wisely.demo.domain.one") //设置实体类所在位置
                  .persistenceUnit("primaryPersistenceUnit")
                  .build();
      }

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

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

}

 

 

 

第二个数据源

 

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(entityManagerFactoryRef="entityManagerFactorySecondary",transactionManagerRef="transactionManagerSecondary",basePackages= { "com.wisely.demo.dao.two" })
public class RepositorySecondaryConfig {
      @Autowired
      private JpaProperties jpaProperties;

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

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

      @Bean(name = "entityManagerFactorySecondary")
      public LocalContainerEntityManagerFactoryBean entityManagerFactorySecondary (EntityManagerFactoryBuilder builder) {
          return builder
                  .dataSource(secondaryDS)
                  .properties(getVendorProperties(secondaryDS))
                  .packages("com.wisely.demo.domain.two")
                  .persistenceUnit("secondaryPersistenceUnit")
                  .build();
      }

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

      @Bean(name = "transactionManagerSecondary")
      PlatformTransactionManager transactionManagerSecondary(EntityManagerFactoryBuilder builder) {
          return new JpaTransactionManager(entityManagerFactorySecondary(builder).getObject());
      }

}

 

 

 

4 使用

 

此时来自不同数据库的dao（repo）可以任意在其它的bean里注入

 

 

 

@Controller
public class TestController {
    @Autowired
    SysRoleRepo1 sysRoleRepo1;
    @Autowired
    SysRoleRepo2 sysRoleRepo2;
    @RequestMapping("/test")
    public @ResponseBody String test(){
        System.out.println(Lists.newArrayList(sysRoleRepo1.findAll()).size());
        System.out.println(Lists.newArrayList(sysRoleRepo2.findAll()).size());
        return "ok";
    }
}