bonecp回缩功能实现

起因

bonecp不具备回缩功能，即连接池持有连接之后，不会主动去释放这些连接（即使这些连接始终处于空闲状态），因此在使用一段时间之后，连接池会达到配置的最大值。

这种方式一定程度上造成了资源的浪费。

改造

参考tomcat-jdbc的策略，每隔一段时间（可配置）会启动定时任务扫描partition中的idle队列，判断idle连接数是否大于partition可持有的最小连接数，如果是，则启动清理方法，将连接释放掉。

为了达到这个目的，实现了ConnectionCleanThread类：

package com.jolbox.bonecp;

import java.sql.SQLException;
import java.util.concurrent.BlockingQueue;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ConnectionCleanThread implements Runnable {

	private static final Logger logger = LoggerFactory.getLogger(ConnectionCleanThread.class);
	
	private ConnectionPartition partition;
	
	private BoneCP pool;
	
	protected ConnectionCleanThread(ConnectionPartition connectionPartition, BoneCP pool) {
		this.partition = connectionPartition;
		this.pool = pool;
	}
	
	@Override
	public void run() {
		BlockingQueue freeQueue = null;
		ConnectionHandle connection = null;
		//获得partition的大小
		int partitionSize = this.partition.getAvailableConnections();
		for (int i = 0; i < partitionSize; i++) {
			//得到free连接的queue
			freeQueue = this.partition.getFreeConnections();
			//如果空闲连接大于partition的最小允许连接数,回缩到最小允许连接数
			while (freeQueue.size() > this.partition.getMinConnections()) {
				connection = freeQueue.poll();
				connection.lock();
				closeConnection(connection);
				connection.unlock();
			}
		}
	}
	
	/** Closes off this connection
	 * @param connection to close
	 */
	private void closeConnection(ConnectionHandle connection) {

		if (connection != null && !connection.isClosed()) {
			try {
				connection.internalClose();
			} catch (SQLException e) {
				logger.error("Destroy connection exception", e);
			} finally {
				this.pool.postDestroyConnection(connection);
				connection.getOriginatingPartition().getPoolWatchThreadSignalQueue().offer(new Object()); // item being pushed is not important.
			}
		}
	}

}

同时需要对核心类ConnectionHandle进行改造，加上连接的上锁方法：

protected void lock() {
	lock.writeLock().lock();
}
	
protected void unlock() {
	lock.writeLock().unlock();
}

在BoneCP类的构造器内加上该线程的定时任务：

/**
 * 空闲连接清理任务
*/
private ScheduledExecutorService connectionCleanScheduler;

...

this.connectionCleanScheduler = Executors.newScheduledThreadPool(this.config.getPartitionCount(), new CustomThreadFactory("BoneCP-connection-clean-thread"+suffix, true));

...

//定期启动一个线程清理空闲连接
//add 2017-2-10
final Runnable connectionCleaner = new ConnectionCleanThread(connectionPartition, this);
this.connectionCleanScheduler.scheduleAtFixedRate(connectionCleaner, this.config.getConnectionCleanTimeInSeconds(), this.config.getConnectionCleanTimeInSeconds(), TimeUnit.SECONDS);

效果

经过实际测试，可以在后台自动的回收idle连接。

现在只是实现了功能，各种情况暂时没有加入考虑，比如没有判断该连接是否应该被释放。

GitHub地址

bonecp