Day07 Druid ha中的数据源选择器——DataSourceSelector

DataSourceSelector

Druid 中的数据源选择器都会实现 DataSourceSelector 接口：

public interface DataSourceSelector {
    /**
     * Return a DataSource according to the implemention.
     */
    DataSource get();

    /**
     * Set the target DataSource name to return.
     * Wether to use this or not, it's decided by the implemention.
     */
    void setTarget(String name);

    /**
     * Return the name of this DataSourceSelector.
     * e.g. byName
     */
    String getName();

    /**
     * Init the DataSourceSelector before use it.
     */
    void init();

    /**
     * Destroy the DataSourceSelector, maybe interrupt the Thread.
     */
    void destroy();
}

Druid 中的数据源选择器有：

• NamedDataSourceSelector

• RandomDataSourceSelector

• StickyRandomDataSourceSelector

一、NamedDataSourceSelector

NamedDataSourceSelector 根据名字来选择数据源，逻辑实现都在 NamedDataSourceSelector#get() 方法：

@Override
    public DataSource get() {
        if (highAvailableDataSource == null) {
            return null;
        }

        // 获取可用数据源集合
        Map<String, DataSource> dataSourceMap = highAvailableDataSource.getAvailableDataSourceMap();
        if (dataSourceMap == null || dataSourceMap.isEmpty()) {
            return null;
        }
        // 只有一个可用数据源，直接返回这个数据源
        if (dataSourceMap.size() == 1) {
            for (DataSource v : dataSourceMap.values()) {
                return v;
            }
        }
        // 目标数据源的名称
        String name = getTarget();
        if (name == null) {
            if (dataSourceMap.get(getDefaultName()) != null) {
                // 返回默认数据源
                return dataSourceMap.get(getDefaultName());
            }
        } else {
            // 根据名称来获取数据源
            return dataSourceMap.get(name);
        }
        return null;
    }

二、RandomDataSourceSelector

RandomDataSourceSelector——随机数据源选择器。

    public DataSource get() {
        // 获取数据源集合
        Map<String, DataSource> dataSourceMap = getDataSourceMap();
        if (dataSourceMap == null || dataSourceMap.isEmpty()) {
            return null;
        }

        // 移除掉数据源集合中的黑名单数据源
        Collection<DataSource> targetDataSourceSet = removeBlackList(dataSourceMap);
        // 移除掉没有空闲连接的数据源
        removeBusyDataSource(targetDataSourceSet);
        // 从剩下的数据源集合中通过随机函数 random.nextInt 随机选择一个数据源
        DataSource dataSource = getRandomDataSource(targetDataSourceSet);
        return dataSource;
    }

三、StickyRandomDataSourceSelector

StickyRandomDataSourceSelector ——粘性数据源选择器，它基于RandomDataSourceSelector，如果是第一次获取数据源，那么就是随机选取，之后会把这个数据源绑定到当前线程（ThreadLocal），会持续一段时间（这个时间可以配置）。这里的原理类似 CPU 的亲和性。

	private ThreadLocal<StickyDataSourceHolder> holders = new ThreadLocal<StickyDataSourceHolder>();
	
	@Override
    public DataSource get() {
        StickyDataSourceHolder holder = holders.get();
        if (holder != null && isAvailable(holder)) {
            // 如果当前线程的 ThreadLocal 中有绑定数据源，那么直接返回该数据源
            LOG.debug("Return the sticky DataSource " + holder.getDataSource().toString() + " directly.");
            return holder.getDataSource();
        }
        
        // 返回一个随机数据源
        LOG.debug("Return a random DataSource.");
        DataSource dataSource = super.get();
        holder = new StickyDataSourceHolder(dataSource);
        // 清空 ThreadLocal
        holders.remove();
        // 将随机选择出来的数据源对象放入 ThreadLocal.
        holders.set(holder);
        return dataSource;
    }

这里通过 ThreadLocal 实现粘性选择，类似 CPU的亲和性，思路不错，值得借鉴。