目录:

  • 实现思想
  • 导入依赖、配置说明
  • 代码实现
  • 问题总结

一.实现思想

  接手一个旧系统,SpringBoot 使用的是纯粹的 mybatis ,既没有使用规范的代码生成器,也没有使用 JPA 或者 mybatis-plus

  想着接入 mybatis-plus,为以后敲代码省点力气。普通的接入 mybatis-plus 可以直接参考官方文档 https://mp.baomidou.com/ 

  但我接手的系统是个多数据源系统,本来最优的方法是使用官方的 动态数据源 支持 https://mp.baomidou.com/guide/dynamic-datasource.html 

  但我因为乱七八糟的依赖冲突,决定自己实现 动态数据源 的支持。

  实现的核心逻辑:使用一个 代理数据源,来管理 其他数据源 的分发请求。(通过AOP分发)

二.导入依赖、配置说明

  因为依赖的冲突,我没有直接使用

<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.3.1.tmp</version>
</dependency>

  而是引入的

<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus</artifactId>
<version>3.3.1.tmp</version>
</dependency>

  数据库配置文件大致如下

spring:
datasource: #数据库配置
primary: #数据库1
driver-class-name: com.mysql.cj.jdbc.Driver
jdbc-url: jdbc:mysql://
username: root
password: 3
type: com.alibaba.druid.pool.DruidDataSource
second: #数据库2
driver-class-name: com.mysql.cj.jdbc.Driver
jdbc-url: jdbc:mysql://
username: root
password: 3
type: com.alibaba.druid.pool.DruidDataSource
third: #数据库3
driver-class-name: com.mysql.cj.jdbc.Driver
jdbc-url: jdbc:mysql://
username: root
password: 3
type: com.alibaba.druid.pool.DruidDataSource

  MybatisPlus 配置大致如下(mybatis 的配置可以删除)

mybatis-plus:
# 扫描 mapper.xml
mapper-locations: classpath:mapper/*.xml #也可以不配置,在代码中设置
# configuration:
# map-underscore-to-camel-case: false

三.代码实现

  1.我们先新建 数据源的枚举

public enum DataSourceEnums {

    PRIMARY("primaryDataSource"),
SECOND("secondDataSource"),
THIRD("thirdDataSource"); private String value; DataSourceEnums(String value){this.value=value;} public String getValue() {
return value;
} }

  2.用来标记数据源的 注解(在哪里使用哪个数据源)。

/**
* @author zhaww
* @date 2020/4/14
* @Description .自定义 - 区分数据源的注解
*/
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface MyDataSource {
DataSourceEnums value() default DataSourceEnums.PRIMARY;
}

  3.动态数据源管理器,继承 AbstractRoutingDataSource  

