Undertow服务器基础分析 - XNIO
阅读原文时间:2023年07月09日阅读:1

阅读更多

我们从名字上就能看出这是一个NIO思想为基础的IO框架,X是指这个框架可以有多种实现,我们可以从代码库 https://github.com/xnio 中发现一个项目xnio-native,里面有用C实现的nio层,就能体会到这个X的含义,可以直接基于操作系统C库。目前在Xnio中默认的实现是nio-impl,也就是JDK的NIO。我们可以认为xnio是在JDK的NIO之上,进行了扩展,融入了一些JBoss开发人员对于并发访问异步通信的理解和思考,总结出一套API,并用基于JDK NIO进行实现。

要学习XNIO,必须得有JDK NIO的基础知识,本文假设读者已经学习过NIO,如果还没有可以阅读参考书籍[1],[2]。另外对Netty有所了解的话,就会融会贯通,比较容易的理解XNIO的基本原理,因为两个项目有很多相似之处。

XNIO有两个重要的概念:
1. Channel,是传输管道的抽象概念,在NIO的Channel上进行的扩展加强,使用ChannelListener API进行事件通知。在创建Channel时,就赋予IO线程,用于执行所有的ChannelListener回调方法。
2. 区分IO线程和工作线程,创建一个工作线程池可以用来执行阻塞任务。一般情况下,非阻塞的Handler由IO线程执行,而阻塞任务比如Servlet则被调度到工作线程池执行。这样就很好的区分了阻塞和非阻塞的两种情形。

我们知道NIO的基本要求是不阻塞当前线程的执行,对于非阻塞请求的结果,可以用两种方式获得:一种是对于请求很快返回一个引用(如JDK中Future,XNIO中称为IoFuture,其中很多方法是类似的),过一段时间再查询结果;还有一种是当结果就绪时,调用事先注册的回调方法来通知(如NIO2的CompletionHandler,XNIO的ChannelListener)。显而易见后者效率更高一些,避免了数据未就绪情景下的无用处理过程。但JDK7之前无法将函数作为方法参数,所以只能用Java的匿名内部类来模拟函数式方法,造成代码嵌套层次过多,难以理解和维护,所以Netty和XNIO这样的框架通过调度方法调用过程,简化了编程工作。

XNIO和Netty的最主要的一个区别是,XNIO继承重用了JDK NIO的ByteBuffer类,而不像Netty另起炉灶,完全重建自己的ByteBuf体系。我们知道NIO的ByteBuffer使用时有个状态切换的过程,读和写要显式的通过调用slice, reset等方法切换,就和unix使用vi编辑器编辑和处理文本需要用'i'和Esc切换状态类似。Netty通过读写指针索引值移除了这个“不便操作”,但XNIO保留了和JDK NIO一致的做法。

无论NIO还是Netty,都有heap buffer和direct buffer的概念,前者可以认为是byte数组的封装,缓冲区存放在堆上,后者可以直接通过调用操作系统的系统调用在内存上分配缓冲,这样在一些IO操作时,比如从网卡上读出大量数据,再写到硬盘文件中,就不必拷贝数据到应用层,直接在操作系统内核或者驱动上进行数据复制。ByteBuffer的管理和应用是NIO最核心的思想,开发人员应该根据应用类型,对其反复调优,做到占用资源最少和效率最大。

XNIO和Netty都对ByteBuffer进行池化管理,简单来说就是开发者在程序开始时就计划好读写缓存区大小,统一分配好放到池中,Xnio中有Pool和Pooled接口用来管理池化缓存区。开发过高并发应用就知道,JVM GC经常出现并难以控制是很头疼的问题。我们通常在接收网络数据时,往往简单的new出一块数据区,填充,解析,使用,最后丢弃,这种方法随着大量的数据读入,必然造成GC反复出现。重用缓存区就可以在这个方面解决一部分问题。

和Netty的ChannelHandler不同,XNIO对应的ChannelListener只有一个方法handleEvent(),也就意味着所有的事件都要经由这个方法。在实际实行过程中,会进行若干状态机的转变,比如在服务器端,开始时accept状态就绪,当连接建立后转变为可读或者可写状态。请参见下面的例子。

在一些情况下,阻塞的IO调用也是很有用的,比如事务过程中。XNIO也提供了阻塞方法awaitReadable()和awaitWritable()。

利用Stream channel,可以在数据源头和目的地之间直接读写数据,有一种zero-copy(零拷贝)的方式,即读写过程使用同一块缓存区,这样就不必进行数据拷贝移动过程。XNIO通过封装NIO FileChannel中的方法transferTo和transferFrom来实现。

还有一种Messgae channel,用来传输帧数据,因为有些报文格式是固定长度或者按照某种已知帧式格式定义的,缓冲区的长度可以按照报文帧来定义,当数据填满后就及时发送,减少了数据长度的计算工作,代码逻辑简洁很多,所以这种channel用于传递'消息',websocket就是这样的。

阅读XNIO时,有几个词出现频率很高,Source表示信息源头,Sink是信息目的地,Conduit是源头到目的地管道的抽象。

