温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

baomidou中怎么利用dynamic-datasource实现读写分离

发布时间:2021-08-09 11:38:31 来源:亿速云 阅读:184 作者:Leah 栏目:大数据

这篇文章给大家介绍baomidou中怎么利用dynamic-datasource实现读写分离,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。

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();
    }
}

关于baomidou中怎么利用dynamic-datasource实现读写分离就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI