spring-jdbc之AbstractRoutingDataSource源码解析

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

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

下面我们来看下AbstractRoutingDataSource类

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

        Map targetDataSources = new HashMap();
        //把所有数据库都放在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对应的数据源,从而完成,数据源的动态切换。

你可能感兴趣的:(源码解析)