Spring Boot 多数据源配置之 MyBatis多数据源配置

前置准备

1、首先在application配置文件中配置多个数据源，这里我用的是yaml

spring:
  datasource:
    test:
      driver-class-name: com.mysql.jdbc.Driver
      username: 
      password:
      type: com.alibaba.druid.pool.DruidDataSource
      url: 
    pro:
      driver-class-name: com.mysql.jdbc.Driver
      username: 
      password:
      type: com.alibaba.druid.pool.DruidDataSource
      url: 
mybatis:
  type-aliases-package: app.entity
  mapper-locations: classpath:mapper/*.xml

配置SqlSessionFactory和SqlSessionTemplate

1、配置DataSource

@Configuration
public class DataSourceConfig {

//    设定一个主要的配置文件
    @Primary
//  把DataSourceProperties注入到Spring容器中
    @Bean(name = "testDataProperties")
//    识别配置文件前缀
    @ConfigurationProperties(prefix = "spring.datasource.test")
    public DataSourceProperties testDataSourceProperties(){
        return new DataSourceProperties();
    }
    
    
//  把DataSourceProperties注入到Spring容器中
    @Bean(name = "proDataProperties")
//    识别配置文件前缀
    @ConfigurationProperties(prefix = "spring.datasource.pro")
    public DataSourceProperties proDataSourceProperties(){
        return new DataSourceProperties();
    }
    
//    设定主要的数据源
    @Primary
//    把DataSource注入到Spring容器中，参数：DataSourceProperties，根据@Bean中的name进行区分
    @Bean(name = "testDataSource")
    public DataSource testDataSource(@Qualifier("testDataProperties") DataSourceProperties dataSourceProperties){
        return dataSourceProperties.initializeDataSourceBuilder().build();
    }


//    把DataSource注入到Spring容器中，参数：DataSourceProperties，根据@Bean中的name进行区分
    @Bean(name = "proDataSource")
    public DataSource proDataSource(@Qualifier("proDataProperties") DataSourceProperties dataSourceProperties){
        return dataSourceProperties.initializeDataSourceBuilder().build();
    }

}

2、配置数据源1

@Configuration
//设置扫描的mapper在哪个包下，此包下的mapper都使用该配置类的DataSource，配置SqlSessionFactory
@MapperScan(basePackages = "app.pro.dao",sqlSessionFactoryRef = "proSqlSessionFactory")
public class ProDataSourceConfig {
    @Primary
//    配置SqlSessionFactory,使用相应的DataSource
    @Bean("proSqlSessionFactory")
    public SqlSessionFactory proSqlSessionFactory(@Qualifier("proDataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/*.xml"));
        return sqlSessionFactoryBean.getObject();
    }

    @Primary
//    配置SqlSessionTemplate，使用相应SqlSessionFactory
    @Bean("proSqlSessionTemplate")
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("proSqlSessionFactory") SqlSessionFactory sqlSessionFactory){
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

3、配置数据源2

@Configuration
//设置扫描的mapper在哪个包下，此包下的mapper都使用该配置类的DataSource，配置SqlSessionFactory
@MapperScan(basePackages = "app.dao",sqlSessionFactoryRef = "testSqlSessionFactory")
public class TestDataSourceConfig {
    @Primary
//    配置SqlSessionFactory,使用相应的DataSource
    @Bean("testSqlSessionFactory")
    public SqlSessionFactory testSqlSessionFactory(@Qualifier("testDataSource")DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/*.xml"));
        return sqlSessionFactoryBean.getObject();
    }

    @Primary
//    配置SqlSessionTemplate，使用相应SqlSessionFactory
    @Bean("testSqlSessionTemplate")
    public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("testSqlSessionFactory") SqlSessionFactory sqlSessionFactory){
        return new SqlSessionTemplate(sqlSessionFactory);
    }

}

使用

可以看到这两个mapper是要在不同的包下的，因为之前的MapperScan注解中是用包来区分不同数据源的mapper（mapper.xml是否在一个文件夹下取决于之前配置数据源时，sqlSessionFactoryBean.setMapperLocations()是如何书写的，多个数据源都是设置相同的，则放在一个文件夹下即可，如果不一样，请按需安放）

    @Resource
    private AccessTokenDao accessTokenDao;

    @Resource
    private ProAccessTokenDao proAccessTokenDao;

这样就可以使用不同的数据源操作不同的数据库了