关于MybatisPlus配置双数据库驱动连接数据库问题


Posted in Java/Android onJanuary 22, 2022

最近项目中需要用到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/*Mapper.xml
  global-config:
    db-config:
      id-type: auto
      field-strategy: not_empty
      logic-delete-value: 1
      logic-not-delete-value: 0
  configuration:
    map-underscore-to-camel-case: true
    cache-enabled: false
    call-setters-on-nulls: true
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl

3、新建DataSourceEnum枚举类,如下:

public enum DataSourceEnum {
 
    DB1("db1"),
    DB2("db2");
 
    private String value;
 
    DataSourceEnum(String value){this.value=value;}
 
    public String getValue() {
        return value;
    }
}

4、新建DataSourceContextHolder类,如下:

public class DataSourceContextHolder {
 
    // 默认数据源
    public static final String DEFAULT_DS = DataSourceEnum.DB1.getValue();
 
    private static final ThreadLocal<String> contextHolder = new InheritableThreadLocal<>();
 
    /**
     *  设置数据源
     * @param db
     */
    public static void setDataSource(String db){
        contextHolder.set(db);
    }
 
    /**
     * 取得当前数据源
     * @return
     */
    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();
    }
 
    /**
     * 动态数据源配置
     * @return
     */
    @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配置双数据库驱动连接数据库问题的文章就介绍到这了,更多相关MybatisPlus配置双数据库内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Java/Android 相关文章推荐
JVM入门之类加载与字节码技术(类加载与类的加载器)
Jun 15 Java/Android
springcloud之Feign超时问题的解决
Jun 24 Java/Android
springboot集成flyway自动创表的详细配置
Jun 26 Java/Android
JavaWeb 入门篇:创建Web项目,Idea配置tomcat
Jul 16 Java/Android
Java Spring 控制反转(IOC)容器详解
Oct 05 Java/Android
Java实战之课程信息管理系统的实现
Apr 01 Java/Android
Java中Quartz高可用定时任务快速入门
Apr 03 Java/Android
mapstruct的用法之qualifiedByName示例详解
Apr 06 Java/Android
Elasticsearch Recovery 详细介绍
Apr 19 Java/Android
Java工作中实用的代码优化技巧分享
Apr 21 Java/Android
Android存储中最基本的文件存储方式
Apr 30 Java/Android
Java Redisson多策略注解限流
Sep 23 Java/Android
JavaCV实现照片马赛克效果
Jan 22 #Java/Android
maven依赖的version声明控制方式
深入浅出讲解Java8函数式编程
Jan 18 #Java/Android
关于maven依赖 ${xxx.version}报错问题
Jan 18 #Java/Android
Eclipse+Java+Swing+Mysql实现电影购票系统(详细代码)
关于Spring配置文件加载方式变化引发的异常详解
Jan 18 #Java/Android
springboot中的pom文件 project报错问题
Jan 18 #Java/Android
You might like
MySQL授权问题总结
2007/05/06 PHP
php读取txt文件组成SQL并插入数据库的代码(原创自Zjmainstay)
2012/07/31 PHP
浅析Mysql 数据回滚错误的解决方法
2013/08/05 PHP
php把数据表导出为Excel表的最简单、最快的方法(不用插件)
2014/05/10 PHP
Laravel 4 初级教程之Pages、表单验证
2014/10/30 PHP
php unlink()函数使用教程
2018/07/12 PHP
php如何利用pecl安装mongodb扩展详解
2019/01/09 PHP
在一个form用一个SUBMIT(或button)分别提交到两个处理表单页面的代码
2007/02/15 Javascript
学习js所必须要知道的一些
2007/03/07 Javascript
jQuery 网易相册鼠标移动显示隐藏效果实现代码
2013/03/31 Javascript
js操作checkbox遇到的问题解决
2013/06/29 Javascript
js取值中form.all和不加all的区别介绍
2014/01/20 Javascript
jquery取消选择select下拉框示例代码
2014/02/22 Javascript
Javascript中的数据类型之旅
2015/10/18 Javascript
JS验证逗号隔开可以是中文字母数字
2016/04/22 Javascript
javascript简单实现等比例缩小图片的方法
2016/07/27 Javascript
Javascript vue.js表格分页,ajax异步加载数据
2016/10/24 Javascript
简单的网页广告特效实例
2017/08/19 Javascript
浅谈webpack下的AOP式无侵入注入
2017/11/12 Javascript
微信小程序如何获取用户手机号
2018/01/26 Javascript
vue观察模式浅析
2018/09/25 Javascript
[58:42]DOTA2上海特级锦标赛C组败者赛 Newbee VS Archon第一局
2016/02/27 DOTA
[04:46]2018年度玩家喜爱的电竞媒体-完美盛典
2018/12/16 DOTA
从零学Python之入门(三)序列
2014/05/25 Python
python ftp 按目录结构上传下载的实现代码
2018/09/12 Python
Python设计模式之模板方法模式实例详解
2019/01/17 Python
python飞机大战pygame游戏背景设计详解
2019/12/17 Python
python3利用Axes3D库画3D模型图
2020/03/25 Python
使用matplotlib动态刷新指定曲线实例
2020/04/23 Python
详解CSS3浏览器兼容
2016/12/14 HTML / CSS
CSS3实现div从下往上滑入滑出效果示例
2020/04/28 HTML / CSS
Urban Outfitters德国官网:美国跨国生活方式零售公司
2018/05/21 全球购物
管理站站长岗位职责
2013/11/27 职场文书
实践单位评语
2014/04/26 职场文书
2014学习优秀共产党员先进事迹材料思想汇报
2014/09/14 职场文书
python numpy中setdiff1d的用法说明
2021/04/22 Python