zookeeper之服务端接收数据请求和集群模式下的处理流程

服务端接收数据请求

服务端收到的数据包应该在哪里呢？在上节课分析过了，zookeeper 启动的时候，通过下面的代码构建了一个ServerCnxnFactory cnxnFactory = ServerCnxnFactory.createFactory();

NIOServerCnxnFactory，它实现了 Thread，所以在启动的时候，会在 run 方法中不断循环接收客户端的请求进行分发。

NIOServerCnxnFactory.run

public void run() {
        while (!ss.socket().isClosed()) {
            try {
                for (SelectionKey k : selectedList) {
                    // 获取 client 的连接请求
                    if ((k.readyOps() & SelectionKey.OP_ACCEPT) != 0) {
 
                    } else if ((k.readyOps() & (SelectionKey.OP_READ | SelectionKey.OP_WRITE)) != 0) {
                        // 处理客户端的读/写请求
                        NIOServerCnxn c = (NIOServerCnxn) k.attachment();
                        c.doIO(k);// 处理 IO 操作
                    } else {
                        if (LOG.isDebugEnabled()) {
                            LOG.debug("Unexpected ops in select " + k.readyOps());
                        }
                    }
                }
                selected.clear();
            } catch (RuntimeException e) {
                LOG.warn("Ignoring unexpected runtime exception", e);
            } catch (Exception e) {
                LOG.warn("Ignoring exception", e);
            }
        }
        closeAll();
        LOG.info("NIOServerCnxn factory exited run method");
    }

NIOServerCnxn.doIO

void doIO(SelectionKey k) {
        try {
            //省略部分代码..
            if (k.isReadable()) {//处理读请求，表示接收
                //中间这部分逻辑用来处理报文以及粘包问题
                if (isPayload) { // not the case for 4letterword
                    readPayload();//处理报文
                }
                else {
                    // four letter words take care
                    // need not do anything else
                    return;
                }
            }
        }
    }

NIOServerCnxn.readRequest

读取客户端的请求，进行具体的处理

private void readRequest() throws IOException {
     zkServer.processPacket(this, incomingBuffer);
}

ZookeeperServer.processPacket

这个方法根据数据包的类型来处理不同的数据包，对于读写请求，我们主要关注下面这块代码即可：

Request si = new Request(cnxn, cnxn.getSessionId(), h.getXid(),
            h.getType(), incomingBuffer, cnxn.getAuthInfo());
            si.setOwner(ServerCnxn.me);
            submitRequest(si);

后续的流程，在前面的源码分析中有些，就不做重复粘贴了。

集群模式下的处理流程

集群模式下，涉及到 zab 协议，所以处理流程比较复杂，大家可以基于这个图来定位代码的流程：