Spring AOP 实现基于业务逻辑的动态数据源切换

DynamicDataSource 路由策略

继承 AbstractRoutingDataSource，重写 determineCurrentLookupKey 方法以获取当前线程的数据源标识。

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
import cn.mayongfa.common.JdbcContextHolder;

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return JdbcContextHolder.getJdbcType();
    }
}

枚举与注解定义

通过枚举明确数据源类型，避免字符串硬编码。

public enum DataSourceType {
    Master("master"),
    Slave("slave");

    private final String name;

    DataSourceType(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }
}

自定义注解用于标记方法所需的数据源，默认值为 Master。

import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
@Documented
public @interface DataSource {
    DataSourceType value() default DataSourceType.Master;
}

切面拦截逻辑

通过 AOP 拦截带有 @DataSource 注解的方法，在方法执行前设置数据源上下文。

import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.reflect.MethodSignature;
import cn.mayongfa.common.JdbcContextHolder;

public class DataSourceChoose {

    public void before(JoinPoint point) {
        Object target = point.getTarget();
        String methodName = point.getSignature().getName();
        Class<?>[] interfaces = target.getClass().getInterfaces();
        MethodSignature methodSignature = (MethodSignature) point.getSignature();
        Class<?>[] parameterTypes = methodSignature.getMethod().getParameterTypes();

        try {
            if (interfaces.length > 0) {
                Method m = interfaces[0].getMethod(methodName, parameterTypes);
                if (m != null && m.isAnnotationPresent(DataSource.class)) {
                    DataSource data = m.getAnnotation(DataSource.class);
                    JdbcContextHolder.clearJdbcType();
                    JdbcContextHolder.setJdbcType(data.value().getName());
                }
            }
        } catch (Exception e) {
            // 处理异常，建议记录日志
        }
    }
}

注意：上述反射逻辑假设目标对象实现了接口。若直接操作类，需调整 getInterfaces() 为 getMethods() 遍历查找。实际运行时请确保注解位置正确，并在切面配置中引用该拦截器。