当前位置：首页 > 文章列表 > 文章 > java教程 > Java多数据源动态切换配置详解

Java多数据源动态切换配置详解

2025-07-28 18:41:51 0浏览收藏

本文详细介绍了如何在Java项目中实现多数据源动态切换配置，旨在帮助开发者应对复杂的数据库访问需求，提升系统灵活性和可维护性。首先，需要在Spring配置文件中定义多个数据库连接信息，并通过数据源配置类初始化各个数据源Bean。核心在于继承`AbstractRoutingDataSource`并重写`determineCurrentLookupKey()`方法，结合`ThreadLocal`实现动态数据源选择。此外，文章还阐述了如何配置`DynamicDataSource`、利用AOP自动切换数据源、手动控制数据源切换、配置Mybatis使用`DynamicDataSource`、确保事务一致性、处理连接失败以及在微服务架构中的应用场景，例如读写分离等。通过本教程，开发者可以掌握一套完整的Java多数据源动态切换方案，轻松应对多变的业务需求。

配置多个数据源：在Spring配置文件中定义多个数据库连接信息。2. 创建数据源配置类：使用@ConfigurationProperties初始化各个数据源Bean。3. 实现AbstractRoutingDataSource：继承该类并重写determineCurrentLookupKey()方法，结合ThreadLocal实现动态选择。4. 配置DynamicDataSource：将多个数据源注入到DynamicDataSource中，并设置默认数据源。5. 使用AOP切换数据源：通过自定义注解和切面实现自动切换。6. 手动切换数据源：调用setDataSource()和clearDataSource()进行手动控制。7. 配置Mybatis：确保SqlSessionFactory使用DynamicDataSource。8. 确保事务一致性：通过PlatformTransactionManager与@Transactional注解管理事务。9. 处理连接失败：添加健康检查、重试机制及日志记录。10. 应用于微服务架构：支持灵活访问不同数据库，适用于读写分离等场景。

Java实现多数据源动态切换的详细配置教程

Java实现多数据源动态切换，核心在于配置和管理多个数据源，并在运行时根据需要选择合适的数据源进行操作。这通常涉及Spring框架、AbstractRoutingDataSource以及ThreadLocal等技术。

解决方案：

配置多个数据源： 首先，在Spring配置文件（例如application.properties或application.yml）中定义多个数据源。每个数据源对应一个数据库连接配置，包括URL、用户名、密码、驱动类等。

spring.datasource.primary.url=jdbc:mysql://localhost:3306/db_primary?useSSL=false&serverTimezone=UTC
spring.datasource.primary.username=root
spring.datasource.primary.password=password
spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver

spring.datasource.secondary.url=jdbc:mysql://localhost:3306/db_secondary?useSSL=false&serverTimezone=UTC
spring.datasource.secondary.username=root
spring.datasource.secondary.password=password
spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver

创建数据源配置类： 创建一个Java配置类，用于初始化这些数据源Bean。

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import javax.sql.DataSource;

@Configuration
public class DataSourceConfig {

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

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

实现AbstractRoutingDataSource： 创建一个类继承AbstractRoutingDataSource，并重写determineCurrentLookupKey()方法。这个方法负责决定使用哪个数据源。

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {

    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();
    }

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

配置DynamicDataSource： 在配置类中配置DynamicDataSource，并将之前定义的数据源注入。

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

@Configuration
public class DynamicDataSourceConfig {

    @Bean
    public DynamicDataSource dynamicDataSource(@Qualifier("primaryDataSource") DataSource primaryDataSource,
                                              @Qualifier("secondaryDataSource") DataSource secondaryDataSource) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("primary", primaryDataSource);
        targetDataSources.put("secondary", secondaryDataSource);

        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.setTargetDataSources(targetDataSources);
        dynamicDataSource.setDefaultTargetDataSource(primaryDataSource); // 默认数据源
        return dynamicDataSource;
    }

    @Bean
    public PlatformTransactionManager transactionManager(DynamicDataSource dynamicDataSource) {
        return new DataSourceTransactionManager(dynamicDataSource);
    }
}

使用AOP切换数据源： (可选) 可以使用AOP来自动切换数据源，例如基于方法注解。

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

@Aspect
@Component
public class DataSourceAspect {

    @Before("@annotation(UsePrimaryDataSource)") // 自定义注解
    public void setPrimaryDataSource(JoinPoint joinPoint) {
        DynamicDataSource.setDataSource("primary");
    }

    @Before("@annotation(UseSecondaryDataSource)") // 自定义注解
    public void setSecondaryDataSource(JoinPoint joinPoint) {
        DynamicDataSource.setDataSource("secondary");
    }

    @After("@annotation(UsePrimaryDataSource) || @annotation(UseSecondaryDataSource)")
    public void restoreDataSource(JoinPoint joinPoint) {
        DynamicDataSource.clearDataSource();
    }
}

需要定义UsePrimaryDataSource和UseSecondaryDataSource注解。

手动切换数据源： 在需要切换数据源的地方，调用DynamicDataSource.setDataSource("dataSourceName")方法。完成后，调用DynamicDataSource.clearDataSource()清除ThreadLocal中的数据源信息。

配置Mybatis： 配置Mybatis使用DynamicDataSource。

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;

import javax.sql.DataSource;

@Configuration
public class MybatisConfig {

    @Autowired
    private DynamicDataSource dynamicDataSource;

    @Bean
    public SqlSessionFactory sqlSessionFactory() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(dynamicDataSource);
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        factoryBean.setMapperLocations(resolver.getResources("classpath:mapper/*.xml")); // mapper文件路径
        return factoryBean.getObject();
    }

    @Bean
    public SqlSessionTemplate sqlSessionTemplate(SqlSessionFactory sqlSessionFactory) throws Exception {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

如何确保数据源切换的事务一致性？

使用@Transactional注解结合PlatformTransactionManager可以管理事务。确保PlatformTransactionManager使用的是DynamicDataSource。在切换数据源前开启事务，切换数据源后，事务会自动应用到当前数据源。

如何优雅地处理数据源连接失败的情况？

在DynamicDataSource中，可以添加数据源健康检查机制。定期检查数据源的连接状态，如果连接失败，可以尝试重新连接或者切换到备用数据源。可以使用Spring Retry模块实现自动重试。此外，可以记录数据源连接失败的日志，方便排查问题。

多数据源动态切换在微服务架构中的应用场景？

在微服务架构中，每个服务可能需要访问不同的数据库。多数据源动态切换可以帮助服务灵活地访问不同的数据源，而无需修改代码。例如，一个服务可能需要访问订单数据库、用户数据库和商品数据库。通过多数据源动态切换，可以根据业务需求选择合适的数据源。此外，多数据源动态切换还可以用于读写分离，将读操作路由到只读数据库，将写操作路由到主数据库。

到这里，我们也就讲完了《Java多数据源动态切换配置详解》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于Spring,MyBatis,threadlocal,abstractroutingdatasource,多数据源动态切换的知识点！

Spring MyBatis threadlocal abstractroutingdatasource 多数据源动态切换