Java注解实现动态数据源切换的实例代码

我们来分步骤详细讲解“Java注解实现动态数据源切换的实例代码”的完整攻略。

步骤一：创建动态数据源切换注解

我们需要创建一个自定义的注解来实现动态数据源切换。注解的结构如下：

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DS {
    String value() default "default";
}

该注解有一个 value 属性，用于设置数据源名称，默认值为 "default"。我们将在需要进行数据源切换的方法上添加该注解。

步骤二：创建动态数据源切换拦截器

接下来，我们需要创建一个拦截器，用于拦截被 @DS 注解标注的方法，以实现数据源的动态切换。代码如下所示：

@Aspect
@Component
public class DynamicDataSourceAspect {
    @Around("@annotation(ds)")
    public Object switchDataSource(ProceedingJoinPoint point, DS ds) throws Throwable {
        String dataSourceName = ds.value();
        DynamicDataSourceContextHolder.setDataSource(dataSourceName);
        try {
            return point.proceed();
        } finally {
            DynamicDataSourceContextHolder.clearDataSource();
        }
    }
}

该拦截器使用 @Aspect 注解标识，并在 @Around 注解中指定需要拦截的方法及其参数。在方法执行前，通过 setDataSource() 方法将需要使用的数据源名称存入 DynamicDataSourceContextHolder 中，方法执行完毕后再通过 clearDataSource() 方法清空存储的数据源名称。

需要注意的是，DynamicDataSourceContextHolder 是一个线程本地变量，用于保存当前线程的数据源名称。

步骤三：创建数据源配置类

接下来，我们需要创建数据源配置类，用于动态生成多个数据源，并将其注册到 Spring 容器中。代码如下所示：

@Configuration
public class DataSourceConfig {
    @Primary
    @Bean(name = "defaultDataSource")
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource defaultDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "testDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.test")
    public DataSource testDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "prodDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.prod")
    public DataSource prodDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "dynamicDataSource")
    public DynamicDataSource dataSource() {
        Map<Object, Object> dataSourceMap = new HashMap<>();
        dataSourceMap.put("default", defaultDataSource());
        dataSourceMap.put("test", testDataSource());
        dataSourceMap.put("prod", prodDataSource());

        DynamicDataSource dataSource = new DynamicDataSource();
        dataSource.setTargetDataSources(dataSourceMap);
        dataSource.setDefaultTargetDataSource(defaultDataSource());
        return dataSource;
    }
}

该配置类创建了三个数据源（defaultDataSource、testDataSource、prodDataSource），并将这些数据源存入 DynamicDataSource 中。需要注意的是，defaultDataSource 是使用 @Primary 注解标记的，这意味着如果没有注释 DS 注释的方法，则它将是默认选择的数据源。

步骤四：创建数据源上下文持有者

最后，我们需要创建一个数据源上下文持有者，用于存储当前线程使用的数据源名称。代码如下所示：

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

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

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

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

该类包含三个静态方法，分别用于设置、获取和清除当前线程的数据源名称。

示例一：动态切换默认数据源

假设我们有一个使用默认数据源的 DAO 层方法，我们可以直接在方法上添加 @DS 注解，如下所示：

public interface UserDao {
    @DS
    List<User> getAllUsers();
}

这样，在调用 getAllUsers() 方法时，就会使用默认数据源。

示例二：动态切换其他数据源

假设我们有一个使用测试数据源的 DAO 层方法，我们可以在方法上添加 @DS 注解，并将其值设置为 "test"，如下所示：

public interface UserDao {
    @DS("test")
    List<User> getTestUsers();
}

这样，在调用 getTestUsers() 方法时，就会使用测试数据源。

好了，这样就实现了动态数据源切换。希望这个攻略可以帮助你。