/**
* @author zhaww
* @date 2020/4/10
* @Description .动态数据源管理器
*/public class DataSourceContextHolder extends AbstractRoutingDataSource { private static final ThreadLocal<String> contextHolder = new InheritableThreadLocal<>(); /**
* 重写这个方法,这里返回使用的数据源 key 值
* @return
*/
@Override
protected Object determineCurrentLookupKey() {
// log.info("动态切换数据源:" + DataSourceContextHolder.getDataSource());
return contextHolder.get();
} /**
* 设置数据源
* @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();
} }

  4. mybatis-plus 的配置类

/**
* @author zhaww
* @date 2020/4/10
* @Description .
*/
//@EnableTransactionManagement //开启事务
@Configuration
@MapperScan(value = {"com.zydd.admin.dao"}) //扫描Mapper 层的类
public class MybatisPlusConfig { @Bean(name = "primaryDataSource")
@ConfigurationProperties(prefix = "spring.datasource.primary")
public DataSource primaryDataSource() {
return DataSourceBuilder.create().build();
} @Bean(name = "secondDataSource")
@ConfigurationProperties(prefix = "spring.datasource.second")
public DataSource secondDataSource() {
return DataSourceBuilder.create().build();
} @Bean(name = "thirdDataSource")
@ConfigurationProperties(prefix = "spring.datasource.third")
public DataSource thirdDataSource() {
return DataSourceBuilder.create().build();
} @Bean(name = "multipleTransactionManager")
@Primary
public DataSourceTransactionManager multipleTransactionManager(@Qualifier("multipleDataSource") DataSource dataSource) {
// return new MyDataSourceTransactionManager(dataSource);
return new DataSourceTransactionManager(dataSource);
} /**
* 动态数据源配置
*
* @return
*/
@Bean(name = "multipleDataSource")
@Primary
public DataSource multipleDataSource(@Qualifier("primaryDataSource") DataSource primaryDataSource,
@Qualifier("secondDataSource") DataSource secondDataSource,
@Qualifier("thirdDataSource") DataSource thirdDataSource) {
DataSourceContextHolder dynamicDataSource = new DataSourceContextHolder();
Map<Object, Object> targetDataSources = new HashMap<>();
targetDataSources.put(DataSourceEnums.PRIMARY.getValue(), primaryDataSource);
targetDataSources.put(DataSourceEnums.SECOND.getValue(), secondDataSource);
targetDataSources.put(DataSourceEnums.THIRD.getValue(), thirdDataSource);
dynamicDataSource.setTargetDataSources(targetDataSources);
dynamicDataSource.setDefaultTargetDataSource(thirdDataSource); // 默认使用的数据源
return dynamicDataSource;
} @Bean("sqlSessionFactory")
public SqlSessionFactory sqlSessionFactory() throws Exception {
MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean();
sqlSessionFactory.setDataSource(multipleDataSource(primaryDataSource(), secondDataSource(), thirdDataSource()));

      //mybatis-plus yml 配置不生效,要在这里代码里配置
MybatisConfiguration configuration = new MybatisConfiguration();
configuration.setJdbcTypeForNull(JdbcType.NULL);
//是否使用转驼峰
configuration.setMapUnderscoreToCamelCase(false);
configuration.setCacheEnabled(false);
sqlSessionFactory.setConfiguration(configuration); //添加分页功能
Interceptor[] plugins = {paginationInterceptor()};
sqlSessionFactory.setPlugins(plugins); //扫描 mapper 路径
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
Resource[] resource = resolver.getResources("classpath:mapper/**/*.xml");
sqlSessionFactory.setMapperLocations(resource);
return sqlSessionFactory.getObject();
} /**
* @Description : mybatis-plus分页插件
*/
@Bean
public PaginationInterceptor paginationInterceptor() {
PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
// 设置请求的页面大于最大页后操作, true调回到首页,false 继续请求 默认false
paginationInterceptor.setOverflow(true);
// 设置最大单页限制数量,默认 500 条,-1 不受限制
// paginationInterceptor.setLimit(30);
return paginationInterceptor;
} }

  5.使用AOP来实现数据源的动态设置。

/**
* @author zhaww
* @date 2020/4/9
* @Description .AOP通用日志记录、动态数据源分发
*/
@Aspect
@Component
@Slf4j
@Order(-100)
public class AOP { /**
* Controller层路径
*/
@Pointcut("within(com.zydd.admin.controller..*)")
public void controllerPointcut() {
} /**
* Service层路径
*/
@Pointcut("within(com.zydd.admin.service..*)")
public void servicePointcut() {
} @Around("servicePointcut()")
public Object doServiceLogging(ProceedingJoinPoint joinPoint) throws Throwable {
changeDataSource(joinPoint); //检查数据源
} /**
* Mapper层拦截,动态切换 mybatisPlus 数据源
*/
@Before("execution(* com.zydd.admin.dao.primary..*(..))")
public void doAdmin(){
log.info("选择数据源---" + DataSourceEnums.PRIMARY.getValue());
DataSourceContextHolder.setDataSource(DataSourceEnums.PRIMARY.getValue());
} /**
* Mapper层拦截,动态切换 mybatisPlus 数据源
*/
@Before("execution(* com.zydd.admin.dao.second..*(..))")
public void doZYDD(){
log.info("选择数据源---" + DataSourceEnums.SECOND.getValue());
DataSourceContextHolder.setDataSource(DataSourceEnums.SECOND.getValue());
} /**
* Mapper层拦截,动态切换 mybatisPlus 数据源
*/
@Before("execution(* com.zydd.admin.dao.third..*(..))")
public void doDW(){
log.info("选择数据源---" + DataSourceEnums.THIRD.getValue());
DataSourceContextHolder.setDataSource(DataSourceEnums.THIRD.getValue());
} /**
* 通过注解 变更数据源
* @param joinPoint
*/
private void changeDataSource(ProceedingJoinPoint joinPoint) {
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
Method method = signature.getMethod();
MyDataSource myDataSource = null;
//优先判断方法上的注解
if (method.isAnnotationPresent(MyDataSource.class)) {
myDataSource = method.getAnnotation(MyDataSource.class);
DataSourceContextHolder.setDataSource(myDataSource.value().getValue());
} else if (method.getDeclaringClass().isAnnotationPresent(MyDataSource.class)) { //其次判断类上的注解
myDataSource = method.getDeclaringClass().getAnnotation(MyDataSource.class);
DataSourceContextHolder.setDataSource(myDataSource.value().getValue());
}
     if (myDataSource != null) {
     log.info("注解方式选择数据源---" + myDataSource.value().getValue());
     }
  } 
}

 注意:我们不但默认通过 Mapper 的路径来切换数据源,还通过 Service 方法层来切换数据源。

  因为如果 service 有事务的话,进入service方法的时候,DataSourceTransactionManager 就设置好了默认数据源,就算通过Mapper层重新设置数据源,

  DataSourceTransactionManager 的默认数据源还是没有变。

  所以在 事务管理器 设置默认数据源之前,就切换数据源,实现动态事务+动态数据源。

  6.实际使用,只要 MyDataSource 注解就ok了。也可以在 ServiceImpl 类上加注解。

    @Override
@MyDataSource(DataSourceEnums.THIRD)
@Transactional
public void test() {
DwUserMPEntity test2 = new DwUserMPEntity();
test2.setUuid("test");
test2.setNickname("test");
test2.setPhone("test");
dwUserDao.insert(test2);
// throw new RuntimeException("lala");
} @Override
@MyDataSource(DataSourceEnums.PRIMARY)
@Transactional
public void test1() {
SysRoleMPEntity test = new SysRoleMPEntity();
test.setRole("test");
test.setDescription("test");
sysRoleDao.insert(test);
// throw new RuntimeException("lala");
}

五.问题总结

1.配置文件里 mybatis-plus的配置不生效:因为我们在 SqlSessionFactory 里重新写了 MybatisConfiguration 。

2.启用事务的话,动态数据源不生效:因为 service 有事务的话,在进入service方法时,DataSourceTransactionManager 就设置好了默认数据源。

多数据源系统接入mybatis-plus, 实现动态数据源、动态事务。的更多相关文章

  1. myBatis源码解析-数据源篇(3)

    前言:我们使用mybatis时,关于数据源的配置多使用如c3p0,druid等第三方的数据源.其实mybatis内置了数据源的实现,提供了连接数据库,池的功能.在分析了缓存和日志包的源码后,接下来分析 ...

  2. Spring Boot + Mybatis 实现动态数据源

    动态数据源 在很多具体应用场景的时候,我们需要用到动态数据源的情况,比如多租户的场景,系统登录时需要根据用户信息切换到用户对应的数据库.又比如业务A要访问A数据库,业务B要访问B数据库等,都可以使用动 ...

  3. Spring Boot:实现MyBatis动态数据源

    综合概述 在很多具体应用场景中,我们需要用到动态数据源的情况,比如多租户的场景,系统登录时需要根据用户信息切换到用户对应的数据库.又比如业务A要访问A数据库,业务B要访问B数据库等,都可以使用动态数据 ...

  4. SpringBoot+Mybatis 实现动态数据源切换方案

    背景 最近让我做一个大数据的系统,分析了一下,麻烦的地方就是多数据源切换抽取数据.考虑到可以跨服务器跨数据库抽数,再整理数据,就配置了这个动态数据源的解决方案.在此分享给大家. 实现方案 数据库配置文 ...

  5. SaaS 系统架构,Spring Boot 动态数据源实现!

    这段时候在准备从零开始做一套SaaS系统,之前的经验都是开发单数据库系统并没有接触过SaaS系统,所以接到这个任务的时候也有也些头疼,不过办法部比困难多,难得的机会. 在网上找了很多关于SaaS的资料 ...

  6. Spring Boot + Mybatis多数据源和动态数据源配置

    文章转自 https://blog.csdn.net/neosmith/article/details/61202084 网上的文章基本上都是只有多数据源或只有动态数据源,而最近的项目需要同时使用两种 ...

  7. SpringBoot集成Mybatis配置动态数据源

    很多人在项目里边都会用到多个数据源,下面记录一次SpringBoot集成Mybatis配置多数据源的过程. pom.xml <?xml version="1.0" encod ...

  8. 使用 mybatis plus 动态数据源

    1.pom.xml 增加 <dependency> <groupId>com.baomidou</groupId> <artifactId>dynami ...

  9. mybatis多数据库切换,(动态数据源)。

    项目中将一个库的某些标的某些数据保存到另一个库. 使用spring的aop编程动态切换数据源,代码如下,以备下次用到! 1.先将两个数据库连接,创建两个数据源,交于spring管理! <bean ...

随机推荐

  1. 微信小程序修改request合法域名不生效及解决方法

    在小程序微信公众平台修改后请求,依然在console中显示修改前的域名. 解决:在小程序开发者工具中点击“详情”后点击“域名信息”,就会自动刷新

  2. JDK14发布了,快来看看有什么新特性吧

    序言

  3. C++之 ostream详细用法

    前言 在 C++中,ostream表示输出流,英文”output stream“的简称.在 C++中常见的输出流对象就是标准输出流cout,很少自定义ostream的对象,更多的是直接使用cout.那 ...

  4. Natas0 Writeup(前端源码、敏感信息)

    Natas0: 提示密码就在本页,右键查看源码,注释中发现flag. flag:gtVrDuiDfck831PqWsLEZy5gyDz1clto 总结:客户端代码注释中切勿保存敏感信息.  

  5. Scapy编写ICMP扫描脚本

    使用Scapy模块编写ICMP扫描脚本: from scapy.all import * import optparse import threading import os def scan(ipt ...

  6. Vue中使用axios发送ajax请求

    作为前后端交互的重要技巧--发送ajax请求,在Vue中我们使用axio来完成这一需求: 首先是下载axios的依赖, npm install --save axios vue-axios 然后在ma ...

  7. 安装sql server 2005时出现“安装汇编”错误的解决办法

    今天安装sql server 2005 management studio到最后步骤的时候报“安装汇编”错误,卸载重装的几遍还是不行,最后将net framework 3.5删除后,终于安装成功了.

  8. python环境变量忘记配置

    Python安装没有勾选配置环境变量安装 解决方法一: 于是,便用安装包卸载后重新安装. 重新安装勾选 安装成功 方法二: 配置环境变量 A.右键点击“我的电脑”,点击“属性”: B.在弹出的界面中点 ...

  9. http服务部署

    [root@xiaoyi ~]# vim /etc/ssh/sshd_config 17行 Port 22 49行 PermitRootLogin yes 观察IP地址与物理机属于同一个网段 打开Fi ...

  10. TensorFlow 趣题

    checkpoint 文件夹 Tensorflow训练后的模型可以保存checkpoint文件,checkpoint文件是结构与权重分离的四个文件,便于训练. 1)checkpoint 文件 保存断点 ...