配置多个数据源：在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等技术。
解决方案：
-
配置多个数据源： 首先，在Spring配置文件（例如
application.properties或application.yml）中定义多个数据源。每个数据源对应一个数据库连接配置，包括URL、用户名、密码、驱动类等。立即学习“Java免费学习笔记（深入）”；
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 ThreadLocalcontextHolder = 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 -
使用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模块实现自动重试。 此外，可以记录数据源连接失败的日志，方便排查问题。
多数据源动态切换在微服务架构中的应用场景？
在微服务架构中，每个服务可能需要访问不同的数据库。 多数据源动态切换可以帮助服务灵活地访问不同的数据源，而无需修改代码。 例如，一个服务可能需要访问订单数据库、用户数据库和商品数据库。 通过多数据源动态切换，可以根据业务需求选择合适的数据源。 此外，多数据源动态切换还可以用于读写分离，将读操作路由到只读数据库，将写操作路由到主数据库。
