java动态切换数据库mybatis

原创

mob64ca12e2442a 2024-03-08 03:11:39 ©著作权

文章标签 数据源 spring Source 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e2442a的原创作品，请联系作者获取转载授权，否则将追究法律责任

动态切换数据库实现mybatis

在实际开发中，有时候我们需要动态切换数据库连接，比如在多租户系统中，不同租户的数据需要连接不同的数据库。在使用MyBatis这样的ORM框架时，我们可以通过配置动态数据源来实现数据库的动态切换。

配置多数据源

首先，我们需要在Spring Boot项目的配置文件中配置多个数据源，例如在application.properties文件中配置两个数据源：

# 数据源1
spring.datasource.url1=jdbc:mysql://localhost:3306/db1
spring.datasource.username1=root
spring.datasource.password1=123456
spring.datasource.driver-class-name1=com.mysql.cj.jdbc.Driver

# 数据源2
spring.datasource.url2=jdbc:mysql://localhost:3306/db2
spring.datasource.username2=root
spring.datasource.password2=123456
spring.datasource.driver-class-name2=com.mysql.cj.jdbc.Driver

动态数据源切换

接下来，我们需要实现一个动态数据源切换的工具类DynamicDataSource，继承AbstractRoutingDataSource，并重写determineCurrentLookupKey方法：

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }
}

数据源上下文

为了实现动态切换数据源，我们还需要一个数据源上下文DataSourceContextHolder类，用于保存和获取当前数据源的标识：

public class DataSourceContextHolder {

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

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

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

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

数据源切换注解

为了在代码中方便地切换数据源，可以定义一个注解DataSource：

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {

    String value();
}

在需要切换数据源的Service方法上添加@DataSource注解，传入对应的数据源标识。

切面拦截

最后，通过AOP技术，实现一个切面DataSourceAspect，在方法执行前根据注解的值切换数据源：

@Aspect
@Component
public class DataSourceAspect {

    @Before("@annotation(dataSource)")
    public void switchDataSource(JoinPoint joinPoint, DataSource dataSource) {
        DataSourceContextHolder.setDataSource(dataSource.value());
    }

    @After("@annotation(dataSource)")
    public void restoreDataSource(JoinPoint joinPoint, DataSource dataSource) {
        DataSourceContextHolder.clearDataSource();
    }
}

现在，当我们在Service方法上使用@DataSource注解指定数据源后，就可以实现动态切换数据库连接了。