SpringBoot +DynamicDataSource切换多数据源的全过程

下面我就来详细讲解SpringBoot + DynamicDataSource切换多数据源的全过程。

1. 概述

在实际项目中，经常会遇到需要切换多数据源的情况，SpringBoot + DynamicDataSource可以很好地解决这个问题。本文将介绍如何使用SpringBoot + DynamicDataSource实现多数据源的切换过程。

2. 示例说明

2.1 示例1

首先，我们需要在pom.xml中添加DynamicDataSource的依赖：

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
    <version>2.5.3</version>
</dependency>

接着，我们需要在application.yml中配置数据源信息：

spring:
  datasource:
    driver-class-name: com.mysql.jdbc.Driver
    dynamic:
      datasource:
        primary:
          url: jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf-8&useSSL=false
          username: root
          password: root
          driver-class-name: com.mysql.jdbc.Driver
        secondary:
          url: jdbc:mysql://localhost:3306/test2?useUnicode=true&characterEncoding=utf-8&useSSL=false
          username: root
          password: root
          driver-class-name: com.mysql.jdbc.Driver

其中，primary和secondary是我们定义的两个数据源的名称。

接着，我们需要定义一个注解用来指定数据源：

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

然后，在切换数据源的时候，我们可以在Service层中使用该注解：

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper;

    @Override
    @DataSource("primary")
    public List<User> getUsersFromPrimaryDataSource() {
        return userMapper.getUsers();
    }

    @Override
    @DataSource("secondary")
    public List<User> getUsersFromSecondaryDataSource() {
        return userMapper.getUsers();
    }
}

在该示例中，我们定义了一个UserService接口，并使用@DataSource注解来指定不同的数据源。当调用getUsersFromPrimaryDataSource方法时，数据源会切换到primary，而当调用getUsersFromSecondaryDataSource方法时，数据源会切换到secondary。

2.2 示例2

在实际应用中，我们可能需要动态切换数据源。这时候，我们可以使用AOP来实现。

首先，我们需要定义一个AOP切面：

@Component
@Aspect
public class DataSourceAspect {
    @Pointcut("@annotation(com.example.demo.datasource.DataSource)")
    public void dsPointCut() {}

    @Before("dsPointCut()")
    public void beforeSwitchDS(JoinPoint joinPoint) {
        String dataSourceKey = String.valueOf(joinPoint.getSignature());
        if (dataSourceKey.contains("setPrimary")) {
            DataSourceContextHolder.setDataSourceKey("primary");
        } else {
            DataSourceContextHolder.setDataSourceKey("secondary");
        }
    }

    @After("dsPointCut()")
    public void afterSwitchDS(JoinPoint joinPoint) {
        DataSourceContextHolder.removeDataSourceKey();
    }
}

在该切面中，我们使用了@annotation注解来指定切入点，并在方法执行前后进行数据源的切换操作。具体来说，当方法名包含setPrimary时，数据源切换为primary，否则数据源切换为secondary。

接着，我们需要定义一个DataSourceContextHolder类，用来存储数据源的名称：

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

    public static void setDataSourceKey(String dataSourceKey) {
        DATA_SOURCE_KEY_HOLDER.set(dataSourceKey);
    }

    public static String getDataSourceKey() {
        return DATA_SOURCE_KEY_HOLDER.get();
    }

    public static void removeDataSourceKey() {
        DATA_SOURCE_KEY_HOLDER.remove();
    }
}

最后，在Service层中使用@DataSource注解来指定数据源：

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper;

    @Override
    @DataSource("primary")
    public void addUserToPrimaryDataSource(User user) {
        userMapper.insert(user);
    }

    @Override
    @DataSource("secondary")
    public void addUserToSecondaryDataSource(User user) {
        userMapper.insert(user);
    }
}

在该示例中，我们使用AOP来实现动态切换数据源。具体来说，在切入点方法执行前后，会根据方法名来设置数据源的名称，然后通过@DataSource注解来指定数据源。

3. 结论

使用SpringBoot + DynamicDataSource可以很方便地切换多数据源。在实际应用中，我们可以使用两种不同的方式来实现数据源的切换：通过@DataSource注解来指定数据源，或者使用AOP来动态切换数据源。