@DS注解实现动态数据源切换

动态数据源切换是指在程序运行时根据需要切换数据源。实现动态数据源切换可以灵活地处理不同数据源的读写操作，提高系统的可扩展性和可维护性。下面介绍一种基于注解的实现方式。

首先，定义一个注解类`@DataSource`，用于标识需要切换的数据源：

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
    String value() default "default";
}

其中，`value`属性用于指定数据源的名称。如果不指定，则使用默认数据源。

接着，定义一个数据源切换的切面类`DataSourceAspect`，用于切换数据源：

@Aspect
@Component
public class DataSourceAspect {
    @Pointcut("@within(com.example.datasource.annotation.DataSource) || @annotation(com.example.datasource.annotation.DataSource)")
    public void dataSourcePointCut() {
    }

    @Around("dataSourcePointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        DataSource dataSource = method.getAnnotation(DataSource.class);
        if (dataSource == null) {
            dataSource = method.getDeclaringClass().getAnnotation(DataSource.class);
        }
        if (dataSource != null) {
            DynamicDataSourceContextHolder.setDataSource(dataSource.value());
        }
        try {
            return point.proceed();
        } finally {
            DynamicDataSourceContextHolder.clearDataSource();
        }
    }
}

在该切面类中，使用`@Pointcut`注解定义一个切点`dataSourcePointCut`，用于匹配所有被`@DataSource`注解标识的类或方法。在`around`方法中，通过反射获取方法上的`@DataSource`注解，并根据注解中的数据源名称切换数据源。在方法执行完毕后，清空数据源上下文。

最后，定义一个数据源上下文类`DynamicDataSourceContextHolder`，用于保存当前数据源的名称：

public class DynamicDataSourceContextHolder {
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

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

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

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

在需要切换数据源的类或方法上，添加`@DataSource`注解即可：

@Service
@DataSource("db1")
public class UserServiceImpl implements UserService {
    @Override
    public User getUserById(Long id) {
        return userDao.getUserById(id);
    }
}

上述代码中，`UserServiceImpl`类上的`@DataSource("db1")`表示使用名为`db1`的数据源。在调用`getUserById`方法时，切换到`db1`数据源进行操作。

以上就是基于注解实现动态数据源切换的方法。这种实现方式比较简单，但需要手动在需要切换数据源的类或方法上添加注解。如果需要更加自动化地切换数据源，可以考虑使用AOP或动态代理来实现。