Spring Boot + Spring JPA + JDBC + Druid实现动态数据源切换

最新推荐文章于 2024-08-06 10:17:52 发布

AprChell

最新推荐文章于 2024-08-06 10:17:52 发布

阅读量805

点赞数

分类专栏：代码片段文章标签： spring java 数据库

本文链接：https://blog.csdn.net/ChiZng/article/details/113700060

版权

代码片段专栏收录该内容

2 篇文章 0 订阅

订阅专栏

Spring Boot + Spring JPA + JDBC + Druid实现动态数据源切换

AbstractRoutingDataSource源码分析

AllDataSourcesExecuteSQLDevice

application.properties

参考：https://blog.csdn.net/aiyo92/article/details/101064167#comments_12008727

最后：

总结：

背景：公司开发一个系统，需要满足根据传入的数据库信息和sql，对应的去该库中执行该sql。

关键点：spring-jdbc的AbstractRoutingDataSource抽象类。

AbstractRoutingDataSource源码分析

属性：

targetDataSources: 目标数据源
defaultTargetDataSource: 默认目标数据源
resolvedDataSources: 存储是数据库标识和数据源的映射关系

方法：

determineCurrentLookupKey()
setTargetDataSources()
setDefaultTargetDataSource()

分析：

先看关系，AbstractRoutingDataSource继承AbstractDataSource，AbstractDataSource实现DataSource，既然是实现DataSource接口那就应该有getConnection()这个方法，一看果然有

@Override
public Connection getConnection() throws SQLException {
    return determineTargetDataSource().getConnection();
}

@Override
public Connection getConnection(String username, String password) throws SQLException {
    return determineTargetDataSource().getConnection(username, password);
}

getConnection()方法内容也很简单,determineTargetDataSource().getConnection();determineTargetDataSource(): 看名知意，确定目标数据DataSource，再从DataSource中获取Connection。

下面看determineTargetDataSource()方法：

/**
* Retrieve the current target DataSource. Determines the
* {@link #determineCurrentLookupKey() current lookup key}, performs
* a lookup in the {@link #setTargetDataSources targetDataSources} map,
* falls back to the specified
* {@link #setDefaultTargetDataSource default target DataSource} if necessary.
* @see #determineCurrentLookupKey()
*/
protected DataSource determineTargetDataSource() {
    // resolvedDataSources数据源是否为空
    Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
    // 确定当前数据源的key
    Object lookupKey = determineCurrentLookupKey();
    // 从resolvedDataSources集合中根据key获取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 + "]");
    }
    return dataSource;
}

关键代码是determineCurrentLookupKey()方法：

/**
* Determine the current lookup key. This will typically be
* implemented to check a thread-bound transaction context.
* <p>Allows for arbitrary keys. The returned key needs
* to match the stored lookup key type, as resolved by the
* {@link #resolveSpecifiedLookupKey} method.
*/
@Nullable
protected abstract Object determineCurrentLookupKey();

这是一个抽象方法，翻译一下注释：确定当前的lookupKey。通常将其实现为检查线程绑定事务的上下文。允许任意key。返回的key需要匹配由resolveSpecifiedLookupKey()方法存储的lookup key类型。