SpringBoot+Mybatis多数据源配置和切换

前言

在项目开发中，经常会涉及到一个应用程序调用多个数据的情况。今天介绍一个SpringBoot+mybatis的多数据源的解决方案。

数据库准备

创建两个数据库，两个数据库都有Im_person表，两个表中无数据。

代码结构

说明：我这里只是为了体现效果，就省略了service步骤。各位大牛开发，不喜勿喷，理解万岁，嘻嘻！！

application.yml中配置两个数据源，配置如下：

master:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://127.0.0.1:3306/db1?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&serverTimezone=GMT%2B8
    username: root
    password:123456

slave:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://127.0.0.1:3306/db2?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&serverTimezone=GMT%2B8
    username: root
    password:123456

2.SpringBoot多数据源配置类。

@Configuration
@MapperScan(basePackages="com.zhangls.multipledatasource.dao.master", sqlSessionFactoryRef="masterSqlSessionFactory")
publicclassMasterDataSourceConfig {
@Value("${master.datasource.driver-class-name}")
privateStringdriverClassName;

@Value("${master.datasource.url}")
privateStringurl;

@Value("${master.datasource.username}")
privateStringusername;

@Value("${master.datasource.password}")
privateStringpassword;

@Bean(name="masterDataSource")
@Primary
publicDataSourcedataSource() {
DruidDataSourcedataSource=newDruidDataSource();
dataSource.setDriverClassName(this.driverClassName);
dataSource.setUrl(this.url);
dataSource.setUsername(this.username);
dataSource.setPassword(this.password);
returndataSource;
    }

@Bean(name="masterSqlSessionFactory")
@Primary
publicSqlSessionFactorysqlSessionFactory(@Qualifier("masterDataSource") DataSourcedataSource) throwsException {
SqlSessionFactoryBeanbean=newSqlSessionFactoryBean();
bean.setDataSource(dataSource);
ResourcePatternResolverresourceResolver=newPathMatchingResourcePatternResolver();
List<String>mapperLocations=newArrayList<>();
mapperLocations.add("classpath*:/mapper/master/*.xml");
List<Resource>resources=newArrayList();
if (mapperLocations!=null) {
for (StringmapperLocation : mapperLocations) {
try {
Resource[] mappers=resourceResolver.getResources(mapperLocation);
resources.addAll(Arrays.asList(mappers));
                } catch (IOExceptione) {
// ignore
                }
            }
        }

bean.setMapperLocations(resources.toArray(newResource[resources.size()]));
returnbean.getObject();
    }

@Bean(name="masterTransactionManager")
@Primary
publicDataSourceTransactionManagertransactionManager(@Qualifier("masterDataSource") DataSourcedataSource) {
returnnewDataSourceTransactionManager(dataSource);
    }

@Bean(name="masterSqlSessionTemplate")
@Primary
publicSqlSessionTemplatetestSqlSessionTemplate(
@Qualifier("masterSqlSessionFactory") SqlSessionFactorysqlSessionFactory) throwsException {
returnnewSqlSessionTemplate(sqlSessionFactory);
    }
}

@Configuration
@MapperScan(basePackages="com.zhangls.multipledatasource.dao.slave", sqlSessionFactoryRef="slaveSqlSessionFactory")
publicclassSlaveDataSourceConfig {
@Value("${slave.datasource.driver-class-name}")
privateStringdriverClassName;

@Value("${slave.datasource.url}")
privateStringurl;

@Value("${slave.datasource.username}")
privateStringusername;

@Value("${slave.datasource.password}")
privateStringpassword;

@Bean(name="slaveDataSource")
publicDataSourcedataSource() {
DruidDataSourcedataSource=newDruidDataSource();
dataSource.setDriverClassName(this.driverClassName);
dataSource.setUrl(this.url);
dataSource.setUsername(this.username);
dataSource.setPassword(this.password);
returndataSource;
    }

@Bean(name="slaveSqlSessionFactory")
publicSqlSessionFactorysqlSessionFactory(@Qualifier("slaveDataSource") DataSourcedataSource) throwsException {
SqlSessionFactoryBeanbean=newSqlSessionFactoryBean();
bean.setDataSource(dataSource);
bean.setMapperLocations(
newPathMatchingResourcePatternResolver().getResources("classpath*:/mapper/slave/*.xml"));
returnbean.getObject();
    }

@Bean(name="slaveTransactionManager")
publicDataSourceTransactionManagertransactionManager(@Qualifier("slaveDataSource") DataSourcedataSource) {
returnnewDataSourceTransactionManager(dataSource);
    }

@Bean(name="slaveSqlSessionTemplate")
publicSqlSessionTemplatetestSqlSessionTemplate(
@Qualifier("masterSqlSessionFactory") SqlSessionFactorysqlSessionFactory) throwsException {
returnnewSqlSessionTemplate(sqlSessionFactory);
    }
}

说明：两个数据的代码基本相同，不同的是要配置好扫描包的路径，以及mybatis的SQL文件的路径。所以不同的连接DAO和mapper文件要放在不同的文件夹，方便配置管理。

实现效果

以上的配置已经实现了多数据源的配置，下面实现一个功能。通过一个接口同时向两个库的两个表中导入不同的数据。

controller代码如下：

@RestController
publicclassDataSourceController {
@Resource
privateMasterPersonMappermasterPersonMapper;
@Resource
privateSlavePersonMapperslavePersonMapper;

@GetMapping("/datasource")
publicStringdatasource() {
ImPersonperson1=newImPerson();
person1.setPersonId("1");
person1.setGender("男");
person1.setBirthday(newDate());
person1.setLocation("中国");
masterPersonMapper.insertSelective(person1);

ImPersonperson2=newImPerson();
person2.setPersonId("2");
person2.setGender("女");
person2.setBirthday(newDate());
person2.setLocation("中国北京");
slavePersonMapper.insertSelective(person2);
return"导入成功";
    }
}