ShardingSphere数据源切换偶尔失效

快来打我* 2023-10-09 23:38 28阅读 0赞

#### 目录 ####

*   *  一、场景
     *   *  说明
     *  二、代码
     *   *  1、数据源管理类
         *  2、数据源切换AOP
         *  3、方法调用代码
     *  三、排查
     *  四、原因
     *   *  调用方法的代码
     *  五、解决
     *   *  改动后的调用方式

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

### 一、场景 ###

1、该项目使用多个数据源

2、通过AOP + 注解的方式切换数据源

3、绝大部分service方法调用时都能正常切换数据源

4、在调用`updateSessionInfo()`方法时数据源切换**偶尔失效【该方法在被SessionMsgServiceImpl调用到时，数据源就会切换失败】**

#### 说明 ####

**1、持有`updateSessionInfo()`方法的类就是`SessionMsgServiceImpl`，也就是说，在本类调用本类的`updateSessionInfo()`方法时，数据源切换就会失效**

**2、如果是其他类调用`updateSessionInfo()`则数据源切换是正常的**

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

### 二、代码 ###

#### 1、数据源管理类 ####

package com.xxx.datasource;
    
    import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    import javax.sql.DataSource;
    import java.util.Map;
    
    public class DynamicDataSource extends AbstractRoutingDataSource {
        
        private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
    
        public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
        
            super.setDefaultTargetDataSource(defaultTargetDataSource);
            super.setTargetDataSources(targetDataSources);
            super.afterPropertiesSet();
        }
    
        @Override
        protected Object determineCurrentLookupKey() {
        
            return getDataSource();
        }
    
        public static void setDataSource(String dataSource) {
        
            contextHolder.set(dataSource);
        }
    
        public static String getDataSource() {
        
            return contextHolder.get();
        }
    
        public static void clearDataSource() {
        
            contextHolder.remove();
        }
    }

#### 2、数据源切换AOP ####

package com.xxx.datasource;
    
    import org.aspectj.lang.ProceedingJoinPoint;
    import org.aspectj.lang.annotation.Around;
    import org.aspectj.lang.annotation.Aspect;
    import org.aspectj.lang.annotation.Pointcut;
    import org.aspectj.lang.reflect.MethodSignature;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.core.Ordered;
    import org.springframework.stereotype.Component;
    
    import java.lang.reflect.Method;
    
    @Aspect
    @Component
    public class DataSourceAspect implements Ordered {
        
        private static Logger logger = LoggerFactory.getLogger(DataSourceAspect.class);
    
        @Pointcut("@annotation(com.xxx.datasource.DataSource)")
        public void dataSourcePointCut() {
        
    
        }
    
        @Around("dataSourcePointCut()")
        public Object around(ProceedingJoinPoint point) throws Throwable {
        
            MethodSignature signature = (MethodSignature) point.getSignature();
            Method method = signature.getMethod();
    
            DataSource ds = method.getAnnotation(DataSource.class);
            if(ds == null){
        
                DynamicDataSource.setDataSource(DataSourceNames.MANAGER);
                logger.debug("set datasource is " + DataSourceNames.MANAGER);
            }else {
        
                DynamicDataSource.setDataSource(ds.name());
                logger.debug("set datasource is " + ds.name());
            }
    
            try {
        
                return point.proceed();
            } finally {
        
                DynamicDataSource.clearDataSource();
                logger.debug("clean datasource");
            }
        }
    
        @Override
        public int getOrder() {
        
            return -1;
        }
    }

#### 3、方法调用代码 ####

//注：该方法所在类为：SessionMsgServiceImpl
    updateSessionInfo(sessionInfo);

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

### 三、排查 ###

1、通过debug，发现在调用`updateSessionInfo()`时，**有时**会进入AOP进行数据源切换

1.1、即使进入AOP进行数据源切换，也不是每次都能切换成功【从debug看，确实有进行数据源的切换】

1.2、没有进入AOP进行数据源切换，则执行代码SQL必然报错【数据库找不到表】  
  
  
2、经过一整天的百度，使用以下方法试图解决问题，但都失败

2.1、在数据源切换AOP类上使用`@Order(-1)`，在事务开启前切换数据源

2.2、去除`@Transactional`，百度上说是事务引起的切换失效，去除事务，企图确认原因

2.3、将数据源切换AOP的`@Around`(环绕)，改为`@Before`(前置-注入数据源)和`@After`(后置-重置数据源)

2.4、为`updateSessionInfo()`再加一个切入点

@Pointcut("execution(* com.xxx.service.impl.SessionMsgServiceImpl.updateSessionInfo(..))")
        public void dataSourcePointCut2() {
        
    }
    
    //@Before注入数据源相关代码
    ......
    
    //@After重置数据源相关代码
    ......

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

### 四、原因 ###

在调用updateSessionInfo()时，没有注入该方法所在类的实例，调用时没有使用该实例来调用方法

#### 调用方法的代码 ####

//因为是在本类调用，所以这里没有指定调用者
    updateSessionInfo(sessionInfo);

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

### 五、解决 ###

注入该方法所在类的对象，使用该对象调用方法

#### 改动后的调用方式 ####

//1、先注入该方法持有类的实例
    @Autowired
    private SessionMsgService sessionMsgService;
    
    //2、使用该实例调用方法
    sessionMsgService.updateSessionInfo(sessionInfo);