我们了解完 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;
}
}
- 首选双重检查，防止初始化多次
HighAvailableDataSource。 - 假如
dataSourceMap为空，初始化poolUpdater来获取和更新dataSourceMap，这里接下来会细讲。 - 判断
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");
}
}
-
loadProperties该方法会从传入的HighAvailableDataSource中查找相关的配置信息，包括checkingIntervalSeconds检查间隔时间，recoveryIntervalSeconds苏醒间隔时间，validationSleepSeconds 验证睡眠时间，blacklistThreshold黑名单阈值。 -
initThreads根据配置初始化线程，这里主要涉及到两条线程validateThread&recoverThread。
我们首先看一下validateThread他主要是检查dataSourceMap中的datasource是否可用。
@Override
public void run() {
while (true) {
if (selector != null) {
checkAllDataSources();
maintainBlacklist();
cleanup();
} else {
break;
}
sleepForNextValidation();
}
}
-
checkAllDataSources检查datasource是否正常，这里会根据检查时间看是否需要跳过，主要是根据checkingIntervalSeconds来判断，接着到了真正检查的逻辑，他会先从datasource中获取链接的信息，并新建一条链接，而非重datasource中获取，然后使用这个链接执行一条指令，在 MySQL 中是执行pingInternal方法。 - 检查上面的执行检查结果是否成功，假如成功，就从
blacklist中移除，并重置 errorcounter ，假如不成功且失败次数大于blacklistThreshold黑名单阈值，将其放入blacklist中，等待后续操作。 -
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<String, DataSource> dataSourceMap = getDataSourceMap();
if (dataSourceMap == null || dataSourceMap.isEmpty()) {
return null;
}
Collection<DataSource> targetDataSourceSet = removeBlackList(dataSourceMap);
removeBusyDataSource(targetDataSourceSet);
DataSource dataSource = getRandomDataSource(targetDataSourceSet);
return dataSource;
}
这里的逻辑就会比较简单，主要先将黑名单的和 busy 的DataSource 移除，这里校验是否 busy ，只检查改 DataSource 的 poolcount 是否 <= 0； 假如是就代表空闲连接为 0 。最后这里 getRandomDataSource 就是通过 radom 函数求余获取到真正的 DataSource 。
