基于ZooKeeper的分布式Session实现-已发布

基于ZooKeeper的分布式Session实现

1． 认识ZooKeeper

ZooKeeper—— “动物园管理员”。动物园里当然有好多的动物，游客可以根据动物园提供的向导图到不同的场馆观赏各种类型的动物，而不是像走在原始丛林里，心惊胆颤的被动 物所观赏。为了让各种不同的动物呆在它们应该呆的地方，而不是相互串门，或是相互厮杀，就需要动物园管理员按照动物的各种习性加以分类和管理，这样我们才 能更加放心安全的观赏动物。回到我们企业级应用系统中，随着信息化水平的不断提高，我们的企业级系统变得越来越庞大臃肿，性能急剧下降，客户抱怨频频。拆 分系统是目前我们可选择的解决系统可伸缩性和性能问题的唯一行之有效的方法。但是拆分系统同时也带来了系统的复杂性——各子系统不是孤立存在的，它们彼此 之间需要协作和交互，这就是我们常说的分布式系统。各个子系统就好比动物园里的动物，为了使各个子系统能正常为用户提供统一的服务，必须需要一种机制来进 行协调——这就是ZooKeeper——动物园管理员。

关于ZooKeeper更正式的介绍——ZooKeeper是一个为分布式应用程序提供高性能协调服务的工具集合。它可以应用在一些需要提供统一协调服务的case中，例如命名、配置管理、同步和组服务等。而在我们的case中，它被作为一个协调分布式环境中各子系统之间共享状态数据的基础设施。

2． ZooKeeper之特性

ZooKeeper本质上是一个分布式的小文件存储系统。原本是Apache Hadoop的一个组件，现在被拆分为一个Hadoop的独立子项目，在HBase

（Hadoop的另外一个被拆分出来的子项目，用于分布式环境下的超大数据量的DBMS）中也用到了ZooKeeper集群。ZooKeeper有如下的特性：

1） 简单

ZooKeeper核心是一个精简的文件系统，它提供了一些简单的文件操作以及附加的功能，例如排序和通知。

2） 易表达

ZooKeeper的数据结构原型是一棵znode树（类似Linux的文件系统），并且它们是一些已经被构建好的块，可以用来构建大型的协作数据结构和协议。

3） 高可用性

ZooKeeper可以运行在一组服务器上，同时它们被设计成高可用性，为你的应用程序避免单点故障。

4） 松耦合交互

ZooKeeper提供的Watcher机制使得各客户端与服务器的交互变得松耦合，每个客户端无需知晓其他客户端的存在，就可以和其他客户端进行数据交互。

5） 丰富的API

ZooKeeper为开发人员提供了一套丰富的API，减轻了开发人员编写通用协议的负担。

这篇文章是关于如何在ZooKeeper上创建分布式Session系统，所以关于ZooKeeper的安装、使用、管理等主题不在本文的讨论范围内，如果想了解ZooKeeper更加详细的情况，请看另外一篇文章《ZooKeeper实战》。

3． 为什么使用ZooKeeper

目前有关于分布式Session的实现基本上都是基于memcached。memcached本质上是一个内存缓存系统。虽然memcached也可以是分布式集群环境的，但是对于一份数据来说，它总是存储在某一台memcached服务器上。如果发生网络故障或是服务器当机，则存储在这台服务器上的所有数据都将不可访问。由于数据是存储在内存中的，重启服务器，将导致数据全部丢失。当然你可以自己实现一套机制，用来在分布式memcached之间进行数据的同步和持久化，但是实现这套机制谈何容易！

由上述ZooKeeper的特性可知，ZooKeeper是一个分布式小文件系统，并且被设计为高可用性。通过选举算法和集群复制可以避免单点故障，由于是文件系统，所以即使所有的ZooKeeper节点全部挂掉，数据也不会丢失，重启服务器之后，数据即可恢复。另外ZooKeeper的节点更新是原子的，也就是说更新不是成功就是失败。通过版本号，ZooKeeper实现了更新的乐观锁，当版本号不相符时，则表示待更新的节点已经被其他客户端提前更新了，而当前的整个更新操作将全部失败。当然所有的一切ZooKeeper已经为开发者提供了保障，我们需要做的只是调用API。

