Druid源码分析（三）-- 初始化DataSource2

init

昨天分析了init 函数的双重校验+锁，今天接着往下看源码

// DruidDataSource.java#init()
// 获取堆栈信息
initStackTrace = Utils.toString(Thread.currentThread().getStackTrace());
// 创建数据源id 
this.id = DruidDriver.createDataSourceId();

这里用的是原子类Integer，来保证自增是线程安全的

// DruidDriver.java
private final static AtomicInteger dataSourceIdSeed = new AtomicInteger(0);
public static int createDataSourceId() {
	return dataSourceIdSeed.incrementAndGet();
}

多数据源处理

if (this.id > 1) {
    long delta = (this.id - 1) * 100000;
    this.connectionIdSeedUpdater.addAndGet(this, delta);
    this.statementIdSeedUpdater.addAndGet(this, delta);
    this.resultSetIdSeedUpdater.addAndGet(this, delta);
    this.transactionIdSeedUpdater.addAndGet(this, delta);
}

接下来是加载filters，自带的监控filter等在这里加载

for (Filter filter : filters) {
	// filter初始化
	filter.init(this);
}

// StatFilter.java
// 以监控filter为例，查看下init的实现，其他filter类似
public void init(DataSourceProxy dataSource) {
	// 加锁，避免重复初始化
    lock.lock();
    try {
    	// 如果filter的 dbType 为空，用数据源的dbType 
        if (this.dbType == null) {
            this.dbType = DbType.of(dataSource.getDbType());
        }
		// 从配置文件加载配置
        configFromProperties(dataSource.getConnectProperties());
        configFromProperties(System.getProperties());
    } finally {
    	// 解锁
        lock.unlock();
    }
}

接着往下看，是校验dbTypeName ，没配置的话会调用JdbcUtils 的函数通过jdbcUrl来获取

// 如果没有设置数据源TypeName，或通过jdbcUrl来获取
if (this.dbTypeName == null || this.dbTypeName.length() == 0) {
    this.dbTypeName = JdbcUtils.getDbType(jdbcUrl, null);
}

// JdbcUtils.java
// 从这个函数可以看出druid连接池支持的数据库类型是非常多的
public static DbType getDbTypeRaw(String rawUrl, String driverClassName) {
   	...
    if (rawUrl.startsWith("jdbc:mysql:") || rawUrl.startsWith("jdbc:cobar:")
            || rawUrl.startsWith("jdbc:log4jdbc:mysql:")) {
        return DbType.mysql;
    } else if (rawUrl.startsWith("jdbc:mariadb:")) {
        return DbType.mariadb;
    } else if (rawUrl.startsWith("jdbc:oracle:") || rawUrl.startsWith("jdbc:log4jdbc:oracle:")) {
        return DbType.oracle;
    ...
 }

下面就是校验一些参数的合法性

// 校验最大连接池数量
if (maxActive <= 0) {
    throw new IllegalArgumentException("illegal maxActive " + maxActive);
}

if (maxActive < minIdle) {
    throw new IllegalArgumentException("illegal maxActive " + maxActive);
}

// 这里默认连接池数量 DEFAULT_INITIAL_SIZE = 0;
if (getInitialSize() > maxActive) {
    throw new IllegalArgumentException("illegal initialSize " + this.initialSize + ", maxActive " + maxActive);
}

SPI机制

自定义的filter在这里加载

private void initFromSPIServiceLoader() {
	// 默认值是false，可以在配置文件通过druid.load.spifilter.skip 设置值为true，即可跳过这个加载
    if (loadSpifilterSkip) {
        return;
    }

    if (autoFilters == null) {
        List<Filter> filters = new ArrayList<Filter>();
        // 通过SPI机制去加载Filter
        ServiceLoader<Filter> autoFilterLoader = ServiceLoader.load(Filter.class);
		// 开启了自动加载 AutoLoad 注解的filter 才会被加载
        for (Filter filter : autoFilterLoader) {
            AutoLoad autoLoad = filter.getClass().getAnnotation(AutoLoad.class);
            if (autoLoad != null && autoLoad.value()) {
                filters.add(filter);
            }
        }
        autoFilters = filters;
    }

    for (Filter filter : autoFilters) {
        ...
        // 把通过SPI加载的filter 加到filter 列表里
        addFilter(filter);
    }
}