spring-boot-2.0.3之quartz集成，數據源問題，源碼探究

前言開心一刻著火了，他報警說：119嗎，我家發生火災了。 119問：在哪裡？他說：在我家。 119問：具體點。他說：在我家的廚房裡。 119問：我說你現在的位置。他說：我趴在桌子底下。 119：我們怎樣才能到你家？他說：你們不是有消防車嗎？ 119說：燒死你個傻B算了。路漫漫其修遠兮， ...

前言

　　開心一刻　　

　　　　著火了，他報警說：119嗎，我家發生火災了。
　　　　119問：在哪裡？
　　　　他說：在我家。
　　　　119問：具體點。
　　　　他說：在我家的廚房裡。
　　　　119問：我說你現在的位置。
　　　　他說：我趴在桌子底下。
　　　　119：我們怎樣才能到你家？
　　　　他說：你們不是有消防車嗎？
　　　　119說：燒死你個傻B算了。

　　路漫漫其修遠兮，吾將上下而求索！
　　github：https://github.com/youzhibing
　　碼雲(gitee)：https://gitee.com/youzhibing

前情回顧

　　上篇博客中，講到了springboot與quartz的集成，非常簡單，pow.xml中引入spring-boot-starter-quartz依賴即可，工程中就可以通過

@Override
private Scheduler scheduler;

　　自動註入quartz調度器，然後我們就可以通過調度器對quartz組件：Trigger、JobDetail進行添加與刪除等操作，實現對任務的調度。

　　結果也如我們預期一樣，每隔10s我們的MyJob的executeinternal方法就被調用，列印一條信息：MyJob...

　　似乎一切都是那麼順利，感覺集成quartz就是這麼簡單！

　　測試工程：spring-boot-quartz

數據源問題

　　產生背景

　　　　如果定時任務不服務於業務，那將毫無意義；我們不能讓定時任務只是空跑(或者列印一句：MyJob...)，如果是，那麼相信我，把這個定時任務刪了吧，不要有任何留戀！

　　　　既然是服務於我們的業務，那麼很大程度上就會操作資料庫；我的業務需求就是凌晨某個時間點進行一次數據統計，既要從資料庫查數據，也要將統計後的數據插入到資料庫。那麼問題來了，業務job中如何操作資料庫？

　　　　業務job示例

package com.lee.quartz.job;

import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;
import org.springframework.scheduling.quartz.QuartzJobBean;

public class MyJob extends QuartzJobBean {

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

    @Override
    protected void executeInternal(JobExecutionContext context) throws JobExecutionException {
        // TODO 如何進行資料庫的操作
        System.out.println("MyJob...")
    }
}

　　　　可以從4個方面來考慮（業務job中如何操作資料庫）：

　　　　　　1、既然是springboot與quartz的集成，那麼我們能不能用spring的註入功能，將我們的mapper（集成了mybatis）註入到業務job中了？

　　　　　　2、利用JobDetail的jobDataMap，將我們的mapper傳到業務job中

　　　　　　3、quartz不是有它自己的11張表嗎，那它肯定有對資料庫進行操作，我們參考quartz是如何操作資料庫的

　　　　　　4、實在是不行，我們自己創建資料庫連接總行了吧

　　　　我們來逐個分析下以上4種方案

　　　　　　方案4，個人不推薦，個人比較推薦連接池的方式來管理資料庫連接，但個人實現資料庫連接池已經是個不小的挑戰了，沒必要；不到萬不得已不採用此方案。

　　　　　　方案1，這個聽起來好像很不錯，連接交由spring的數據源管理，我們只需要用其中的連接操作資料庫即可。但看上面的MyJob，spring管理的bean能註入進來嗎，顯然不能，因為MyJob實例不受spring管理；有小伙伴可能會認為這很簡單，MyJob實例讓spring管理起來不就OK 了！ ok，問題又來了，spring管理的MyJob實例能用到quartz中嗎，不能！ quartz如何獲取MyJob實例？我們把MyJob的類全路徑：com.lee.quartz.job.MyJob傳給了quartz，那麼很顯然quartz會根據這個類全路徑，然後通過反射來實例化MyJob（這也是為什麼業務Job一定要有無參構造方法的原因），也就是quartz會重新創建MyJob實例，與spring管理MyJob實例沒有任何關係。顯然通過spring註入的方式是行不通的。

　　　　　　方案2，我們知道可以通過JobDetail進行參數的傳遞，但有要求：傳遞的參數必須能序列化（實現Serializable）；我沒測試此方案，不過我想實現起來會有點麻煩。

　　　　　　方案3，這個好像可行，我們可以看看quartz是如何進行資料庫操作的，我們把quartz的那套拿過來用是不是就行了呢？

　　　　說了這麼多，方案總結下：

　　　　　　1、如何利用quartz的數據源（或者資料庫連接）進行資料庫操作

　　　　　　2、引申下，能不能將quart的數據源設置成我們應用的數據源，讓quartz與應用共用一個數據源，方便統一管理？

