DynamicDatasource动态数据源实现及使用

在现代微服务架构中，动态数据源的管理显得尤为重要。应用程序常常需要根据不同的逻辑来切换数据源，尤其是在多租户的场景中，或者当同一应用需要同时访问多个数据库的时候，动态数据源为我们提供了便利。下面我们就来探讨一下如何实现和使用动态数据源。

一、动态数据源的基本概念

动态数据源是指能够在运行时根据业务需求动态切换数据源的机制。通常在Spring框架中，我们可以通过拦截器、AOP或自定义注解等方式来实现动态数据源的切换。这种机制能够极大地提高系统的灵活性和扩展性。

二、实现步骤

1.创建数据源配置类

首先，我们需要定义一个数据源配置类，用于定义数据源的信息。

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.lookup.JndiDataSourceLookup;

import javax.sql.DataSource;

@Configuration
public class DynamicDataSourceConfig {

    @Bean("dataSource1")
    @ConfigurationProperties("spring.datasource.datasource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }

    @Bean("dataSource2")
    @ConfigurationProperties("spring.datasource.datasource2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public DataSource dynamicDataSource(
            @Qualifier("dataSource1") DataSource dataSource1,
            @Qualifier("dataSource2") DataSource dataSource2) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("dataSource1", dataSource1);
        targetDataSources.put("dataSource2", dataSource2);

        AbstractRoutingDataSource dataSource = new AbstractRoutingDataSource() {
            @Override
            protected Object determineCurrentLookupKey() {
                return DataSourceContextHolder.getDataSourceType();
            }
        };
        dataSource.setTargetDataSources(targetDataSources);
        return dataSource;
    }
}

2.创建数据源上下文

接下来，我们需要创建一个上下文来存放当前线程的数据源类型。

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

    public static void setDataSourceType(String dataSourceType) {
        contextHolder.set(dataSourceType);
    }

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

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

3.创建切面进行数据源切换

然后，我们可以通过AOP来进行数据源的切换。

import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class DataSourceAspect {

    @Pointcut("@annotation(DataSourceSwitch)")
    public void dataSourcePointcut() {}

    @Before("dataSourcePointcut() && @annotation(dataSourceSwitch)")
    public void beforeSwitchDataSource(DataSourceSwitch dataSourceSwitch) {
        String dataSourceType = dataSourceSwitch.value();
        DataSourceContextHolder.setDataSourceType(dataSourceType);
    }
}

4.自定义注解

我们自定义一个注解，用于标记需要切换数据源的方法。

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

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

5.使用动态数据源

最后，我们可以在服务层中使用自定义注解来切换数据源。

@Service
public class UserService {

    @DataSourceSwitch("dataSource1")
    public User getUserFromDataSource1(Long id) {
        // 从数据源1获取用户信息的逻辑
    }

    @DataSourceSwitch("dataSource2")
    public User getUserFromDataSource2(Long id) {
        // 从数据源2获取用户信息的逻辑
    }
}

三、总结

通过上述步骤，我们实现了一个简单的动态数据源切换机制。使用自定义注解和AOP，我们能够在运行时根据业务需求灵活地切换数据源。这种方式不仅易于扩展，同时也保持了代码的整洁性。在实际应用中，可以进一步根据具体业务需求来优化和调整。