SpringBoot AOP+注解方式实现多数据源切换可能遇到的问题

た入场券 2024-03-16 23:43 42阅读 0赞

#### 文章目录 ####

*  AOP+注解方式实现多数据源原理
 *  可能遇到的问题
 *   *  情景1：指定数据源的请求中发生报错
     *  情景2：指定数据源的请求中使用新的线程

--------------------

## AOP+注解方式实现多数据源原理 ##

通过ThreadLocal的线程隔离性将设线程与数据源ID进行绑定：

*  若不设置则使用默认数据源
 *  若设置则使用该数据源ID对应的数据源（注意：使用完后需要清除该数据源ID）

## 可能遇到的问题 ##

### 情景1：指定数据源的请求中发生报错 ###

问题描述：指定数据源的请求中发生报错，后面未指定数据源的请求却使用了该指定数据源（应该使用默认数据源）。

代码：

@Slf4j
    @Aspect
    @Order(-2)
    @Component
    public class DataSourceAspect {
        
    
        @Pointcut("@annotation(com.joker.datasource.aopannotation.DataSource) || @within(com.joker.datasource.aopannotation.DataSource)")
        public void run(){
        
    
        }
    
        @Around("run()")
        public Object around(ProceedingJoinPoint point) throws Throwable {
        
            MethodSignature signature = (MethodSignature) point.getSignature();
            // 方法上获取
            // AnnotatedElementUtils.hasAnnotation()
            DataSource dataSource = AnnotationUtils.findAnnotation(signature.getMethod(), DataSource.class);
            if (Objects.isNull(dataSource)) {
        
                // 类上获取
                dataSource =  AnnotationUtils.findAnnotation(signature.getDeclaringType(), DataSource.class);
            }
            // 设置数据源
            DynamicDataSourceContextHolder.setDataSourceType(dataSource.value().name());
            Object obj = point.proceed();
            // 清除数据源
            DynamicDataSourceContextHolder.clearDataSourceType();
            return obj;
        }
    }

问题分析：

1.  指定数据源的请求中发生报错，导致清除数据源ID未执行，当前线程thread1仍然绑定了该数据源；
2.  由于接口请求使用的线程是通过线程池来管理的，后续该线程thread1可能会继续分配给其它请求使用
3.  如果后续请求使用了该线程thread1且未指定数据源，使用的仍然是前面绑定的数据源，导致使用错数据源（原本应该使用默认数据源）
4.  但如果后续请求使用了该线程thread1但指定了数据源，则不会有问题

### 情景2：指定数据源的请求中使用新的线程 ###

问题描述：指定数据源的请求中使用新的线程，导致指定数据源无效，使用的是默认数据源。

问题分析：因为数据源是和线程绑定的，即使在当前线程绑定了指定数据源，但如果在请求中使用了新的线程，新线程是没有绑定数据源的（默认使用默认数据源）。使用新线程的一下场景：

1.  使用了new Thread()创建的新线程，在新线程中使用数据源
2.  使用了线程池，通过线程池中的线程来使用数据源
3.  使用了Java8中list.parallelStream()来并行处理（多线程处理），处理过程中使用数据