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

还在为Java项目中多数据源管理烦恼吗？本教程将带你一步步掌握Java多数据源动态切换配置，轻松应对复杂业务场景。我们将深入讲解如何在Spring框架下，利用AbstractRoutingDataSource和ThreadLocal技术，实现运行时数据源的灵活选择。内容涵盖：**配置多个数据源**、**创建数据源配置类**、**实现DynamicDataSource**、**AOP自动切换**、**手动控制**、**Mybatis集成**、**事务一致性保障**以及**连接失败处理**等关键环节。无论你是需要实现读写分离，还是在微服务架构中灵活访问不同数据库，本教程都能为你提供清晰、实用的指导，助你构建高效稳定的多数据源应用。立即学习，告别数据源切换难题！

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

解决方案：

1. 配置多个数据源： 首先，在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

2. 创建数据源配置类： 创建一个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();
       }
   }

3. 实现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();
       }
   }

4. 配置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);
       }
   }

5. 使用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注解。

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

7. 配置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学习网公众号，给大家分享更多文章知识！