Spring Boot 动态数据源

Spring Boot 动态数据源

• 1.前言
• 2.实现思路
• 3.项目实战
• • 3.1 项目初始化
  • 3.2 配置
  • 3.3 配置类
  • 3.4 加载数据源
  • 3.5 切换数据源
  • 3.6 标记数据源
  • 3.7 解析自定义注解
  • 3.8 动态使用数据源
  • 3.9 注解Bean
• 4. 测试
• • 4.1 数据库准备
  • 4.2 dao
  • 4.2 service
  • 4.3 controller
  • 4.4 执行测试

利用AOP来实现多数据源的动态切换功能。

1.前言

AbstractRoutingDataSource是Spring2.0.1版本引入的一个抽象类，它提供了多数据源的支持能力。AbstractRoutingDataSource抽象类定义了抽象的determineCurrentLookupKey方法，子类只需实现此方法，进而动态确定要使用的数据源。

AbstractRoutingDataSource

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
    @Nullable
    private Map targetDataSources;
    @Nullable
    private Object defaultTargetDataSource;
    private boolean lenientFallback = true;
    private DataSourceLookup dataSourceLookup = new JndiDataSourceLookup();
    @Nullable
    private Map resolvedDataSources;
    @Nullable
    private DataSource resolvedDefaultDataSource;
    
    ......

    protected DataSource determineTargetDataSource() {
        Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
        Object lookupKey = this.determineCurrentLookupKey();
        DataSource dataSource = (DataSource)this.resolvedDataSources.get(lookupKey);
        if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
            dataSource = this.resolvedDefaultDataSource;
        }

        if (dataSource == null) {
            throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
        } else {
            return dataSource;
        }
    }

    @Nullable
    protected abstract Object determineCurrentLookupKey();
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

1. 调用determineCurrentLookupKey方法来获取数据源名称key
2. 从resolvedDataSources属性中得到对应的DataSource对象。
3. 如果找不到DataSource对象或者数据源名称key不存在则使用resolvedDefaultDataSource。

2.实现思路

1. 提前准备好多个数据源
2. 将其存入一个Map中 （Map的Key是对应数据源的名称，而Value则是对应的数据源）
3. 将Map设置到AbstractRoutingDataSource对象的resolvedDataSources属性中
4. 当执行数据库操作的时候就通过一个Key来从Map中获取对应的数据源实例
5. 执行对应的数据库操作

3.项目实战

3.1 项目初始化

新建一个spring的项目：multi-source

    4.0.0
    
        org.springframework.boot
        spring-boot-starter-parent
        2.7.0
         
    
    cn.flowboot
    multi-source
    0.0.1-SNAPSHOT
    multi-source
    multi-source
    
        1.8
        
        3.5.2
  
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19