baomidou的dynamic-datasource讀寫分離實作和加入AOP根據方法名選擇庫

1. <dependency>
2. <groupId>com.baomidou</groupId>
3. <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
4. <version>2.5.7</version>
5. </dependency>

純讀寫分離(mybatis環境)

場景：

1. 在純的讀寫分離環境，寫操作全部是master，讀操作全部是slave。
2. 不想通過注解配置完成以上功能。

答：在mybatis環境下可以基于mybatis插件結合本資料源完成以上功能。 手動注入插件。

1. @Bean
2. public MasterSlaveAutoRoutingPlugin masterSlaveAutoRoutingPlugin(){
3. return new MasterSlaveAutoRoutingPlugin();
4. }

預設主庫名稱master,從庫名稱slave。

問題

       我在配置好了之後，調試發現對資料庫讀的操作不得進入MasterSlaveAutoRoutingPlugin，而且進入了預設的庫。隻有寫進入了MasterSlaveAutoRoutingPlugin中。當然也可以預設為從庫，但是感覺就不是很好。

       于是我自定義了一個aop切面來，來完成庫的選擇，代碼如下：

1. import java.lang.reflect.Method;
2. import org.aspectj.lang.JoinPoint;
3. import org.aspectj.lang.annotation.After;
4. import org.aspectj.lang.annotation.Aspect;
5. import org.aspectj.lang.annotation.Before;
6. import org.aspectj.lang.annotation.Pointcut;
7. import org.aspectj.lang.reflect.MethodSignature;
8. import org.springframework.context.annotation.Lazy;
9. import org.springframework.core.annotation.Order;
10. import org.springframework.stereotype.Component;
11. import com.baomidou.dynamic.datasource.annotation.DS;
12. import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
13. import lombok.extern.java.Log;
14. /**
15. * Copyright: Copyright (c) 2019
16. * <p> 說明：動态資料源配置 </P>
17. *
18. * @version: V1.0
19. * @author: BianPeng
21. */
22. @Aspect
23. @Component
24. @Order(0)
25. @Lazy(false)
26. @Log
27. public class DataSourceAop{
28. private static final String MASTER = "master";
29. private static final String SLAVE = "slave";
30. @Pointcut("execution(* com.buybit.power.service..*.*(..)) || execution(* com.baomidou.mybatisplus.extension.service..*.*(..))")
31. public void checkArgs() {
33. // 這裡切到你的方法目錄
34. @Before("checkArgs()")
35. public void process(JoinPoint joinPoint) throws NoSuchMethodException, SecurityException {
36. String methodName = joinPoint.getSignature().getName();
37. if (methodName.startsWith("get")
38. || methodName.startsWith("count")
39. || methodName.startsWith("find")
40. || methodName.startsWith("list")
41. || methodName.startsWith("select")
42. || methodName.startsWith("check")
43. || methodName.startsWith("page")) {
44. log.info("目前執行的庫："+SLAVE);
45. DynamicDataSourceContextHolder.push(SLAVE);
46. } else {
47. log.info("目前執行的庫："+MASTER);
48. DynamicDataSourceContextHolder.push(MASTER);
51. @After("checkArgs()")
52. public void afterAdvice(){
53. DynamicDataSourceContextHolder.clear();

但是發現，​​baomidou/dynamic-datasource​​自帶的@DS沒失去了着用，于是我把有@DS的類和方法排除掉，代碼入下：

37. Class clazz = joinPoint.getTarget().getClass();
38. if(clazz.isAnnotationPresent(DS.class)){
39. //擷取類上注解
40. return;
42. String targetName = clazz.getSimpleName();
43. Class[] parameterTypes =
44. ((MethodSignature)joinPoint.getSignature()).getMethod().getParameterTypes();
45. Method methdo = clazz.getMethod(methodName,parameterTypes);
46. if (methdo.isAnnotationPresent(DS.class)) {

這樣可以讓你有@DS的注解依然生效，而且也會根據方法名來自動切換資料源。