java 配置多数据源的核心做法是定义多个 datasource bean,并通过 spring 的 @primary 注解或自定义路由机制(如继承 abstractroutingdatasource)将它们区分开来。具体实现时,需要为每个数据源创建独立的连接池实例(如 hikaricp),并在 service 层通过 transactionmanager 绑定特定的数据源,从而确保业务逻辑能准确访问目标数据库。

如何定义并注册多个独立的数据源 Bean
在 Spring Boot 应用中,首先需要从配置文件(如 application.yml)中读取不同数据库的连接信息,并将其映射为具体的 DataSource Bean。通常使用 HikariDataSource 作为底层连接池,因为它性能优异且配置简单。每个数据源 Bean 必须拥有唯一的名称,以便后续注入和路由时使用。

import com.zaxxer.hikari.HikariDataSource; import org.springframework.boot.context.properties.ConfigurationProperties; 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 MultiDataSourceConfig { // 主数据源配置 @Bean(name = "primaryDataSource") @Primary @ConfigurationProperties(prefix = "spring.datasource.primary") public DataSource primaryDataSource() { return new HikariDataSource(); } // 从数据源配置 @Bean(name = "secondaryDataSource") @ConfigurationProperties(prefix = "spring.datasource.secondary") public DataSource secondaryDataSource() { return new HikariDataSource(); } }上述代码展示了如何定义两个数据源:primaryDataSource 和 secondaryDataSource。@Primary 注解标记了默认使用的数据源,当 Spring 需要自动注入 DataSource 但未指定具体 Bean 名称时,将优先使用此数据源。配置文件中的 spring.datasource.primary 和 spring.datasource.secondary 前缀分别对应不同的 JDBC URL、用户名和密码。这种配置方式确保了每个数据源拥有独立的连接池实例,互不干扰。
立即学习“Java免费学习笔记(深入)”;
如何实现动态数据源路由与事务管理
定义了多个数据源后,关键步骤是实现动态路由,使应用能根据业务需求选择正确的数据源。Spring 提供了 AbstractRoutingDataSource 抽象类,通过重写 determineCurrentLookupKey() 方法,可以基于当前线程上下文(如用户 ID、租户 ID 或业务类型)动态返回目标数据源的 Key。同时,为了支持分布式事务或多数据源事务,需要为每个数据源配置独立的 PlatformTransactionManager,或者使用 JtaTransactionManager 进行全局事务管理。

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; import org.springframework.transaction.PlatformTransactionManager; import org.springframework.transaction.annotation.EnableTransactionManagement; import org.springframework.transaction.annotation.TransactionManagementConfigurer; import javax.sql.DataSource; import java.util.HashMap; import java.util.Map; @EnableTransactionManagement public class DynamicDataSource extends AbstractRoutingDataSource implements TransactionManagementConfigurer { private static final ThreadLocal<String> contextHolder = new ThreadLocal<>(); @Override protected Object determineCurrentLookupKey() { return contextHolder.get(); } public void setDataSource(String dataSource) { contextHolder.set(dataSource); } public static void clearDataSource() { contextHolder.remove(); } // 为动态数据源提供事务管理器 @Override public PlatformTransactionManager annotationDrivenTransactionManager() { return new DataSourceTransactionManager(primaryDataSource()); } }该示例代码实现了 AbstractRoutingDataSource,并通过 ThreadLocal 存储当前请求的目标数据源 Key。在 Service 层调用前,需先执行 dynamicDataSource.setDataSource("secondaryDataSource") 设置路由键,业务逻辑执行完毕后调用 clearDataSource() 清理上下文,防止内存泄漏和路由错误。对于事务管理,若仅涉及单个数据源的事务,可使用对应的 DataSourceTransactionManager;若涉及跨库事务,则需引入 JTA 实现。注意,annotationDrivenTransactionManager() 方法返回的是默认事务管理器,实际使用中可能需要为不同数据源配置多个事务管理器 Bean,并通过 @Transactional 的 value 属性指定具体事务管理器名称。
在 Service 层如何切换并使用特定数据源
在 Service 层使用多数据源时,最直接的方式是通过 AOP 切面或手动调用路由器的 setDataSource() 方法进行切换。例如,在查询从库的业务方法上添加自定义注解 @SecondaryDB,切面拦截该方法并设置路由键为 "secondaryDataSource"。这种方式解耦了业务逻辑与数据源选择逻辑,提高了代码的可维护性。同时,需确保在事务边界内完成数据源切换,避免事务启动后切换数据源导致事务失效或数据不一致。
import org.aspectj.lang.annotation.Aspect; import org.aspectj.lang.annotation.Before; import org.springframework.stereotype.Component; @Aspect @Component public class DataSourceSwitchAspect { @Autowired private DynamicDataSource dynamicDataSource; @Before("@annotation(com.example.SecondaryDB)") public void switchToSecondary() { dynamicDataSource.setDataSource("secondaryDataSource"); } @After("@annotation(com.example.SecondaryDB)") public void restorePrimary() { dynamicDataSource.clearDataSource(); } }上述 AOP 切面代码演示了如何在方法执行前后自动切换数据源。当带有 @SecondaryDB 注解的方法被调用时,切面会在方法开始前将数据源切换到 secondaryDataSource,并在方法结束后恢复默认状态。这种机制确保了数据源切换的原子性和安全性。在实际开发中,建议结合 @Transactional 注解使用,确保事务和数据源的一致性。例如,在 @SecondaryDB 方法上同时添加 @Transactional,Spring 会根据当前数据源选择相应的事务管理器,保证事务的正确提交或回滚。


















