分桶策略:
ZooKeeper的会话管理主要是由SessionTracker负责的,其采用了一种特殊的会话管理方式,称其为“分桶策略”。所谓分桶策略,是将超时时间相近的会议放到同一个桶中来进行管理,以减少管理的复杂度。在检查超时的会话时,只需要检查桶中剩下的会话即可(没有被转移走的会话全是超时的)。
会话超时时间的计算方法如下:
long expireTime = roundToInterval(System.currentTimeMillis() + timeout);
private long roundToInterval(long time) {
// We give a one interval grace period
return (time / expirationInterval + 1) * expirationInterval;
}
可见,超时检查的粒度就是expirationInterval,以这个粒度为单位,就可以分成多个桶,同一个桶中存放着超时时间一致的会话。
会话激活:
有了桶之后,接下来就是激活。每次客户端和ZooKeeper服务器之间有通信时,会话就会被激活,会话的超时时间就会被重新计算,然后会话就会被放到其他的桶中。顺便说一句,每个桶就是一个Set,SessionTracker还维护了另外一个Map:
HashMap sessionSets= new HashMap();
sessionSets存放超时时间到桶的映射。这样通过某个超时时间,就可以获取这个时间超时的所有会话了。这样一次可以检查多个会话。
会话检查:
@Override
synchronizedpublic void run() {
try {
while (running) {
currentTime =System.currentTimeMillis();
if (nextExpirationTime >currentTime) {
this.wait(nextExpirationTime - currentTime);
continue;
}
SessionSet set;
set =sessionSets.remove(nextExpirationTime);
if (set != null) {
for (SessionImpl s :set.sessions) {
setSessionClosing(s.sessionId);
expirer.expire(s);
}
}
nextExpirationTime +=expirationInterval;
}
} catch (InterruptedException e) {
handleException(this.getName(), e);
}
LOG.info("SessionTrackerImpl exited loop!");
}
从源代码中可以看出,分桶后,每次把一个桶中剩下的会话全认为是超时的即可(不超时的会话会被移走)。检查一个桶后,经过expirationInterval时间后,再检查下一个桶。
ZooKeeper的会话分桶策略,在一些需要管理大量超时会话的场景下,会非常有用。