Java 动态数据源切换的核心代码

Java动态数据源切换的核心代码分为两部分，一部分是切面类用于拦截方法并根据注解切换数据源，另一部分是数据源上下文类用于保存当前线程使用的数据源。

1. 切面类代码

@Aspect
@Component
public class DataSourceAspect {

    @Around("@annotation(dataSource)")
    public Object switchDataSource(ProceedingJoinPoint point, DataSource dataSource) throws Throwable {
        String dataSourceKey = dataSource.value();
        DataSourceContextHolder.setDataSource(dataSourceKey);
        try {
            return point.proceed();
        } finally {
            DataSourceContextHolder.clearDataSource();
        }
    }
}

在切面类中定义了`switchDataSource`方法，使用`@Around`注解标识需要拦截的方法，并根据方法上的`@DataSource`注解值设置当前线程使用的数据源。在方法执行完成后，清除当前数据源。

2. 数据源上下文类代码

public class DataSourceContextHolder {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

    public static String getDataSource() {
        return contextHolder.get();
    }

    public static void clearDataSource() {
        contextHolder.remove();
    }
}

在数据源上下文类中，使用`ThreadLocal`保存当前线程使用的数据源，提供了`setDataSource`、`getDataSource`、`clearDataSource`等方法用于设置、获取和清除当前数据源。

以上是Java动态数据源切换的核心代码，通过切面类和数据源上下文类的配合，可以实现在运行时动态切换数据源。