spring实现动态切换、添加数据源及源码分析

Spring 实现动态切换、添加数据源及源码分析 Spring 框架提供了 AbstractRoutingDataSource 抽象类来实现动态切换、添加数据源，通过将多个数据源配置在一个 Bean 中，然后使用 determineTargetDataSource() 方法来确定当前使用的数据源。下面是关于 Spring 实现动态切换、添加数据源及源码分析的详细知识点： 一、数据源切换的必要性 在实际项目中，随着数据量的增加，单个 MySQL 数据库可能无法满足性能要求，需要对数据库进行水平切分，通过Hash 等方式将数据分布到多个数据库中。为了实现这种需求，需要在后端服务中实现动态切换、添加数据源的功能。 二、AbstractRoutingDataSource 抽象类 AbstractRoutingDataSource 是 Spring 框架提供的一个抽象类，用于实现动态切换、添加数据源。它提供了 determineTargetDataSource() 方法来确定当前使用的数据源，通过将多个数据源配置在一个 Bean 中，实现了动态切换、添加数据源的功能。 三、DynamicRoutingDataSource 实现动态切换、添加数据源 DynamicRoutingDataSource 是 AbstractRoutingDataSource 的一个实现类，用于实现动态切换、添加数据源。它提供了两个重要的方法：setTargetDataSources() 和 afterPropertiesSet()，前者用于设置多个数据源，后者用于解析数据源和初始化数据源。 四、determineTargetDataSource() 方法 determineTargetDataSource() 方法是 AbstractRoutingDataSource 的一个抽象方法，用于确定当前使用的数据源。它通过调用 determineCurrentLookupKey() 方法来获取当前的数据源 Key，然后从 resolvedDataSources 中获取对应的数据源。 五、determineCurrentLookupKey() 方法 determineCurrentLookupKey() 方法是 AbstractRoutingDataSource 的一个抽象方法，用于获取当前的数据源 Key。它需要用户实现，用于确定当前使用的数据源。 六、SpringBoot 实现动态切换、添加数据源 在 SpringBoot 项目中，可以使用 @Configuration 注解来配置多个数据源，然后使用 @Bean 注解来注册数据源。例如： ```java @Configuration @MapperScan("com.gitee.taven.mapper") public class DataSourceConfigurer { @Bean("db0") @Primary @ConfigurationProperties(prefix = "datasource.db0") public DataSource dataSource0() { return DruidDataSourceBuilder.create().build(); } @Bean("db1") @ConfigurationProperties(prefix = "datasource.db1") public DataSource dataSource1() { return DruidDataSourceBuilder.create().build(); } @Bean("dynamicDataSource") public DataSource dynamicDataSource() { DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource(); Map<Object, Object> dataSourceMap = new HashMap<>(); dataSourceMap.put("dynamic_db0", dataSource0()); dataSourceMap.put("dynamic_db1", dataSource1()); dynamicRoutingDataSource.setDefaultTargetDataSource(dataSource0()); dynamicRoutingDataSource.setTargetDataSources(dataSourceMap); return dynamicRoutingDataSource; } } ``` 七、源码分析 从源码中可以看到，AbstractRoutingDataSource 的afterPropertiesSet() 方法会调用 determineTargetDataSource() 方法来确定当前使用的数据源，然后从 resolvedDataSources 中获取对应的数据源。DynamicRoutingDataSource 的 setTargetDataSources() 方法用于设置多个数据源，afterPropertiesSet() 方法用于解析数据源和初始化数据源。 Spring 框架提供了 AbstractRoutingDataSource 抽象类来实现动态切换、添加数据源，通过将多个数据源配置在一个 Bean 中，然后使用 determineTargetDataSource() 方法来确定当前使用的数据源。