有人会怀疑ZooKeeper的执行能力，在ZooKeeper诞生的地方——Yahoo!给出了一组数据将打消你的怀疑。它的吞吐量标准已经达到大约每秒10000基于写操作的工作量。对于读操作的工作量来说，它的吞吐量标准还要高几倍。

4． 实现分布式Session所面临的挑战

实现分布式session最大的挑战莫过于如何实现session在分布式系统之间的共享。在分布式环境下，每个子系统都是跨网络的独立JVM，在这些JVM之间实现共享数据的方式无非就是TCP/IP通讯。无论是memcached，还是ZooKeeper，底层都是基于TCP/IP的。所以，我认为使用何种工具实现分布式Session都是可行的，没有那种实现优于另外一种实现，在不同的应用场景，各有优缺点。世间万物，无十全十美，不要盲目的崇拜某种技术，唯有适合才是真理。

1） Session ID的共享

Session ID是一个实例化Session对象的唯一标识，也是它在Web容器中可以被识别的唯一身份标签。Jetty和Tomcat容器会通过一个Hash算法，得到一个唯一的ID字符串，然后赋值给某个实例化的Session，此时，这个Session就可以被放入Web容器的

SessionManager中开始它短暂的一生。在Servlet中，我们可以通过HttpSession的getId()方法得到这个值，但是我们无法改变这个值。当Session走到它一生尽头的时候，Web容器的SessionManager会根据这个ID将其“火化”。所以Session ID是非常重要的一个属性，并且要保证它的唯一性。在单系统中，Session ID只需要被自身的Web容器读写，但是在分布式环境中，多个Web容器需要共享同一个Session ID。因此，当某个子系统的Web容器产生一个新的ID时，它必须需要一种机制来通知其他子系统，并且告知新ID是什么。

2） Session中数据的复制

和共享Session ID的问题一样，在分布式环境下，Session中的用户数据也需要在各个子系统中共享。当用户通过HttpSession的setAttribute()方法在Session中设置了一个用户数据时，它只存在于当前与用户交互的那个Web容器中，而对其他子系统的Web容器来说，这些数据是不可见的。当用户在下一步跳转到另外一个Web容器时，则又会创建一个新的Session对象，而此Session中并不包含上一步骤用户设置的数据。其实Session在分布式系统之间的复制实现是简单的，但是每次在Session数据发生变化时，都在子系统之间复制一次数据，会大大降低用户的响应速度。因此我们需要一种机制，即可以保证Session数据的一致性，又不会降低用户操作的响应度。

3） Session的失效

Session是有生命周期的，当Session的空闲时间(maxIdle属性值)超出限制时，Session就失效了，这种设计主要是考虑到了Web容器的可靠性。当一个系统有上万人使用时，就会产生上万个Session对象，由于HTTP的无状态特性，服务器无法确切的知道用户是否真的离开了系统。因此如果没有失效机制，所有被Session占据的内存资源将永远无法被释放，直到系统崩溃为止。在分布式环境下，Session被简单的创建，并且通过某种机制被复制到了其他系统中。你无法保证每个子系统的时钟都是一致的，可能相差几秒，甚至相差几分钟。当某个Web容器的Session失效时，可能其他的子系统中的Session并未失效，这时会产生一个有趣的现象，一个用户在各个子系统之间跳转时，有时会提示Session超时，而有时又能正常操作。因此我们需要一种机制，当某个系

统的Session失效时，其他所有系统的与之相关联的Session也要同步失效。

4） 类装载问题