　　源碼探究

　　　　1、quartz自身是如何操作資料庫的

　　　　　　我們通過暫停任務來跟下源代碼，如下圖所示

　　　　　　發現獲取connection的方式如下所示：

conn = DBConnectionManager.getInstance().getConnection(getDataSource());

　　　　　　很明顯，DBConnectionManager是單例的，通過DBConnectionManager從數據源中獲取資料庫連接（conn），既然都拿到conn了，那操作資料庫也就簡單了。註意：getDataSource()獲取的是數據源的名稱，不是數據源！

　　　　　　接下來我們再看看數據源是什麼數據源，druid？還是quartz自己的數據源？

　　　　　　數據源還是用的我們應用的數據源（druid數據源），springboot自動將我們應用的數據源配置給了quartz。

　　　　　　至此，該問題也就清晰了，總結下：springboot會自動將我們的應用數據源（druid數據源）配置給quartz，quartz操作資料庫的時候從數據源中獲取資料庫連接，然後通過資料庫連接對資料庫進行操作。

　　　　2、springboot是如何設置quartz數據源的

　　　　　　凡是涉及到springboot自動配置的，去找spring-boot-autoconfigure-2.0.3.RELEASE.jar中spring.factories就對了，如下所示

　　　　　　關於spring.factories文件內容的讀取，大家查閱此篇博文；關於springboot的自動配置，我的springboot啟動源碼系列篇中還沒有講到。大家姑且先這樣認為：

　　　　　　　　當在類路徑下能找到Scheduler.class, SchedulerFactoryBean.class,PlatformTransactionManager.class時（只要pom.xml有spring-boot-starter-quartz依賴，這些類就能在類路徑下找到），QuartzAutoConfiguration就會被springboot當成配置類進行自動配置。

　　　　　　將quartz的配置屬性設置給SchedulerFactoryBean；將數據源設置給SchedulerFactoryBean：如果有@QuartzDataSource修飾的數據源，則將@QuartzDataSource修飾的數據源設置給SchedulerFactoryBean，否則將應用的數據源（druid數據源）設置給SchedulerFactoryBean，顯然我們的應用中沒有@QuartzDataSource修飾的數據源，那麼SchedulerFactoryBean中的數據源就是應用的數據源；將事務管理器設置給SchedulerFactoryBean。

　　　　　　SchedulerFactoryBean，Scheduler的工程bean，負責創建和配置quartz Scheduler；它實現了FactoryBean、InitializingBean，FactoryBean的getObject方法實現的很簡單，如下

@Override
@Nullable
public Scheduler getObject() {
    return this.scheduler;
}

　　　　　　就是返回scheduler實例，註冊到spring容器中，那麼scheduler是在哪裡實例化的呢，就是在afterPropertiesSet中完成的，關於FactoryBean、InitializingBean本文不做過多的講解，不瞭解的可以先去查閱下資料（註意：InitializingBean的afterPropertiesSet()先於FactoryBean的getObject()執行）。接下來我們仔細看看SchedulerFactoryBean實現InitializingBean的afterPropertiesSet方法

@Override
public void afterPropertiesSet() throws Exception {
    if (this.dataSource == null && this.nonTransactionalDataSource != null) {
        this.dataSource = this.nonTransactionalDataSource;
    }

    if (this.applicationContext != null && this.resourceLoader == null) {
        this.resourceLoader = this.applicationContext;
    }

    // Initialize the Scheduler instance... 初始化Scheduler實例
    this.scheduler = prepareScheduler(prepareSchedulerFactory());
    try {
        registerListeners();                // 註冊Scheduler相關監聽器，一般沒有
        registerJobsAndTriggers();            // 註冊jobs和triggers， 一般沒有
    }
    catch (Exception ex) {
        try {
            this.scheduler.shutdown(true);
        }
        catch (Exception ex2) {
            logger.debug("Scheduler shutdown exception after registration failure", ex2);
        }
        throw ex;
    }
}

View Code

　　　　　　我們來重點跟下：this.scheduler = prepareScheduler(prepareSchedulerFactory());

　　　　　　可以看到我們通過org.quartz.jobStore.dataSource設置的dsName（quartzDs）最後會被替換成springTxDataSource.加scheduler實例名（我們的應用中是：springTxDataSource.quartzScheduler），這也就是為什麼我們通過DBConnectionManager.getInstance().getConnection("quartzDs")報以下錯誤的原因

java.sql.SQLException: There is no DataSource named 'quartzDs'
    at org.quartz.utils.DBConnectionManager.getConnection(DBConnectionManager.java:104)
    at com.lee.quartz.job.FetchDataJob.executeInternal(FetchDataJob.java:24)
    at org.springframework.scheduling.quartz.QuartzJobBean.execute(QuartzJobBean.java:75)
    at org.quartz.core.JobRunShell.run(JobRunShell.java:202)
    at org.quartz.simpl.SimpleThreadPool$WorkerThread.run(SimpleThreadPool.java:573)

