spring boot多数据源动态切换代码实例

下面将为您详细讲解如何实现在Spring Boot应用中实现多数据源动态切换，并提供两个示例。

一、前置条件

在开始编写代码之前，需要满足以下条件：

1. 确保已经正确配置了多个数据源，这些数据源需要连接的数据库表结构和数据内容都应当是相同的；
2. 当前应用中必须已经引入了相关依赖，这里采用Spring Boot 2.x版本为例：

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-jdbc</artifactId>
    </dependency>
    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-java</artifactId>
        <version>${mysql.version}</version>
    </dependency>
</dependencies>

二、实现代码

1. 配置多个数据源

在进行数据库数据查询操作时，需要根据请求参数动态切换不同的数据源。因此你需要在配置文件中为每个数据源添加一个前缀（例如：datasource.ds1 和 datasource.ds2），指定对应的数据库连接信息。

datasource:
  ds1:
    url: jdbc:mysql://localhost:3306/db1
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver
  ds2:
    url: jdbc:mysql://localhost:3306/db2
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver

2. 实现多数据源动态切换

在实现多数据源动态切换时，需要借助Spring Boot的AOP（面向切面编程）技术，在方法调用之前拦截请求并切换数据源。具体实现过程如下：

2.1 自定义注解

需要在自定义注解中指定要使用的数据源，这里定义了一个@DataSource注解来实现此功能。

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {

    String value() default "ds1";

}

2.2 切面实现

在使用自定义注解的方法上，使用AOP拦截器实现数据源的切换。

@Component
@Aspect
public class DataSourceAspect {

    @Around("@annotation(com.example.demo.annotation.DataSource)")
    public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();

        DataSource ds = method.getAnnotation(DataSource.class);
        if (ds == null) {
            DbContextHolder.setDB("ds1");
        } else {
            DbContextHolder.setDB(ds.value());
        }

        try {
            return joinPoint.proceed();
        } finally {
            DbContextHolder.clearDB();
        }
    }

}

2.3 数据源上下文实现

在实际使用中，需要通过一个ContextHolder类来存储当前要使用的数据源，以方便在程序运行时动态切换数据源。这里定义了一个DbContextHolder类来实现此功能。

public class DbContextHolder {

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

    public static void setDB(String dbType) {
        contextHolder.set(dbType);
    }

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

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

}

3. 示例代码

3.1 查询学生信息

@Service
public class StudentService {

    @Autowired
    JdbcTemplate jdbcTemplate;

    @DataSource("ds1")
    public List<Student> findStudentsFromDs1() {
        return jdbcTemplate.query("SELECT * FROM student", new BeanPropertyRowMapper<>(Student.class));
    }

    @DataSource("ds2")
    public List<Student> findStudentsFromDs2() {
        return jdbcTemplate.query("SELECT * FROM student", new BeanPropertyRowMapper<>(Student.class));
    }

}

3.2 查询教师信息

@Service
public class TeacherService {

    @Autowired
    JdbcTemplate jdbcTemplate;

    @DataSource("ds1")
    public List<Teacher> findTeachersFromDs1() {
        return jdbcTemplate.query("SELECT * FROM teacher", new BeanPropertyRowMapper<>(Teacher.class));
    }

    @DataSource("ds2")
    public List<Teacher> findTeachersFromDs2() {
        return jdbcTemplate.query("SELECT * FROM teacher", new BeanPropertyRowMapper<>(Teacher.class));
    }

}

三、测试代码

@SpringBootTest
@RunWith(SpringRunner.class)
public class ApplicationTests {

    @Autowired
    StudentService studentService;

    @Autowired
    TeacherService teacherService;

    @Test
    public void testFindStudents() {
        DbContextHolder.setDB("ds1");
        List<Student> students1 = studentService.findStudentsFromDs1();
        students1.forEach(System.out::println);

        System.out.println("----------");

        DbContextHolder.setDB("ds2");
        List<Student> students2 = studentService.findStudentsFromDs2();
        students2.forEach(System.out::println);
    }

    @Test
    public void testFindTeachers() {
        DbContextHolder.setDB("ds1");
        List<Teacher> teachers1 = teacherService.findTeachersFromDs1();
        teachers1.forEach(System.out::println);

        System.out.println("----------");

        DbContextHolder.setDB("ds2");
        List<Teacher> teachers2 = teacherService.findTeachersFromDs2();
        teachers2.forEach(System.out::println);
    }

}

四、总结

通过以上步骤，我们就实现了Spring Boot多数据源动态切换的功能，通过自定义注解和AOP的方式，实现了用户请求时数据库连接自动切换的操作。这样就可以很方便地实现多数据源连接操作，避免了多个数据源开发的麻烦。