MybatisPlus配置双数据库驱动 | 代码搬运工
最近项目中需要用到2种数据库驱动连接数据库，下面我们基于MybatisPlus实现一下
具体实现
1、在pom.xml中添加如下依赖:
1.8 1.18.2 3.2.0 1.1.9 UTF-8 UTF-8 org.springframework.boot spring-boot-starter-webmysql mysql-connector-javaruntime org.postgresql postgresqlruntime com.baomidou mybatis-plus-boot-starter${mybatis-plus.version} com.baomidou mybatis-plus${mybatis-plus.version} org.projectlombok lombok${lombok.version} com.alibaba druid-spring-boot-starter${druid.version} org.springframework.boot spring-boot-starter-aop
2、在yml配置文件中添加如下配置:
server:
port: 8080
spring:
application:
name: xxxx
datasource:
druid:
# mysql数据源配置
db1:
driver-class-name: com.mysql.jdbc.Driver
url: jdbc:mysql://127.0.0.1:3306/db1?useUnicode=true&characterEncoding=UTF-8&useSSL=false
username: ${username}
password: ${password}
initial-size: 5
min-idle: 5
max-active: 50
# postgresql 数据源配置
db2:
driver-class-name: org.postgresql.Driver
url: jdbc:postgresql://127.0.0.1:5432/db2?useUnicode=true&characterEncoding=utf-8
username: ${username}
password: ${password}
initial-size: 5
min-idle: 5
max-active: 50
# mybatis-plus配置
mybatis-plus:
type-aliases-package: com.dms.gateway.api.entity
mapper-locations: classpath:/mapper
public static void setDataSource(String db){
contextHolder.set(db);
}
public static String getDataSource(){
return contextHolder.get();
}
public static void clear(){
contextHolder.remove();
}
}
5、新建MultipleDataSource类，如下：
public class MultipleDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContextHolder.getDataSource();
}
}
5、新建DataSource注解，如下：
@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@documented
public @interface DataSource {
DataSourceEnum value() default DataSourceEnum.DB1;
}
6、新建面向类和方法级别的切面，如下：
@Component
@Slf4j
@Aspect
@Order(-6)
public class DataSourceClassAspect {
@Before("@within(dataSource)")
public void doBefore(JoinPoint point, DataSource dataSource){
log.info("切换到数据源[{}]", dataSource.value().getValue());
DataSourceContextHolder.setDataSource(dataSource.value().getValue());
}
@After("@within(dataSource)")
public void doAfter(JoinPoint point, DataSource dataSource){
log.info("回收数据源[{}]", dataSource.value().getValue());
DataSourceContextHolder.clear();
}
}
@Component
@Slf4j
@Aspect
@Order(-5)
public class DataSourceMethodAspect {
@Before("@annotation(dataSource)")
public void doBefore(JoinPoint point, DataSource dataSource){
log.info("切换到数据源[{}]", dataSource.value().getValue());
DataSourceContextHolder.setDataSource(dataSource.value().getValue());
}
@After("@annotation(dataSource)")
public void doAfter(JoinPoint point, DataSource dataSource){
log.info("回收数据源[{}]", dataSource.value().getValue());
DataSourceContextHolder.clear();
}
}
7、新建多数据源配置类，如下：
@Configuration
@MapperScan("com.dms.gateway.api.mapper")
public class MybatisPlusConfig {
@Bean
public PaginationInterceptor paginationInterceptor() {
PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
return paginationInterceptor;
}
@Bean(name = "db1")
@ConfigurationProperties(prefix = "spring.datasource.druid.db1" )
public DataSource db1() {
return DruidDataSourceBuilder.create().build();
}
@Bean(name = "db2")
@ConfigurationProperties(prefix = "spring.datasource.druid.db2" )
public DataSource db2() {
return DruidDataSourceBuilder.create().build();
}
@Bean
@Primary
public DataSource multipleDataSource(@Qualifier("db1") DataSource db1,
@Qualifier("db2") DataSource db2) {
MultipleDataSource multipleDataSource = new MultipleDataSource();
Map< Object, Object > targetDataSources = new HashMap<>();
targetDataSources.put(DataSourceEnum.DB1.getValue(), db1);
targetDataSources.put(DataSourceEnum.DB2.getValue(), db2);
//添加数据源
multipleDataSource.setTargetDataSources(targetDataSources);
//设置默认数据源
multipleDataSource.setDefaultTargetDataSource(db1);
return multipleDataSource;
}
@Bean("sqlSessionFactory")
public SqlSessionFactory sqlSessionFactory() throws Exception {
MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean();
sqlSessionFactory.setDataSource(multipleDataSource(db1(),db2()));
MybatisConfiguration configuration = new MybatisConfiguration();
configuration.setJdbcTypeForNull(JdbcType.NULL);
configuration.setMapUnderscoreToCamelCase(true);
configuration.setCacheEnabled(false);
sqlSessionFactory.setConfiguration(configuration);
//添加分页功能
sqlSessionFactory.setPlugins(paginationInterceptor());
return sqlSessionFactory.getObject();
}
}
接下来需要具体的业务逻辑，在service层的类或者方法上面添加@DataSource注解来指定该业务需要用到的数据源，如下：
@Service
@DataSource(DataSourceEnum.DB2)
public class GatewayLogService {
@Autowired
private GatewayLogMapper mapper;
@Override
public Result> pageList(GatewayLogDTO gatewayLogDTO) {
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery();
wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getPath()), GatewayLog::getPath, gatewayLogDTO.getPath());
wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getSourceServer()), GatewayLog::getSourceServer, gatewayLogDTO.getSourceServer());
wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getTargetServer()), GatewayLog::getTargetServer, gatewayLogDTO.getTargetServer());
wrapper.eq(StringUtils.isNotEmpty(gatewayLogDTO.getMethod()), GatewayLog::getMethod, gatewayLogDTO.getMethod());
wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getRequestBody()), GatewayLog::getRequestBody, gatewayLogDTO.getRequestBody());
wrapper.like(StringUtils.isNotEmpty(gatewayLogDTO.getResponse()), GatewayLog::getResponse, gatewayLogDTO.getResponse());
wrapper.orderByDesc(GatewayLog::getId);
IPage page = new Page<>(gatewayLogDTO.getPageNo(), gatewayLogDTO.getPageSize());
return Result.success(mapper.selectPage(page, wrapper));
}
}
启动服务，调用相关的接口，我们会在控制台看到如下信息：
说明数据源切换成功！！！
