聊聊Zookeeper技术内幕之客户端与SetData请求处理-526互联

从客户端会话创建到网络连接、请求处理，简单的叙述下流程与逻辑

客户端

客户端是开发人员使用ZooKeeper最主要的途径，ZooKeeper的客户端主要由以下几个核心组件组成。

ZooKeeper实例：客户端的入口。
ClientWatchManager：客户端Watcher管理器。
HostProvider：客户端地址列表管理器(管理服务器地址信息)。
ClientCnxn：客户端核心线程，其内部又包含两个线程，即SendThread和EventThread。前者是一个I/O线程，主要负责ZooKeeper客户端和服务端之间的网络I/O通信；后者是一个事件线程，主要负责对服务端事件进行处理。

对如上的组件，需要有一个初始化的过程，而这就是Zookeeper构造函数执行的结果。

在初始化阶段中，会做如下的事情：

初始化Zookeeper对象。
通过调用ZooKeeper的构造方法来实例化一个ZooKeeper对象，在初始化过程中，会创建一个客户端的Watcher管理器：ClientWatchManager。
设置会话默认Watcher
如果在构造方法中传入了一个Watcher对象，那么客户端会将这个对象作为默认Watcher保存在ClientWatchManager中。
构造Zookeeper服务器地址列表管理器：HostProvider。
对于构造方法中传入的服务器地址，客户端会将其存放在服务器地址列表管理器HostProvider中。
创建并初始化客户端网络连接器：ClientCnxn。
ZooKeeper客户端首先会创建一个网络连接器ClientCnxn，用来管理客户端与服务器的网络交互。另外，客户端在创建ClientCnxn的同时，还会初始化客户端两个核心队列outgoingQueue和pendingQueue，分别作为客户端的请求发送队列和服务端响应的等待队列。
初始化SendThread和EventThread。
客户端会创建两个核心网络线程SendThread和EventThread，前者用于管理客户端和服务端之间的所有网络I/O，后者则用于进行客户端的事件处理。同时，客户端还会将ClientCnxnSocket分配给SendThread作为底层网络I/O处理器，并初始化EventThread的待处理事件队列waitingEvents，用于存放所有等待被客户端处理的事件。

Zookeeper的构造函数有蛮多，但最后都是调用此构造函数方法，其参数列表也正是对应着上述的功能模块的初始化。


public ZooKeeper(
        String connectString,
        int sessionTimeout,
        Watcher watcher,
        boolean canBeReadOnly,
        HostProvider hostProvider,
        ZKClientConfig clientConfig
    ) throws IOException {
        LOG.info(
            "Initiating client connection, connectString={} sessionTimeout={} watcher={}",
            connectString,
            sessionTimeout,
            watcher);

        this.clientConfig = clientConfig != null ? clientConfig : new ZKClientConfig();
        this.hostProvider = hostProvider;
        ConnectStringParser connectStringParser = new ConnectStringParser(connectString);

        cnxn = createConnection(
            connectStringParser.getChrootPath(),
            hostProvider,
            sessionTimeout,
            this.clientConfig,
            watcher,
            getClientCnxnSocket(),
            canBeReadOnly);
        cnxn.start();
    }

初始化过程的泳道图如下

请求处理(SetData请求)

在Zookeeper中，更新类的操作都属于事务操作，即这个操作需要事务保障的。

服务端对于SetData请求的处理，大体可以分为4大步骤，分别是请求的预处理、事务处理、事务应用和请求响应

流程逻辑大概如下所示：

预处理

I/O层接收来自客户端的请求。
判断是否是客户端“会话创建”请求。ZooKeeper对于每一个客户端请求，都会检查是否是“会话创建”请求。对于SetData请求，因为此时已经完成了会话创建，因此按照正常的事务请求进行处理。
将请求交给ZooKeeper的PrepRequestProcessor处理器进行处理。
创建请求事务头。
会话检查。客户端会话检查是指检查该会话是否有效，即是否已经超时。如果该会话已经超时，那么服务端就会向客户端抛出SessionExpiredException异常。
反序列化请求，并创建ChangeRecord记录。面对客户端请求，ZooKeeper首先会将其进行反序列化并生成特定的SetDataRequest请求。SetDataRequest请求中通常包含了数据节点路径path、更新的数据内容data和期望的数据节点版本version。同时，根据请求中对应的path，ZooKeeper会生成一个ChangeRecord记录，并放入outstandingChanges队列中。outstandingChanges队列中存放了当前ZooKeeper服务器正在进行处理的事务请求，以便ZooKeeper在处理后续请求的过程中需要针对之前的客户端请求的相关处理，例如对于“会话关闭”请求来说，其需要根据当前正在处理的事务请求来收集需要清理的临时节点。
ACL检查。由于当前请求是数据更新请求，因此ZooKeeper需要检查该客户端是否具有数据更新的权限。如果没有权限，那么会抛出NoAuthException异常。
数据版本检查。
创建请求事务体SetDataTxn。
保存事务操作到outstandingChanges队列中去。