Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池

一、使用 Druid

Spring Boot 项目中配置数据库连接池为 Druid 并启动项目,Spring 上下文初始化完成后会初始化 DruidDataSource, 如下图:

Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第1张图片

核心类:

  • DruidDataSourceAutoConfigure 负责初始化 DruidDataSource
  • com.alibaba.druid.pool.DruidDataSource 负责创建高效可管理的数据库连接池

二、 druid-spring-boot-starter 源码

DruidDataSourceAutoConfiguredruid-spring-boot-starter下的一个配置类。此 starter 中 spring.factories 文件中定义了自动加载 DruidDataSourceAutoConfigure 配置类:

org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure

DruidDataSourceAutoConfigure 配置类,源码如下,这里初始化了 Bean : DruidDataSourceWrapper(继承了 DruidDataSource类),初始化了DruidDataSourc#init() 方法。

@Configuration
// 判断当前classpath下是否存在 DruidDataSource.class 类,若是则将当前的配置装载入spring 容器
@ConditionalOnClass(DruidDataSource.class)
// DruidDataSourceAutoConfigure 类 在 DataSourceAutoConfiguration 类之前加载
@AutoConfigureBefore(DataSourceAutoConfiguration.class)
// 把使用 @ConfigurationProperties 的类 DruidStatProperties 和 DataSourceProperties 进行了一次注入。
@EnableConfigurationProperties({DruidStatProperties.class, DataSourceProperties.class})
// 把类注入到 IoC 容器
@Import({DruidSpringAopConfiguration.class,
    DruidStatViewServletConfiguration.class,
    DruidWebStatFilterConfiguration.class,
    DruidFilterConfiguration.class})
public class DruidDataSourceAutoConfigure {

    private static final Logger LOGGER = LoggerFactory.getLogger(DruidDataSourceAutoConfigure.class);

    // 初始化 Bean: DruidDataSourceWrapper
    @Bean(initMethod = "init")
    @ConditionalOnMissingBean
    public DataSource dataSource() {
        LOGGER.info("Init DruidDataSource");
        return new DruidDataSourceWrapper();
    }
}

DruidDataSourceWrapper 类的关系图:
Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第2张图片

所以,下面重点看DruidDataSource#init() 方法的源码:

三、Druid 如何初始化数据库连接池

DruidDataSourc#init() 方法主要做了这几件事情:

  1. 初始化 jdbcUrl

  2. 初始化过滤器:
    Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第3张图片

  3. 根据配置的 jdbcUrl 判断数据库的类型

  4. 通过 SPI ServiceLoader ,让类加载器加载 过滤器

  5. 加载数据库驱动 resolveDriver();

  6. 初始化检查,检查数据库类型和驱动是否支持

  7. 初始化数据库连接

  8. 创建线程:连接池日志分析线程、连接池创建线程、连接池销毁线程

  9. 向线程池提交创建连接的任务:this.createSchedulerFuture = createScheduler.submit(task);

其中对于并发以及安全的考虑有:

  1. 整个初始化过程使用可重入锁来保证互斥性:final ReentrantLock lock = this.lock;

  2. 使用 volatile 变量来保证只会初始化一次

    Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第4张图片

  3. DruidDriver 使用静态常量来保证单例
    Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第5张图片

  4. 使用 AtomicLongFieldUpdater 类来原子更新 DruidAbstractDataSourcevolatile long 修饰的变量
    Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第6张图片

  5. 使用 CountDownLatch 来确保 「连接池创建线程」、「连接池销毁线程」 一定会创建
    Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第7张图片

  6. 使用线程池来优化数据库连接的创建
    Spring生态中如何使用 Druid & Druid 如何初始化数据库连接池_第8张图片

你可能感兴趣的