在单系统环境下，所有类被装载到“同一个”ClassLoader中。我在同一个上打了引号，因为实际上并非是同一个ClassLoader，只是逻辑上我们认为是同一个。这里涉及到了JVM的类装载机制，由于这个主题不是本文的讨论重点，所以相关详情可以参考相关的JVM文档。因此即使是由memcached或是ZooKeeper返回的字节数组也可以正常的反序列化成相对应的对象类型。但是在分布式环境下，问题就变得异常的复杂。我们通过一个例子来描述这个问题。用户在某个子系统的Session中设置了一个User类型的对象，通过序列化，将User类型的对象转换成字节数组，并通过网络传输到了memcached或是ZooKeeper上。此时，用户跳转到了另外一个子系统上，需要通过getAttribute方法从memcached或是ZooKeeper上得到先前设置的那个User类型的对象数据。但是问题出现了，在这个子系统的ClassLoader中并没有装载User类型。因此在做反序列化时出现了ClassNotFoundException异常。

当然上面描述的4点挑战只是在实现分布式Session过程中面临的关键问题，并不是全部。其实在我实现分布式Session的整个过程中还遇到了其他的一些挑战。比如，需要通过filter机制拦截HttpServletRequest，以便覆盖其getSession方法。但是在不同的Web容器中（例如Jetty或是Tomcat）对HttpServletRequest的实现是不一样的，虽然都是实现了

HttpServletRequest接口，但是各自又添加了一些特性在其中。例如，在Jetty容器中，HttpSession的实现类是一个保护内部类，无法从其继承并覆盖相关的方法，只能从其实现类的父类中继承更加抽象的Session实现。这样就会造成一个问题，我必须重新实现对Session整个生命周期管理的SessionManager接口。有人会说，那就放弃它的实现吧，我们自己实现HttpSession接口。很不幸，那是不可能的。因为在Jetty的

HttpServletRequest实现类的一些方法中对Session的类型进行了强制转换（转换成它自定义的HttpSession实现类），如果不从其继承，则会出现ClassCastException异常。相比之下，Tomcat的对HttpServletRequest和HttpSession接口的实现还是比较标准的。由此可见，实现分布式Session其实是和某种Web容器紧密耦合的。并不像网上有些人的轻描淡写，仅仅覆盖setAttribute和getAttribute方法是行不通的。

5． 算法实现

从上述的挑战来看，要写一个分布式应用程序是困难的，主要原因是因为局部故障。由于数据需要通过网络传输，而网络是不稳定的，所以如果网络发生故障，则所有的数据通讯都将终止。ZooKeeper并不能解决网络故障的发生，甚至它本身也是基于网络的分布式应用程序。但是它为我们提供了一套工具集合，帮助我们建立安全处理局部故障的分布式应用程序。接下来我们就开始描述如何实现基于ZooKeeper的分布式Session系统。

1） 基于ZooKeeper的分布式Session系统架构

为了实现高可用性，采用了ZooKeeper集群，ZooKeeper集 群是由一台领导者服务器和若干台跟随者服务器构成（总服务器数要奇数）。所有的读操作由跟随者提供，而写操作由领导者提供，并且领导者还负责将写入的数据 复制到集群中其他的跟随者。当领导者服务器由于故障无法访问时，剩下的所有跟随者服务器就开始进行领导者的选举。通过选举算法，最终由一台原本是跟随者的 服务器升级为领导者。当然原来的领导者服务器一旦被恢复，它就只能作为跟随者服务器，并在下一次选举中争夺领导者的位置。

Web容器中的Session容器也将发生变化。它不再对用户的Session进行本地管理，而是委托给ZooKeeper和我们自己实现的Session管理器。也就是说，ZooKeeper负责Session数据的存储，而我们自己实现的Session管理器将负责Session生命周期的管理。

最后是关于在分布式环境下共享Session ID的策略。我们还是通过客户端的Cookie来实现，我们会自定义一个Cookie，并通过一定的算法在多个子系统之间进行共享。下面会对此进行详细的描述。

2） 分布式Session的数据模型

Session数据的存储是有一定格式的，下图展示了一个Session ID为”1gyh0za3qmld7”的Session在ZooKeeper上的存储结构：