前面提过,XNIO中有两类线程:
WORKER_IO_THREADS, IO thread处理非阻塞任务,要保证不做阻塞操作,因为很多连接同时用到这类线程,类似于nodejs中的loop,这个线程只要有任务就去执行,实际配置时每个CPU一个线程比较好。
WORKER_TASK_CORE_THREADS,用于执行阻塞任务,从线程池中获得,任务完成后返回到线程池中。因为不同应用对应的服务器负载不同,所以不易给出具体数值,一般建议每个CPU core设置10个。

代码分析,摘自
https://github.com/ecki/xnio-samples/blob/master/src/main/java/org/xnio/samples/SimpleEchoServer.java

服务器:

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;

import org.xnio.ChannelListener;

import org.xnio.IoUtils;

import org.xnio.OptionMap;

import org.xnio.Xnio;

import org.xnio.XnioWorker;

import org.xnio.channels.AcceptingChannel;

import org.xnio.channels.Channels;

import org.xnio.channels.ConnectedStreamChannel;

public final class SimpleEchoServer {

public static void main(String[] args) throws Exception {

    // 定义读数据listener
    final ChannelListener<ConnectedStreamChannel> readListener =
        new ChannelListener<ConnectedStreamChannel>() {
        public void handleEvent(ConnectedStreamChannel channel) {
            //分配缓冲
            final ByteBuffer buffer = ByteBuffer.allocate(512);
            int res;
            try {
                while ((res = channel.read(buffer)) > 0) {
                    //切换到写的状态并用阻塞的方式写回
                    buffer.flip();
                    Channels.writeBlocking(channel, buffer);
                }
                // 保证全部送出
                Channels.flushBlocking(channel);
                if (res == -1) {
                    channel.close();
                } else {
                    channel.resumeReads();
                }
            } catch (IOException e) {
                e.printStackTrace();
                IoUtils.safeClose(channel);
            }
        }
    };
    // 创建接收 listener.
    final ChannelListener<AcceptingChannel<ConnectedStreamChannel>> acceptListener =
        new ChannelListener<AcceptingChannel<ConnectedStreamChannel>>() {
        public void handleEvent(
                final AcceptingChannel<ConnectedStreamChannel> channel) {
            try {
                ConnectedStreamChannel accepted;
                // channel就绪,准备接收连接请求
                while ((accepted = channel.accept()) != null) {
                    System.out.println("accepted " + accepted.getPeerAddress());
                    // 已经连接,设置读数据listener
                    accepted.getReadSetter().set(readListener);
                    // 恢复读的状态
                    accepted.resumeReads();
                }
            } catch (IOException ignored) {
            }
        }
    };

    //创建Xnio实例,并构造XnioWorker
    final XnioWorker worker = Xnio.getInstance().createWorker(OptionMap.EMPTY);
    // 创建server,在本地12345端口上侦听
    AcceptingChannel<? extends ConnectedStreamChannel> server = worker
            .createStreamServer(new InetSocketAddress(12345),
                    acceptListener, OptionMap.EMPTY);
    // 开始接受连接
    server.resumeAccepts();
    System.out.println("Listening on " + server.getLocalAddress());
}

}

客户端:

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.charset.Charset;

import org.xnio.IoFuture;

import org.xnio.IoUtils;

import org.xnio.OptionMap;

import org.xnio.Xnio;

import org.xnio.XnioWorker;

import org.xnio.channels.Channels;

import org.xnio.channels.ConnectedStreamChannel;

public final class SimpleHelloWorldBlockingClient {

public static void main(String[] args) throws Exception {
    final Charset charset = Charset.forName("utf-8");
    //创建Xnio实例,并构造XnioWorker
    final Xnio xnio = Xnio.getInstance();
    final XnioWorker worker = xnio.createWorker(OptionMap.EMPTY);

    try {
        //连接服务器,本地12345端口,注意返回值是IoFuture类型,并不阻塞,返回后可以做些别的事情
        final IoFuture<ConnectedStreamChannel> futureConnection = worker.connectStream(
            new InetSocketAddress("localhost", 12345), null, OptionMap.EMPTY);
        final ConnectedStreamChannel channel = futureConnection.get(); // get是阻塞调用
        try {
            // 发送消息
            Channels.writeBlocking(channel, ByteBuffer.wrap("Hello world!\n".getBytes(charset)));
            // 保证全部送出
            Channels.flushBlocking(channel);
            // 发送EOF
            channel.shutdownWrites();
            System.out.println("Sent greeting string! The response is...");
            ByteBuffer recvBuf = ByteBuffer.allocate(128);
            // 接收消息
            while (Channels.readBlocking(channel, recvBuf) != -1) {
                recvBuf.flip();
                final CharBuffer chars = charset.decode(recvBuf);
                System.out.print(chars);
                recvBuf.clear();
            }
        } finally {
            IoUtils.safeClose(channel);
        }
    } finally {
        worker.shutdown();
    }
}

}

[1]: JavaNIO http://www.amazon.com/Java-Nio-Ron-Hitchens/dp/0596002882
[2]: Pro Java 7 NIO.2 http://www.amazon.com/Pro-Java-NIO-2-Anghel-Leonard/dp/1430240113