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连接。
现在只是实现了功能,各种情况暂时没有加入考虑,比如没有判断该连接是否应该被释放。