“/SESSIONS”是一个组节点，用来在ZooKeeper上划分不同功能组的定义。你可以把它理解为一个文件夹目录。在这个目录下可以存放0个或N个子节点，我们就把一个Session的实例作为一个节点，节点的名称就是Session ID。在ZooKeeper中，每个节点本身也可以存放一个字节数组。因此，每个节点天然就是一个Key-Value键值对的数据结构。

我们将Session中的用户数据（本质上就是一个Map）设计成多节点，节点名称就是Session的key，而节点的数据就是Session的Value。采用这种设计主要是考虑到性能问题和ZooKeeper对节点大小的限制

问题。当然，我们可以将Session中的用户数据保存在一个Map中，然后将Map序列化之后存储在对应的Session节点中。但是大部分情况下，我们在读取数据时并不需要整个Map，而是Map中的一个或几个值。这样就可以避免一个非常大的Map在网络间传来传去。同理，在写Session的时候，也可以最大限度的减少数据流量。另外由于ZooKeeper是一个小文件系统，为了性能，每个节点的大小为1MB。如果Session中的Map大于1MB，就不能单节点的存储了。当然，一个Key的数据量是很少会超过1MB的，如果真的超过1MB，你就应该考虑一下，是否应该将此数据保存在Session中。

最后我们来关注一下Session节点中的数据——SessionMetaData。它是一个Session实例的元数据，保存了一些与Session生命周期控制有关的数据。以下代码就是SessionMetaData的实现： public class SessionMetaData implements Serializable { private static final long serialVersionUID = -6446174402446690125L;

private String id; /**session的创建时间*/

private Long createTm; /**session的最大空闲时间*/ private Long maxIdle; /**session的最后一次访问时间*/ private Long lastAccessTm;

/**是否可用*/

private Boolean validate = false; /**当前版本*/

private int version = 0; /** * 构造方法 */

public SessionMetaData() {

this.createTm = System.currentTimeMillis(); this.lastAccessTm = this.createTm; this.validate = true; }

……以下是N多getter和setter方法

其中需要关注的属性有：

a) id属性：Session实例的ID。

b) maxIdle属性：Session的最大空闲时间，默认情况下是30分钟。

c) lastAccessTm属性：Session的最后一次访问时间，每次调用Request.getSession方法时都会去更新这个值。用来计算当前

Session是否超时。如果lastAccessTm+maxIdle小于System. currentTimeMillis()，就表示当前Session超时。

d) validate属性：表示当前Session是否可用，如果超时，则此属性为false。

e) version属性：这个属性是为了冗余Znode的version值，用来实现乐观锁，对Session节点的元数据进行更新操作。 这里有必要提一下一个老生常谈的问题，就是所有存储在节点上的对象必须是可序列化的，也就是必须实现Serializable接口，否则无法保存。这个问题在memcached和ZooKeeper上都存在的。

3） 实现过程

实现分布式Session的第一步就是要定义一个filter，用来拦截HttpServletRequest对象。以下代码片段，展现了在Jetty容器下的filter实现。

public class JettyDistributedSessionFilter extends DistributedSessionFilter {

private Logger log = Logger.getLogger(getClass());

@Override

public void init(FilterConfig filterConfig) throws ServletException {

super.init(filterConfig);

// 实例化Jetty容器下的Session管理器

sessionManager = new JettyDistributedSessionManager(conf); try {

sessionManager.start(); // 启动初始化 //创建组节点

ZooKeeperHelper.createGroupNode();

log.debug(\); } catch (Exception e) { log.error(e); } }

@Override

public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)

throws IOException,

ServletException {

// Jetty容器的Request对象包装器，用于重写Session的相关操作 JettyRequestWrapper req

= new JettyRequestWrapper(request, sessionManager);

chain.doFilter(req, response);

} }

