最近项目中需要用到2种数据库驱动连接数据库，下面我们基于MybatisPlus实现一下
具体实现
1、在pom.xml中添加如下依赖:
<properties>
<java.version>1.8</java.version>
<lombok.version>1.18.2</lombok.version>
<mybatis-plus.version>3.2.0</mybatis-plus.version>
<druid.version>1.1.9</druid.version>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
</properties>
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<!-- mysql-->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<scope>runtime</scope>
</dependency>
<!-- postgrepsql-->
<dependency>
<groupId>org.postgresql</groupId>
<artifactId>postgresql</artifactId>
<scope>runtime</scope>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>${mybatis-plus.version}</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus</artifactId>
<version>${mybatis-plus.version}</version>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>${lombok.version}</version>
</dependency>
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid-spring-boot-starter</artifactId>
<version>${druid.version}</version>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
</dependencies>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();
}
}6、新建DataSource注解，如下：
@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
DataSourceEnum value() default DataSourceEnum.DB1;
}7、新建面向类和方法级别的切面，如下：
@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();
}
}8、新建多数据源配置类，如下：
@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<IPage<GatewayLog>> pageList(GatewayLogDTO gatewayLogDTO) {
LambdaQueryWrapper<GatewayLog> 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<GatewayLog> page = new Page<>(gatewayLogDTO.getPageNo(), gatewayLogDTO.getPageSize());
return Result.success(mapper.selectPage(page, wrapper));
}
}启动服务，调用相关的接口，我们会在控制台看到如下信息：
说明数据源切换成功！！！
到此这篇关于关于MybatisPlus配置双数据库驱动连接数据库问题的文章就介绍到这了,更多相关MybatisPlus配置双数据库内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
