SpringBoot2.X基础教程:SpringBoot整合MyBatis多数据源【附源码】
一、引入依赖
org.mybatis.spring.boot
mybatis-spring-boot-starter
2.2.0
mysql
mysql-connector-java
runtime
二、配置&初始化数据源
接下来,在 application.properties 中配置数据源,不同于上篇文章,这里的数据源需要配置两个,如下:
spring.datasource.master.jdbc-url = jdbc:mysql://139.196.20.xxx:3306/spring_boot1?useUnicode=true&characterEncoding=utf-8
spring.datasource.master.username = root
spring.datasource.master.password = 123456
spring.datasource.master.driver-class-name = com.mysql.cj.jdbc.Driver
spring.datasource.slave.jdbc-url = jdbc:mysql://139.196.20.xxx:3306/spring_boot2?useUnicode=true&characterEncoding=utf-8
spring.datasource.slave.username = root
spring.datasource.slave.password = 123456
spring.datasource.slave.driver-class-name = com.mysql.cj.jdbc.Driver
这里通过 master 和 slave 对数据源进行了区分,但是加了 master 和 slave 之后,这里的配置就没法被 SpringBoot 自动加载了 (因为前面的 key 变了),需要我们自己去加载 DataSource 了,此时,需要自己配置一个 DataSourceConfig,用来提供两个 DataSource Bean,如下:
@Configuration
public class DataSourceConfiguration {
@Primary
@Bean
@ConfigurationProperties(prefix = "spring.datasource.master")
public DataSource masterDataSource() {
return DataSourceBuilder.create().build();
}
@Bean
@ConfigurationProperties(prefix = "spring.datasource.slave")
public DataSource slaveDataSource() {
return DataSourceBuilder.create().build();
}
}
这里提供了两个 Bean, @ConfigurationProperties(prefix = “spring.datasource.master”) 表示使用 spring.datasource.master 前缀的数据库配置去创建一个 DataSource,一层一层注入,通过两个不同 DataSource 创建两个 SqlSessionFactory,然后再包装到 SqlSessionTemplate中。
配置类上使用 @MapperScan 注解来指定当前数据源下定义的 Entity 和 Mapper 的包路径,另外需要指定 sqlSessionFactory 和sqlSessionTemplate,这两个具体实现在该配置类中类中初始化。
@Configuration
@MapperScan(
basePackages = "com.fish.chapter7.master",
sqlSessionFactoryRef = "masterSqlSessionFactory",
sqlSessionTemplateRef = "masterSqlSessionTemplate")
public class MasterConfig {
@Bean
public SqlSessionFactory masterSqlSessionFactory(@Qualifier("masterDataSource") DataSource masterDataSource) throws Exception {
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(masterDataSource);
return bean.getObject();
}
@Bean
public SqlSessionTemplate masterSqlSessionTemplate(@Qualifier("masterSqlSessionFactory") SqlSessionFactory masterSqlSessionFactory) throws Exception {
return new SqlSessionTemplate(masterSqlSessionFactory);
}
}
@Configuration
@MapperScan(
basePackages = "com.fish.chapter7.slave",
sqlSessionFactoryRef = "slaveSqlSessionFactory",
sqlSessionTemplateRef = "slaveSqlSessionTemplate")
public class SlaveConfig {
@Bean
public SqlSessionFactory slaveSqlSessionFactory(@Qualifier("slaveDataSource") DataSource slaveDataSource) throws Exception {
SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
bean.setDataSource(slaveDataSource);
return bean.getObject();
}
@Bean
public SqlSessionTemplate slaveSqlSessionTemplate(@Qualifier("slaveSqlSessionFactory") SqlSessionFactory slaveSqlSessionFactory) throws Exception {
return new SqlSessionTemplate(slaveSqlSessionFactory);
}
}
三、准备测试数据
创建spring_boot1、spring_boot2两个数据库
spring_boot1
CREATE TABLE `spring_boot1`.`article` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`title` varchar(255) NULL COMMENT '文章标题',
`author` varchar(255) NULL COMMENT '作者',
PRIMARY KEY (`id`)
) COMMENT = '文章表';
INSERT INTO `spring_boot1`.`article`(`id`, `title`, `author`) VALUES (1, 'master', 'Java程序鱼');
spring_boot2
CREATE TABLE `spring_boot2`.`article` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`title` varchar(255) NULL COMMENT '文章标题',
`author` varchar(255) NULL COMMENT '作者',
PRIMARY KEY (`id`)
) COMMENT = '文章表';
INSERT INTO `spring_boot2`.`article`(`id`, `title`, `author`) VALUES (1, 'slave', 'Java程序鱼');
四、核心代码
@RestController
public class BlogController {
@Autowired
public BlogService blogService;
/**
* 查询所有文章(master数据源)
*
* @return
*/
@GetMapping(value = "/getAllByMasterDatabase")
public List<Blog> getAllByMasterDatabase() {
return blogService.getAllByMasterDatabase();
}
/**
* 查询所有文章(slave数据源)
*
* @return
*/
@GetMapping(value = "/getAllBySlaveDatabase")
public List<Blog> getAllBySlaveDatabase() {
return blogService.getAllBySlaveDatabase();
}
}
@Service
public class BlogServiceImpl implements BlogService {
@Autowired
private BlogMapperMaster blogMapperMaster;
@Autowired
private BlogMapperSlave blogMapperSlave;
@Override
public List<Blog> getAllByMasterDatabase() {
return blogMapperMaster.getAll();
}
@Override
public List<Blog> getAllBySlaveDatabase() {
return blogMapperSlave.getAll();
}
}
五、源码
本文的相关例子可以查看下面仓库中的 chapter7 目录:
- Gitee:https://gitee.com/hezhiyuan007/spring-boot-study
- Github:https://github.com/java-fish-0907/spring-boot-study