[druid 源码解析] 8 HighAvailableDataSource 解析

我们了解完 HighAvailableDataSource 如何使用后，我们继续来了解他是如何运行了，我们先看一下他的初始化方法：

    public void init() {
        // 双重检查防止多次初始化。
        if (inited) {
            return;
        }
        synchronized (this) {
            if (inited) {
                return;
            }
            // 假如 dataSourceMap 为空，启动 Updater ，动态更新 dataSourceMap
            if (dataSourceMap == null || dataSourceMap.isEmpty()) {
                poolUpdater.setIntervalSeconds(poolPurgeIntervalSeconds);
                poolUpdater.setAllowEmptyPool(allowEmptyPoolWhenUpdate);
                poolUpdater.init();
                createNodeMap();
            }
            // 通过 setSelector 初始化，假如没有创建默认 DataSourceSelector
            if (selector == null) {
                setSelector(DataSourceSelectorEnum.RANDOM.getName());
            }
            if (dataSourceMap == null || dataSourceMap.isEmpty()) {
                LOG.warn("There is NO DataSource available!!! Please check your configuration.");
            }
            inited = true;
        }
    }

1. 首选双重检查，防止初始化多次 HighAvailableDataSource 。
2. 假如 dataSourceMap 为空，初始化 poolUpdater 来获取和更新 dataSourceMap ，这里接下来会细讲。
3. 判断 selector 是否为空，假如为空就初始化，我们先看一下 selector 的初始化流程：

    public void setSelector(String name) {
        DataSourceSelector selector = DataSourceSelectorFactory.getSelector(name, this);
        if (selector != null) {
            selector.init();
            setDataSourceSelector(selector);
        }
    }

首先 DataSourceSelectorFactory 会通过 name 去生成真正的 selector ，我们先看一下 DataSourceSelector 的类层次结构：

DataSourceSelector

我们看 init 方法，默认使用的 selector 是 RandomDataSourceSelector , 我们先看一下 RandomDataSourceSelector 的具体实现。

    @Override
    public void init() {
        if (highAvailableDataSource == null) {
            LOG.warn("highAvailableDataSource is NULL!");
            return;
        }
        if (!highAvailableDataSource.isTestOnBorrow() && !highAvailableDataSource.isTestOnReturn()) {
            loadProperties();
            initThreads();
        } else {
            LOG.info("testOnBorrow or testOnReturn has been set to true, ignore validateThread");
        }
    }

1. loadProperties 该方法会从传入的 HighAvailableDataSource 中查找相关的配置信息，包括checkingIntervalSeconds 检查间隔时间，recoveryIntervalSeconds 苏醒间隔时间，validationSleepSeconds 验证睡眠时间，blacklistThreshold 黑名单阈值。
2. initThreads 根据配置初始化线程，这里主要涉及到两条线程 validateThread & recoverThread 。
   我们首先看一下 validateThread 他主要是检查 dataSourceMap 中的 datasource 是否可用。

    @Override
    public void run() {
        while (true) {
            if (selector != null) {
                checkAllDataSources();
                maintainBlacklist();
                cleanup();
            } else {
                break;
            }
            sleepForNextValidation();
        }
    }

1. checkAllDataSources 检查 datasource 是否正常，这里会根据检查时间看是否需要跳过，主要是根据 checkingIntervalSeconds 来判断，接着到了真正检查的逻辑，他会先从 datasource 中获取链接的信息，并新建一条链接，而非重 datasource 中获取，然后使用这个链接执行一条指令，在 MySQL 中是执行 pingInternal 方法。
2. 检查上面的执行检查结果是否成功，假如成功，就从 blacklist 中移除，并重置 errorcounter ，假如不成功且失败次数大于 blacklistThreshold 黑名单阈值，将其放入 blacklist 中，等待后续操作。
3. cleanup 根据 successTimes , errorCounts 和 lastCheckTimes 来清理 datasource,将其移入 blacklist 中。
   接下来我们看一下 recoverThread 。

 @Override
    public void run() {
        while (true) {
            if (selector != null && selector.getBlacklist() != null
                    && !selector.getBlacklist().isEmpty()) {
                LOG.info(selector.getBlacklist().size() + " DataSource in blacklist.");
                for (DataSource dataSource : selector.getBlacklist()) {
                    if (!(dataSource instanceof DruidDataSource)) {
                        continue;
                    }
                    tryOneDataSource((DruidDataSource) dataSource);
                }
            } else if (selector == null) {
                break;
            }
            sleep();
        }
    }

这里的流程就比较简单了，主要是 tryOneDataSource 尝试将 datasource 将其移出 blacklist 。这里的检查的逻辑和 validateThread 的 checkAllDataSources 基本一致，就是会尝试启动一个新的链接检查该数据源是否正常。
接下来我们看一下 RandomDataSourceSelector 最主要的 get 获取数据源的方法：

@Override
    public DataSource get() {
        Map dataSourceMap = getDataSourceMap();
        if (dataSourceMap == null || dataSourceMap.isEmpty()) {
            return null;
        }

        Collection targetDataSourceSet = removeBlackList(dataSourceMap);
        removeBusyDataSource(targetDataSourceSet);
        DataSource dataSource = getRandomDataSource(targetDataSourceSet);
        return dataSource;
    }

这里的逻辑就会比较简单，主要先将黑名单的和 busy 的DataSource 移除，这里校验是否 busy ，只检查改 DataSource 的 poolcount 是否 <= 0； 假如是就代表空闲连接为 0 。最后这里 getRandomDataSource 就是通过 radom 函数求余获取到真正的 DataSource 。