背景介绍
在使用Spring Boot进行开发的过程中，我们通常会遇到多数据源的需求。比如，一个系统需要同时连接多个数据库进行读写操作，或者需要连接不同的数据库进行分库分表。本文将介绍如何使用Spring Boot整合Mybatis-Plus和Druid，实现多数据源的配置和使用。
步骤一：添加依赖
首先，我们需要在pom.xml文件中添加以下依赖：
<dependencies>
...
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.4.0</version>
</dependency>
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid-spring-boot-starter</artifactId>
<version>1.2.6</version>
</dependency>
...
</dependencies>
步骤二：配置数据源
在application.yml中配置数据源信息：
spring:
datasource:
druid:
default:
url: jdbc:mysql://localhost:3306/db1?useUnicode=true&characterEncoding=utf8&useSSL=false
username: root
password: root
driver-class-name: com.mysql.jdbc.Driver
db2:
url: jdbc:mysql://localhost:3306/db2?useUnicode=true&characterEncoding=utf8&useSSL=false
username: root
password: root
driver-class-name: com.mysql.jdbc.Driver
步骤三：配置多数据源
在config包下创建一个DataSourceConfig类，配置多数据源：
@Configuration
public class DataSourceConfig {
@Primary
@Bean(name = "dataSource1")
@ConfigurationProperties(prefix = "spring.datasource.druid.default")
public DataSource dataSource1() {
return DruidDataSourceBuilder.create().build();
}
@Bean(name = "dataSource2")
@ConfigurationProperties(prefix = "spring.datasource.druid.db2")
public DataSource dataSource2() {
return DruidDataSourceBuilder.create().build();
}
@Bean
public DataSourceRouting dataSourceRouting(@Qualifier("dataSource1") DataSource dataSource1,
@Qualifier("dataSource2") DataSource dataSource2) {
DataSourceRouting dataSourceRouting = new DataSourceRouting();
Map<Object, Object> dataSourceMap = new HashMap<>();
dataSourceMap.put("dataSource1", dataSource1);
dataSourceMap.put("dataSource2", dataSource2);
dataSourceRouting.setTargetDataSources(dataSourceMap);
dataSourceRouting.setDefaultTargetDataSource(dataSource1);
return dataSourceRouting;
}
@Bean
public SqlSessionFactory sqlSessionFactory(DataSourceRouting dataSourceRouting) throws Exception {
MybatisSqlSessionFactoryBean factoryBean = new MybatisSqlSessionFactoryBean();
factoryBean.setDataSource(dataSourceRouting);
return factoryBean.getObject();
}
@Bean
public PlatformTransactionManager transactionManager(DataSourceRouting dataSourceRouting) {
return new DataSourceTransactionManager(dataSourceRouting);
}
}
步骤四：编写Mapper和Service
在使用Mybatis-Plus进行操作数据库时，需要编写Mapper和Service。这里以两个数据源为例，分别创建两个包com.example.mapper和com.example.service用于存放不同数据源的Mapper和Service接口。
在com.example.mapper中创建Db1Mapper和Db2Mapper接口，并使用@Mapper注解进行标记：
@Mapper
public interface Db1Mapper extends BaseMapper<Db1Entity> {
}
@Mapper
public interface Db2Mapper extends BaseMapper<Db2Entity> {
}
在com.example.service中创建Db1Service和Db2Service接口：
public interface Db1Service extends IService<Db1Entity> {
}
public interface Db2Service extends IService<Db2Entity> {
}
步骤五：使用多数据源
在需要使用数据源的地方，通过注解的方式指定要使用的数据源。例如，在Service实现类中：
@Service
public class Db1ServiceImpl extends ServiceImpl<Db1Mapper, Db1Entity> implements Db1Service {
@Override
@DataSource("dataSource1")
public List<Db1Entity> getDb1Data() {
return baseMapper.selectList(null);
}
}
@Service
public class Db2ServiceImpl extends ServiceImpl<Db2Mapper, Db2Entity> implements Db2Service {
@Override
@DataSource("dataSource2")
public List<Db2Entity> getDb2Data() {
return baseMapper.selectList(null);
}
}
通过@DataSource注解指定要使用的数据源名称，即@Bean(name = "dataSource1")或@Bean(name = "dataSource2")中定义的名称。
步骤六：测试多数据源
编写测试类进行多数据源的测试：
@SpringBootTest
class MultiDataSourceTest {
@Autowired
private Db1Service db1Service;
@Autowired
private Db2Service db2Service;
@Test
void testMultiDataSource() {
List<Db1Entity> db1Data = db1Service.getDb1Data();
List<Db2Entity> db2Data = db2Service.getDb2Data();
// 进行数据操作和断言
}
}
运行测试类，验证多数据源是否配置成功。
结语
本文介绍了如何使用Spring Boot整合Mybatis-Plus和Druid实现多数据源的配置和使用。通过配置不同的数据源，并使用注解的方式指定要使用的数据源，可以方便地实现多数据源的操作。希望该文章对你有所帮助！
本文来自极简博客，作者：星辰坠落，转载请注明原文链接：Spring Boot整合Mybatis-Plus Druid实现多数据源