View Code

　　　　　　LocalDataSourceJobStore的initialize內容如下

@Override
    public void initialize(ClassLoadHelper loadHelper, SchedulerSignaler signaler) throws SchedulerConfigException {
        // Absolutely needs thread-bound DataSource to initialize.
        this.dataSource = SchedulerFactoryBean.getConfigTimeDataSource();
        if (this.dataSource == null) {
            throw new SchedulerConfigException("No local DataSource found for configuration - " +
                    "'dataSource' property must be set on SchedulerFactoryBean");
        }

        // Configure transactional connection settings for Quartz.
        setDataSource(TX_DATA_SOURCE_PREFIX + getInstanceName());
        setDontSetAutoCommitFalse(true);

        // Register transactional ConnectionProvider for Quartz.
        DBConnectionManager.getInstance().addConnectionProvider(
                TX_DATA_SOURCE_PREFIX + getInstanceName(),
                new ConnectionProvider() {
                    @Override
                    public Connection getConnection() throws SQLException {
                        // Return a transactional Connection, if any.
                        return DataSourceUtils.doGetConnection(dataSource);
                    }
                    @Override
                    public void shutdown() {
                        // Do nothing - a Spring-managed DataSource has its own lifecycle.
                    }
                    /* Quartz 2.2 initialize method */
                    public void initialize() {
                        // Do nothing - a Spring-managed DataSource has its own lifecycle.
                    }
                }
        );

        // Non-transactional DataSource is optional: fall back to default
        // DataSource if not explicitly specified.
        DataSource nonTxDataSource = SchedulerFactoryBean.getConfigTimeNonTransactionalDataSource();
        final DataSource nonTxDataSourceToUse = (nonTxDataSource != null ? nonTxDataSource : this.dataSource);

        // Configure non-transactional connection settings for Quartz.
        setNonManagedTXDataSource(NON_TX_DATA_SOURCE_PREFIX + getInstanceName());

        // Register non-transactional ConnectionProvider for Quartz.
        DBConnectionManager.getInstance().addConnectionProvider(
                NON_TX_DATA_SOURCE_PREFIX + getInstanceName(),
                new ConnectionProvider() {
                    @Override
                    public Connection getConnection() throws SQLException {
                        // Always return a non-transactional Connection.
                        return nonTxDataSourceToUse.getConnection();
                    }
                    @Override
                    public void shutdown() {
                        // Do nothing - a Spring-managed DataSource has its own lifecycle.
                    }
                    /* Quartz 2.2 initialize method */
                    public void initialize() {
                        // Do nothing - a Spring-managed DataSource has its own lifecycle.
                    }
                }
        );

        // No, if HSQL is the platform, we really don't want to use locks...
        try {
            String productName = JdbcUtils.extractDatabaseMetaData(this.dataSource, "getDatabaseProductName");
            productName = JdbcUtils.commonDatabaseName(productName);
            if (productName != null && productName.toLowerCase().contains("hsql")) {
                setUseDBLocks(false);
                setLockHandler(new SimpleSemaphore());
            }
        }
        catch (MetaDataAccessException ex) {
            logWarnIfNonZero(1, "Could not detect database type. Assuming locks can be taken.");
        }

        super.initialize(loadHelper, signaler);

    }

View Code

　　　　　　註冊兩個ConnectionProvider給quartz：一個dsName叫springTxDataSource.quartzScheduler，有事務；一個dsName叫springNonTxDataSource.quartzScheduler，沒事務；所以我們通過DBConnectionManager獲取connection時，通過指定dsName就能獲取支持事務或不支持事務的connection。

　　　　　　另外，SchedulerFactoryBean實現了SmartLifecycle，會在ApplicationContext refresh的時候啟動Schedule，ApplicationContext shutdown的時候停止Schedule。

總結

　　1、springboot集成quartz，應用啟動過程中會自動調用schedule的start方法來啟動調度器，也就相當於啟動了quartz，原因是SchedulerFactoryBean實現了SmartLifecycle介面；

　　2、springboot會自動將我們應用的數據源配置給quartz，在我們示例應用中數據源是druid數據源，應用和quartz都是用的此數據源；

　　3、通過org.quartz.jobStore.dataSource設置的數據源名會被覆蓋掉，當我們通過quartz的DBConnectionManager獲取connection時，預設情況dbName給springTxDataSource.quartzScheduler或者springNonTxDataSource.quartzScheduler，一個支持事務，一個不支持事務；至於怎樣自定義dsName，我還沒去嘗試，有興趣的小伙伴可以自己試試；

　　4、springboot集成quartz，只是將quartz的一些通用配置給配置好了，如果我們對quartz十分熟悉，那麼就很好理解，但如果對quartz不熟悉（樓主對quartz就不熟悉），那麼很多時候出了問題就無從下手了，所以建議大家先熟悉quartz；