这个filter是继承自DistributedSessionFilter的，这个父类主要是负责完成初始化参数设置等通用方法的实现，代码如下所示： public abstract class DistributedSessionFilter implements Filter { protected Logger log = Logger.getLogger(getClass()); /**参数配置*/

protected Configuration conf; /**Session管理器*/

protected SessionManager sessionManager; /**初始化参数名称*/

public static final String SERVERS = \; public static final String TIMEOUT = \; public static final String POOLSIZE = \; /** * 初始化

* @see javax.servlet.Filter#init(javax.servlet.FilterConfig) */

@Override

public void init(FilterConfig filterConfig) throws ServletException {

conf = new Configuration();

String servers = filterConfig.getInitParameter(SERVERS); if (StringUtils.isNotBlank(servers)) { conf.setServers(servers); }

String timeout = filterConfig.getInitParameter(TIMEOUT); if (StringUtils.isNotBlank(timeout)) { try {

conf.setTimeout(Long.valueOf(timeout)); } catch (NumberFormatException ex) {

log.error(\ + timeout + \); } }

String poolsize = filterConfig.getInitParameter(POOLSIZE); if (StringUtils.isNotBlank(poolsize)) { try {

conf.setPoolSize(Integer.valueOf(poolsize)); } catch (NumberFormatException ex) {

log.error(\ + poolsize + \); } }

//初始化ZooKeeper配置参数

} } }

return sessionMap; } catch (KeeperException e) { log.error(e);

} catch (InterruptedException e) { log.error(e); } finally { close(zk); } }

return null; } /**

* 创建一个组节点 */

public static void createGroupNode() { ZooKeeper zk = connect(); if (zk != null) { try {

// 检查节点是否存在

Stat stat = zk.exists(GROUP_NAME, false); //stat为null表示无此节点，需要创建 if (stat == null) { // 创建组件点

String createPath = zk.create(GROUP_NAME, null, Ids.OPEN_ACL_UNSAFE,

CreateMode.PERSISTENT);

log.debug(\创建节点完成:[\ + createPath + \); } else {

log.debug(\组节点已存在，无需创建[\ + GROUP_NAME + \);

}

} catch (KeeperException e) { log.error(e);

} catch (InterruptedException e) { log.error(e); } finally { close(zk); } } }

/**

* 创建指定Session ID的节点 * @param sid Session ID * @return */

public static String createSessionNode(SessionMetaData metadata) {

if (metadata == null) { return null; }

ZooKeeper zk = connect(); //连接服务期 if (zk != null) {

String path = GROUP_NAME + \ + metadata.getId(); try {

// 检查节点是否存在

Stat stat = zk.exists(path, false); //stat为null表示无此节点，需要创建 if (stat == null) { // 创建组件点

String createPath = zk.create(path, null, Ids.OPEN_ACL_UNSAFE,

CreateMode.PERSISTENT);

log.debug(\创建Session节点完成:[\ + createPath + \); //写入节点数据

zk.setData(path, SerializationUtils.serialize(metadata), -1);

return createPath; }

} catch (KeeperException e) { log.error(e);

} catch (InterruptedException e) { log.error(e); } finally { close(zk); } }

return null; } /**

* 创建指定Session ID的节点(异步操作) * @param sid

* @param waitFor 是否等待执行结果

* @return */

public static String

asynCreateSessionNode(final SessionMetaData metadata, boolean waitFor) {

Callable task = new Callable () { @Override

public String call() throws Exception { return createSessionNode(metadata); } }; try {

Future result = pool.submit(task); //如果需要等待执行结果 if (waitFor) { while (true) {

if (result.isDone()) { return result.get(); } } }

} catch (Exception e) {

log.error(e); }

return null; } /**

* 删除指定Session ID的节点 * @param sid Session ID * @return */

public static boolean deleteSessionNode(String sid) { ZooKeeper zk = connect(); //连接服务期 if (zk != null) {

String path = GROUP_NAME + \ + sid; try {

// 检查节点是否存在

Stat stat = zk.exists(path, false); //如果节点存在则删除之 if (stat != null) { //先删除子节点

List nodes = zk.getChildren(path, false); if (nodes != null) {

for (String node : nodes) {

zk.delete(path + \ + node, -1); } }

//删除父节点 zk.delete(path, -1);

log.debug(\删除Session节点完成:[\ + path + \); return true; }

} catch (KeeperException e) { log.error(e);

} catch (InterruptedException e) { log.error(e); } finally { close(zk); } }

