文档章节

baomidou的dynamic-datasource读写分离实现和加入AOP根据方法名选择库

攻城狮-飞牛
 攻城狮-飞牛
发布于 10/22 10:01
字数 620
阅读 646
收藏 12

文档

https://gitee.com/baomidou/dynamic-datasource-spring-boot-starter/wikis/pages

maven

        <dependency>
			<groupId>com.baomidou</groupId>
			<artifactId>dynamic-datasource-spring-boot-starter</artifactId>
			<version>2.5.7</version>
		</dependency>

纯读写分离(mybatis环境)

场景:

  1. 在纯的读写分离环境,写操作全部是master,读操作全部是slave。
  2. 不想通过注解配置完成以上功能。

答:在mybatis环境下可以基于mybatis插件结合本数据源完成以上功能。 手动注入插件。

@Bean
public MasterSlaveAutoRoutingPlugin masterSlaveAutoRoutingPlugin(){
    return new MasterSlaveAutoRoutingPlugin();
}

默认主库名称master,从库名称slave。

问题

       我在配置好了之后,调试发现对数据库读的操作不得进入MasterSlaveAutoRoutingPlugin,而且进入了默认的库。只有写进入了MasterSlaveAutoRoutingPlugin中。当然也可以默认为从库,但是感觉就不是很好。

       于是我自定义了一个aop切面来,来完成库的选择,代码如下:

import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.context.annotation.Lazy;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import lombok.extern.java.Log;

/**
 * Copyright: Copyright (c) 2019
 * <p> 说明:动态数据源配置 </P>
 * 
 * @version: V1.0
 * @author: BianPeng
 * 
 */
@Aspect
@Component
@Order(0)
@Lazy(false)
@Log
public class DataSourceAop{

	private static final String MASTER = "master";

	private static final String SLAVE = "slave";

	
	@Pointcut("execution(* com.buybit.power.service..*.*(..)) || execution(* com.baomidou.mybatisplus.extension.service..*.*(..))")
    public void checkArgs() {
    }
	
	// 这里切到你的方法目录
	@Before("checkArgs()")
	public void process(JoinPoint joinPoint) throws NoSuchMethodException, SecurityException {
		String methodName = joinPoint.getSignature().getName();
		if (methodName.startsWith("get") 
                || methodName.startsWith("count") 
                || methodName.startsWith("find")
				|| methodName.startsWith("list") 
                || methodName.startsWith("select") 
                || methodName.startsWith("check")
				|| methodName.startsWith("page")) {

			log.info("当前执行的库:"+SLAVE);
			DynamicDataSourceContextHolder.push(SLAVE);
		} else {
			log.info("当前执行的库:"+MASTER);
			DynamicDataSourceContextHolder.push(MASTER);
		}
	}
	@After("checkArgs()")
    public void afterAdvice(){
        DynamicDataSourceContextHolder.clear();
    }
}

但是发现,baomidou/dynamic-datasource自带的@DS没失去了着用,于是我把有@DS的类和方法排除掉,代码入下:

import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.context.annotation.Lazy;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import lombok.extern.java.Log;

/**
 * Copyright: Copyright (c) 2019
 * <p> 说明:动态数据源配置 </P>
 * 
 * @version: V1.0
 * @author: BianPeng
 * 
 */
@Aspect
@Component
@Order(0)
@Lazy(false)
@Log
public class DataSourceAop{

	private static final String MASTER = "master";

	private static final String SLAVE = "slave";

	
	@Pointcut("execution(* com.buybit.power.service..*.*(..)) || execution(* com.baomidou.mybatisplus.extension.service..*.*(..))")
    public void checkArgs() {
    }
	
	// 这里切到你的方法目录
	@Before("checkArgs()")
	public void process(JoinPoint joinPoint) throws NoSuchMethodException, SecurityException {
		String methodName = joinPoint.getSignature().getName();
        Class clazz = joinPoint.getTarget().getClass();
		if(clazz.isAnnotationPresent(DS.class)){
		    //获取类上注解
			return;
		} 
		
	    String targetName = clazz.getSimpleName();
	    Class[] parameterTypes = 
        ((MethodSignature)joinPoint.getSignature()).getMethod().getParameterTypes();
	    Method methdo = clazz.getMethod(methodName,parameterTypes);
	    if (methdo.isAnnotationPresent(DS.class)) {
	    	return;
	    }
		if (methodName.startsWith("get") 
                || methodName.startsWith("count") 
                || methodName.startsWith("find")
				|| methodName.startsWith("list") 
                || methodName.startsWith("select") 
                || methodName.startsWith("check")
				|| methodName.startsWith("page")) {

			log.info("当前执行的库:"+SLAVE);
			DynamicDataSourceContextHolder.push(SLAVE);
		} else {
			log.info("当前执行的库:"+MASTER);
			DynamicDataSourceContextHolder.push(MASTER);
		}
	}
	@After("checkArgs()")
    public void afterAdvice(){
        DynamicDataSourceContextHolder.clear();
    }
}

