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

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

2025-07-06 16:00:40 0浏览收藏

本文深入解析了Java多数据源动态切换配置的实现方案，旨在帮助开发者灵活访问不同数据库，提升系统性能和可维护性。文章详细阐述了如何通过Spring框架、`AbstractRoutingDataSource`和`ThreadLocal`等技术，实现数据源的动态选择和管理。具体内容包括：配置多个数据源、创建数据源配置类、实现`AbstractRoutingDataSource`、配置`DynamicDataSource`、使用AOP或手动方式切换数据源，以及配置Mybatis以支持动态数据源。此外，还探讨了如何确保事务一致性、处理连接失败，以及多数据源动态切换在微服务架构中的应用，如读写分离等场景。通过本文，读者将全面掌握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学习网公众号！

java Spring 动态切换 abstractroutingdatasource 多数据源