mybatis多資料來源踩坑，資料庫連線經常斷開問題

Hiway發表於2019-08-29
MyBatis資料庫

問題

最近某專案上出現一個奇怪的問題，就是資料庫經常隔幾小時就報連線已關閉

mybatis多資料來源踩坑，資料庫連線經常斷開問題

mybatis多資料來源踩坑，資料庫連線經常斷開問題

即使是加了如下配置也依然不行，網上也沒找到什麼文章解釋這個坑

test-on-borrow: true
test-while-idle: true
validation-query: select 1 from dual
複製程式碼

關於上面配置為什麼能解決連線中斷可以看這三篇文章

排查

網上查不到，那就只能自己推敲猜測了。因為是mybatis多資料來源的配置，所以每個db我都有專門寫一個config作為連線配置。

mybatis多資料來源踩坑，資料庫連線經常斷開問題

看著DataSourceConfig的程式碼，我突然想到，會不會是因為我使用到是DataSource預設建立方法，所以並沒有讀取到我寫在application.yml的配置：

    @Bean(name = "db1DataSource")
    @ConfigurationProperties(prefix = "spring.datasource.db1")
    @Primary
    public DataSource dbDataSource() {
        return DataSourceBuilder.create().build();
    }
複製程式碼

果斷跟進build()方法

	public DataSource build() {
		Class<? extends DataSource> type = getType();
		DataSource result = BeanUtils.instantiate(type);
		maybeGetDriverClassName();
		bind(result);
		return result;
	}
複製程式碼

打個斷點可以看到此時返回的result的是一個全新的DataSource

mybatis多資料來源踩坑，資料庫連線經常斷開問題

所以我們可以通過修改dbDataSource()方法，寫入我們的配置引數：

 @Value("${spring.datasource.db1.url}")
    private String url;
    @Value("${spring.datasource.db1.username}")
    private String username;
    @Value("${spring.datasource.db1.password}")
    private String password;
    @Value("${spring.datasource.db1.tomcat.test-on-borrow}")
    private boolean testOnBorrow;
    @Value("${spring.datasource.db1.tomcat.test-while-idle}")
    private boolean testWhileIdle;
    @Value("${spring.datasource.db1.tomcat.validation-query}")
    private String validationQuery;
    @Value("${spring.datasource.db1.tomcat.max-idle}")
    private int maxIdle;
    @Value("${spring.datasource.db1.tomcat.min-idle}")
    private int minIdle;
    @Value("${spring.datasource.db1.tomcat.initial-size}")
    private int initialSize;
    @Value("${spring.datasource.db1.tomcat.max-active}")
    private int maxActive;
    @Value("${spring.datasource.db1.tomcat.time-between-eviction-runs-millis}")
    private int timeBetweenEvictionRunsMillis;

    @Bean(name = "db1DataSource")
    @ConfigurationProperties(prefix = "spring.datasource.db1")
    @Primary
    public DataSource dbDataSource() {
        org.apache.tomcat.jdbc.pool.DataSource dataSource = new org.apache.tomcat.jdbc.pool.DataSource();
        dataSource.setDriverClassName("oracle.jdbc.driver.OracleDriver");
        dataSource.setUrl(url);
        dataSource.setUsername(username);
        dataSource.setPassword(password);
        dataSource.setMaxActive(maxActive);
        dataSource.setMinIdle(minIdle);
        dataSource.setMaxIdle(maxIdle);
        dataSource.setTestOnBorrow(testOnBorrow);
        dataSource.setTestWhileIdle(testWhileIdle);
        dataSource.setValidationQuery(validationQuery);
        dataSource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
        dataSource.setInitialSize(initialSize);
        return dataSource;
        //return DataSourceBuilder.create().build();
    }
複製程式碼

二次排查

通過上面方法確實可以解決問題了，但是我突然想到，預設建立的DataSource是沒有url，username，password等必要的基礎資訊的。那這幾個配置引數是為什麼又可以寫入進去呢？

這個時候我看到了我們dbDataSource方法上有一個@Bean(name = "db1DataSource")，於是大膽猜測我們這些配置引數的注入是第一次建立的時候通過Spring的IOC注入的。通過我的Debug發現事實也確實如此。

對DataBinder類的bind方法打斷點，

	public void bind(PropertyValues pvs) {
		MutablePropertyValues mpvs = (pvs instanceof MutablePropertyValues) ?
				(MutablePropertyValues) pvs : new MutablePropertyValues(pvs);
		doBind(mpvs);
	}
複製程式碼

mybatis多資料來源踩坑，資料庫連線經常斷開問題
我們可以看到方法的呼叫路徑

mybatis多資料來源踩坑，資料庫連線經常斷開問題

看到了我們熟悉的refresh大法，這一部分Spring原始碼相關請看Spring原始碼分析

此時只剩最後一個疑惑了，我們的url，username，password等既然能通過IOC注入到DataSource裡，那為什麼其他的引數不可以呢？我隨著DataSource類一路往上，找到他的父類介面PoolConfiguration，看到了所有的引數和getset方法。

mybatis多資料來源踩坑，資料庫連線經常斷開問題

再看一眼我的application.yml配置檔案裡的引數

spring:
    datasource:
        db1:
            url: 
            username: 
            password: 
            driver-class-name: oracle.jdbc.driver.OracleDriver
            tomcat:
                max-wait: 10000
                max-active: 30
                test-on-borrow: true
                max-idle: 5
        db2:
            xxx
            ....
複製程式碼

終於找到這個坑了！
原來Spring data預設使用tomcat-jdbc的連線池的時候，配置的引數是

spring:
    datasource:
        url: 
        username: 
        password: 
        driver-class-name: oracle.jdbc.driver.OracleDriver
        tomcat:
            max-wait: 10000
            max-active: 30
            test-on-borrow: true
            max-idle: 5
複製程式碼

而當使用多資料來源配置的時候，簡單的以為只是複製過去即可，所以Spring IOC注入的時候，讀的到的tomcat.max-wait並不能匹配到DataSource裡的setMaxWait方法。自然就不起作用了。

所以這個問題只需要將配置檔案改為如下即可

spring:
    datasource:
        db1:
            url: 
            username: 
            password: 
            driver-class-name: oracle.jdbc.driver.OracleDriver
            max-wait: 10000
            max-active: 30
            test-on-borrow: true
            test-while-idle: true
            validation-query: select 1 from dual
            max-idle: 5
        db2:
            xxx
            ....
複製程式碼

總結

這個問題從結果上來看，那真是簡單的不行，但是從過程上來說，不僅讓我又複習了一遍spring IOC的流程，也讓我感覺到這種一步一步解剖問題，把多個知識點連線起來的成就感。如果之前沒有學習spring的原始碼，我這次大概率也不會想到去看bean的注入吧

相關文章