return false; } /**

* 删除指定Session ID的节点(异步操作)

* @param sid

* @param waitFor 是否等待执行结果 * @return */

public static boolean asynDeleteSessionNode(final String sid, boolean waitFor) {

Callable task = new Callable () { @Override

public Boolean call() throws Exception { return deleteSessionNode(sid); } }; try {

Future result = pool.submit(task); //如果需要等待执行结果 if (waitFor) { while (true) {

if (result.isDone()) { return result.get(); } } }

} catch (Exception e) { log.error(e); }

return false; } /**

* 在指定Session ID的节点下添加数据节点 * @param sid Session ID * @param name 数据节点的名称 * @param value 数据 * @return */

public static boolean setSessionData(String sid, String name, Object value) {

boolean result = false;

ZooKeeper zk = connect(); //连接服务器 if (zk != null) {

String path = GROUP_NAME + \ + sid; try {

// 检查指定的Session节点是否存在 Stat stat = zk.exists(path, false);

//如果节点存在则删除之 if (stat != null) {

//查找数据节点是否存在，不存在就创建一个 String dataPath = path + \ + name; stat = zk.exists(dataPath, false); if (stat == null) { //创建数据节点

zk.create(dataPath, null, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);

log.debug(\创建数据节点完成[\ + dataPath + \); }

//在节点上设置数据，所有数据必须可序列化 if (value instanceof Serializable) { int dataNodeVer = -1; if (stat != null) { //记录数据节点的版本

dataNodeVer = stat.getVersion(); }

byte[] data = SerializationUtils.serialize((Serializable) value);

stat = zk.setData(dataPath, data, dataNodeVer);

log.debug(\更新数据节点数据完成[\ + dataPath + \ + value + \);

result = true; } }

} catch (KeeperException e) { log.error(e);

} catch (InterruptedException e) { log.error(e); } finally { close(zk); } }

return result; } /**

* 删除指定Session ID的节点(异步操作) * @param sid

* @param waitFor 是否等待执行结果 * @return */

public static boolean asynSetSessionData(final String sid, final String name,

final Object value, boolean waitFor) { Callable task = new Callable () { @Override

public Boolean call() throws Exception { return setSessionData(sid, name, value); } }; try {

Future result = pool.submit(task); //如果需要等待执行结果 if (waitFor) { while (true) {

if (result.isDone()) { return result.get(); } } }

} catch (Exception e) { log.error(e); }

return false; } /**

* 返回指定Session ID的节点下数据 * @param sid Session ID * @param name 数据节点的名称 * @param value 数据 * @return */

public static Object getSessionData(String sid, String name) { ZooKeeper zk = connect(); //连接服务器 if (zk != null) {

String path = GROUP_NAME + \ + sid; try {

// 检查指定的Session节点是否存在 Stat stat = zk.exists(path, false); if (stat != null) {

//查找数据节点是否存在

String dataPath = path + \ + name; stat = zk.exists(dataPath, false); Object obj = null;

if (stat != null) { //获取节点数据

byte[] data = zk.getData(dataPath, false, null); if (data != null) { //反序列化

obj = SerializationUtils.deserialize(data); } }

return obj; }

} catch (KeeperException e) { log.error(e);

} catch (InterruptedException e) { log.error(e); } finally { close(zk); } }

return null; } /**

* 删除指定Session ID的节点下数据 * @param sid Session ID * @param name 数据节点的名称 * @param value 数据 * @return */

public static void removeSessionData(String sid, String name) { ZooKeeper zk = connect(); //连接服务器 if (zk != null) {

String path = GROUP_NAME + \ + sid; try {

// 检查指定的Session节点是否存在 Stat stat = zk.exists(path, false); if (stat != null) {

//查找数据节点是否存在

String dataPath = path + \ + name; stat = zk.exists(dataPath, false); if (stat != null) { //删除节点

zk.delete(dataPath, -1); } }

本文来源：https://www.bwwdw.com/article/ykw5.html