ZooKeeper的会话分桶策略

分桶策略：

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的会话分桶策略，在一些需要管理大量超时会话的场景下，会非常有用。​