監控數據源連接池使用情况

背景

上一篇《數據源連接池未關閉的問題 Could not open JDBC Connection for transaction》分析源碼了解了MyBatis框架和Spring事務管理自動關閉數據源連接池的功能，解决了一個線上數據源連接池占滿的問題。
但解决了一個可能還有下一個，現在需要一個能監控數據源連接池使用情况、快占滿了報警的方案。

方案一

這裏使用的是Druid數據源，其他數據源可以使用下面的方案二。
Druid數據源本身有一個filter，在獲取數據源連接（調用getConnection方法）時會依次調用所有的filter。
於是乎，我們就可以在數據源初始化時增加一個filter，用於查看當前連接池的情况，代碼如下：

List<Filter> filters = new ArrayList<Filter>();
filters.add(new FilterAdapter(){
    
	@Override
	public DruidPooledConnection dataSource_getConnection(FilterChain chain, DruidDataSource dataSource, long maxWaitMillis) throws SQLException {
    
		int activeCount = dataSource.getActiveCount();
		if (activeCount > (dataSource.getMaxActive() * 0.8)) {
    
			log.error("Druid Warning,the current active count of Druid Connection Pool is " + activeCount + "!");
		}
		return chain.dataSource_connect(dataSource, maxWaitMillis);
	}
});
dataSource.setProxyFilters(filters);

這裏當當前連接數大於最大連接數的80%會打印錯誤日志，當然也可以再次增加其他各種報警代碼。

方案二

方案二可以應用在一般的數據源，不限定Druid數據源。核心思想為，寫一個繼承數據源子類，重寫獲取連接的方法，在獲取連接前實現監控的代碼，示例代碼如下：

DruidDataSource dataSource = new DruidDataSource(){
    
	protected Logger log = LoggerFactory.getLogger(BaqiDataSource.class);
	@Override
	public DruidPooledConnection getConnectionDirect(long maxWaitMillis) throws SQLException {
    
		int activeCount = this.getActiveCount();
		if (activeCount > (dataSource.getMaxActive() * 0.8)) {
    
			log.error("Druid Warning,the current active count of Druid Connection Pool is " + activeCount + "!");
		}
		return super.getConnectionDirect(maxWaitMillis);
	}
};
//以下初始化連接池
dataSource.setUrl("xxx");
dataSource.setUsername("xxx");
dataSource.setPassword("xxx"));
dataSource.setDriverClassName("xxx"));
...

結論

方案一適用於Druid數據源，方案二適用於所有的數據源，可以根據情况使用。