欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

spring-jdbc之AbstractRoutingDataSource源码解析

程序员文章站 2022-06-20 20:23:48
...

翻看之前springboot集成的mybatis读写分离,发现还有些疏漏 ,有的还不甚理解,于是翻看下源码;

读写分离 主要就是数据路由的时候重写roundRobinDataSouceProxy方法 ,roundRobinDataSouceProxy中最重要的是 AbstractRoutingDataSource类中的一个抽象方法determineCurrentLookupKey()

下面我们来看下AbstractRoutingDataSource类

 /**
     * 把所有数据库都放在路由中
     * @return
     */
    @Bean(name="roundRobinDataSouceProxy")
    public AbstractRoutingDataSource roundRobinDataSouceProxy() {

        Map<Object, Object> targetDataSources = new HashMap<Object, Object>();
        //把所有数据库都放在targetDataSources中,注意key值要和determineCurrentLookupKey()中代码写的一至,
        //否则切换数据源时找不到正确的数据源
        targetDataSources.put(DataSourceType.write.getType(), writeDataSource);
        targetDataSources.put(DataSourceType.read.getType(), readDataSource);
 
 
        //路由类,寻找对应的数据源
        AbstractRoutingDataSource proxy = new AbstractRoutingDataSource(){
            
            @Override
            protected Object determineCurrentLookupKey() {
                
                return null;
            }
        };
	
        proxy.setDefaultTargetDataSource(writeDataSource);//默认库
	////设置数据源映射
        proxy.setTargetDataSources(targetDataSources);
        return proxy;
    }

AbstractRoutingDataSource源码
 

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean{
  public Connection getConnection() throws SQLException {
        return this.determineTargetDataSource().getConnection();
    }
 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;
        }
    }

        protected abstract Object determineCurrentLookupKey();

AbstractRoutingDataSource源码中有个Object lookupKey = this.determineCurrentLookupKey();

在spring源码中很常见的引用 抽象类中的方法引用抽象方法的实现;

最核心的就determineCurrentLookupKey方法实现, 返回一个datasource实例枚举key,然后当前resolvedDataSources中获取key对应的数据源,从而完成,数据源的动态切换。

相关标签: 源码解析