这样可以让你有@DS的注解依然生效,而且也会根据方法名来自动切换数据源。

© 著作权归作者所有

攻城狮-飞牛

攻城狮-飞牛

粉丝 53
博文 42
码字总数 44828
作品 1
成都
程序员
私信 提问
苞米豆-多数据源快速启动器 V1.4.0 发布

https://gitee.com/baomidou/dynamic-datasource-spring-boot-starter 欢迎关注项目,提出您的宝贵建议。 v1.4.0 升级日志 支持了在类上注解,方法上同时有注解方法优先。 支持的遇到事物强制...

小锅盖
2018/07/17
1K
4
苞米豆-多数据源 2.2.1 发布:全面适配 mybatis-plus 2.x 3.x

dynamic-datasource-spring-boot-starter v2.2.1 发布,更新如下: 适配mybatis2.x版本 简介 dynamic-datasource-spring-boot-starter 是一个基于springboot的快速集成多数据源的启动器。 其...

小锅盖
2018/09/04
2.4K
4
苞米豆多数据源启动器 2.0.1 发布,Bug 修复版本

苞米豆多数据源启动器 2.0.1 发布了。强烈建议升级!更新内容: 修复一个方法缓存的bug,会引起同名方法的注解失效。 底层代码的重命名和部分格式的调整。 源码地址: https://gitee.com/ba...

小锅盖
2018/08/09
778
4
苞米豆-多数据源启动器 2.0.2 发布:修复 sb2.0 集成 bug​​​​​​​

更新内容: 修复springboot2.0以上版本不能默认设置HikariDataSource,type为空抛异常。(原因是sb2.0更改了父类的代码的位置,为兼容springboot2.0采用反射实现,后期优化) 底层代码整理。 ...

小锅盖
2018/08/11
493
1
苞米豆-多数据源七夕特别版 2.1.0 发布:全面适配 mybatis-plus

v2.1.0 修复了底层一个逻辑bug。 提供了对mp的原生支持。 底层代码进行了细微的性能优化 https://gitee.com/baomidou/dynamic-datasource-spring-boot-starter 从2.1.0开始提供对mp的集成,在...

小锅盖
2018/08/18
1K
1

没有更多内容

加载失败,请刷新页面

加载更多

Vue.js学习笔记2 - better-scroll滚动条

better-scroll滚动条 使用作者自制的better-scroll库,实现内容的滚动。 先在package.json加上依赖: "better-scroll": "^0.1.7" 接着再npm install安装依赖。 import BScroll from 'better-......

swanf
今天
7
0
设计模式之适配器模式

定义 将一个类的接口变换成客户端所期待的另一种接口,从而使原本因接口不匹配而无法在一起工作的两个类能够在一起工 作。 UML类图 适配器分为两种,类适配器与对象适配器。 类适配器的UML图...

陈年之后是青葱
今天
8
0
教你玩转Linux—磁盘管理

导读 Linux磁盘管理好坏直接关系到整个系统的性能问题,Linux磁盘管理常用三个命令为df、du和fdisk。 df df命令参数功能:检查文件系统的磁盘空间占用情况。可以利用该命令来获取硬盘被占用了...

问题终结者
今天
13
0
KMP

字符串匹配算法 针对被匹配字段生产一个部分匹配表 A B C D A B D 0 0 0 0 1 2 0 部分匹配表 熟悉前缀与后缀的概念 ,“部分匹配表” 的生产就是根据前缀、后缀的最苍的共有元素的长度 前缀:...

鬼才王
昨天
6
0
快速搭建Jenkins集群

关于Jenkins集群 在Jenkins上同时执行多个任务时,单机性能可能达到瓶颈,使用Jenkins集群可以有效的解决此问题,让多台机器同时处理这些任务可以将压力分散,对单机版Jenkins的单点故障的隐...

程序员欣宸
昨天
13
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部