前言
本章是HikariCP的最后一章,学习HikariCP如何设计归还连接和关闭连接。
一、归还连接
因为暴露给用户使用的是Connection的代理对象
HikariProxyConnection
,用户调用
close
方法关闭连接,实际执行的是
ProxyConnection#close
归还连接。
// 实际数据库连接
protected Connection delegate;
// 关联的PoolEntry
private final PoolEntry poolEntry;
@Overridepublic final void close() throws SQLException {
// 关闭所有打开的Statement
closeStatements();
if (delegate != ClosedConnection.CLOSED_CONNECTION) {
leakTask.cancel();
try {
// 是否需要回滚
if (isCommitStateDirty && !isAutoCommit) {
delegate.rollback();
lastAccess = currentTime();
}
// 用户是否设置过readOnly等连接属性
if (dirtyBits != 0) {
// 如果设置过,则恢复连接属性到创建时的状态
poolEntry.resetConnectionState(this, dirtyBits);
lastAccess = currentTime();
}
// 清除警告信息
delegate.clearWarnings();
} catch (SQLException e) {
if (!poolEntry.isMarkedEvicted()) {
// 发生异常可能会关闭连接
throw checkException(e);
}
} finally {
// 把deletgate改为CLOSED_CONNECTION
delegate = ClosedConnection.CLOSED_CONNECTION;
// 归还PoolEntry
poolEntry.recycle(lastAccess);
}
}
}
PoolEntry#recycle
final class PoolEntry implements IConcurrentBagEntry {
long lastAccessed;
private final HikariPool hikariPool;
void recycle(final long lastAccessed) {
if (connection != null) {
this.lastAccessed = lastAccessed;
hikariPool.recycle(this);
}
}
}
HikariPool#recycle
将PoolEntry归还给ConcurrentBag。
private final ConcurrentBag connectionBag;@Overridevoid recycle(final PoolEntry poolEntry) {
connectionBag.requite(poolEntry);
}
ConcurrentBag#requite
修改Entry状态。
public void requite(final T bagEntry) {
// 修改entry状态为未使用,此时已经可以被别的线程借走了,因为entry在shareList里
bagEntry.setState(STATE_NOT_IN_USE);
// 先尝试通过交接队列给等候线程发放entry
for (int i = 0; waiters.get() > 0; i++) {
if (bagEntry.getState() != STATE_NOT_IN_USE || handoffQueue.offer(bagEntry)) {
return;
} else if ((i & 0xff) == 0xff) {
parkNanos(MICROSECONDS.toNanos(10));
} else {
Thread.yield();
}
}
// 如果没有线程等候,或发放失败,放入threadList
final List<Object> threadLocalList = threadList.get();
if (threadLocalList.size() 50) {
threadLocalList.add(weakThreadLocals ? new WeakReference<>(bagEntry) : bagEntry);
}
}
Hikari连接池如何实现借走和归还?