2.3 I/O 多路復用 - IO multiplexing

簡介：單個線程就可以同時處理多個網(wǎng)絡連接。內(nèi)核負責輪詢所有socket，當某個socket有數(shù)據(jù)到達了，就通知用戶進程。多路復用在Linux內(nèi)核代碼迭代過程中依次支持了三種調(diào)用，即SELECT、POLL、EPOLL三種多路復用的網(wǎng)絡I/O模型。下文將畫圖結合Java代碼解釋。

2.3.1 I/O 多路復用- select

簡介：有連接請求抵達了再檢查處理。

缺點：

• 句柄上限- 默認打開的FD有限制,1024個。
• 重復初始化-每次調(diào)用 select()，需要把 fd 集合從用戶態(tài)拷貝到內(nèi)核態(tài)，內(nèi)核進行遍歷。
• 逐個排查所有FD狀態(tài)效率不高。

服務端的select 就像一塊布滿插口的插排，client端的連接連上其中一個插口，建立了一個通道，然后再在通道依次注冊讀寫事件。一個就緒、讀或寫事件處理時一定記得刪除，要不下次還能處理。

public static void main(String[] args) throws IOException {
        ServerSocketChannel ssc = ServerSocketChannel.open();//管道型ServerSocket
        ssc.socket().bind(new InetSocketAddress(Constant.HOST, Constant.PORT));
        ssc.configureBlocking(false);//設置非阻塞
        System.out.println(" NIO single server started, listening on :" + ssc.getLocalAddress());
        Selector selector = Selector.open();
        ssc.register(selector, SelectionKey.OP_ACCEPT);//在建立好的管道上，注冊關心的事件 就緒
        while(true) {
            selector.select();
            Set<SelectionKey> keys = selector.selectedKeys();
            Iterator<SelectionKey> it = keys.iterator();
            while(it.hasNext()) {
                SelectionKey key = it.next();
                it.remove();//處理的事件，必須刪除
                handle(key);
            }
        }
    }
    private static void handle(SelectionKey key) throws IOException {
        if(key.isAcceptable()) {
                ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
                SocketChannel sc = ssc.accept();
                sc.configureBlocking(false);//設置非阻塞
                sc.register(key.selector(), SelectionKey.OP_READ );//在建立好的管道上，注冊關心的事件 可讀
        } else if (key.isReadable()) { //flip
            SocketChannel sc = null;
                sc = (SocketChannel)key.channel();
                ByteBuffer buffer = ByteBuffer.allocate(512);
                buffer.clear();
                int len = sc.read(buffer);
                if(len != -1) {
                    System.out.println("[" +Thread.currentThread().getName()+"] recv :"+ new String(buffer.array(), 0, len));
                }
                ByteBuffer bufferToWrite = ByteBuffer.wrap("HelloClient".getBytes());
                sc.write(bufferToWrite);
        }
    }

2.3.2 I/O 多路復用 – poll

簡介：設計新的數(shù)據(jù)結構(鏈表)提供使用效率。

poll和select相比在本質(zhì)上變化不大，只是poll沒有了select方式的最大文件描述符數(shù)量的限制。

缺點：逐個排查所有FD狀態(tài)效率不高。

2.3.3 I/O 多路復用- epoll

簡介：沒有fd個數(shù)限制，用戶態(tài)拷貝到內(nèi)核態(tài)只需要一次，使用事件通知機制來觸發(fā)。通過epoll_ctl注冊fd，一旦fd就緒就會通過callback回調(diào)機制來激活對應fd，進行相關的I/O操作。

缺點：

• 跨平臺，Linux 支持最好。
• 底層實現(xiàn)復雜。
• 同步。

 public static void main(String[] args) throws Exception {
        final AsynchronousServerSocketChannel serverChannel = AsynchronousServerSocketChannel.open()
                .bind(new InetSocketAddress(Constant.HOST, Constant.PORT));
        serverChannel.accept(null, new CompletionHandler<AsynchronousSocketChannel, Object>() {
            @Override
            public void completed(final AsynchronousSocketChannel client, Object attachment) {
                serverChannel.accept(null, this);
                ByteBuffer buffer = ByteBuffer.allocate(1024);
                client.read(buffer, buffer, new CompletionHandler<Integer, ByteBuffer>() {
                    @Override
                    public void completed(Integer result, ByteBuffer attachment) {
                        attachment.flip();
                        client.write(ByteBuffer.wrap("HelloClient".getBytes()));//業(yè)務邏輯
                    }
                    @Override
                    public void failed(Throwable exc, ByteBuffer attachment) {
                        System.out.println(exc.getMessage());//失敗處理
                    }
                });
            }

            @Override
            public void failed(Throwable exc, Object attachment) {
                exc.printStackTrace();//失敗處理
            }
        });
        while (true) {
            //不while true main方法一瞬間結束
        }
    }

當然上面的缺點相比較它優(yōu)點都可以忽略。JDK提供了異步方式實現(xiàn)，但在實際的Linux環(huán)境中底層還是epoll，只不過多了一層循環(huán)，不算真正的異步非阻塞。而且就像上圖中代碼調(diào)用，處理網(wǎng)絡連接的代碼和業(yè)務代碼解耦得不夠好。Netty提供了簡潔、解耦、結構清晰的API。

 public static void main(String[] args) {
        new NettyServer().serverStart();
        System.out.println("Netty server started !");
    }

    public void serverStart() {
        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        ServerBootstrap b = new ServerBootstrap();
        b.group(bossGroup, workerGroup)
                .channel(NioServerSocketChannel.class)
                .childHandler(new ChannelInitializer<SocketChannel>() {
                    @Override
                    protected void initChannel(SocketChannel ch) throws Exception {
                        ch.pipeline().addLast(new Handler());
                    }
                });
        try {
            ChannelFuture f = b.localAddress(Constant.HOST, Constant.PORT).bind().sync();
            f.channel().closeFuture().sync();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            workerGroup.shutdownGracefully();
            bossGroup.shutdownGracefully();
        }
    }
}

class Handler extends ChannelInboundHandlerAdapter {
    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        ByteBuf buf = (ByteBuf) msg;
        ctx.writeAndFlush(msg);
        ctx.close();
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        cause.printStackTrace();
        ctx.close();
    }
}

bossGroup 處理網(wǎng)絡請求的大管家（們），網(wǎng)絡連接就緒時，交給workGroup干活的工人（們）。

三、總結

回顧

• 同步/異步，連接建立后，用戶程序讀寫時，如果最終還是需要用戶程序來調(diào)用系統(tǒng)read()來讀數(shù)據(jù)，那就是同步的，反之是異步。Windows實現(xiàn)了真正的異步，內(nèi)核代碼甚為復雜，但對用戶程序來說是透明的。
• 阻塞/非阻塞，連接建立后，用戶程序在等待可讀可寫時，是不是可以干別的事兒。如果可以就是非阻塞，反之阻塞。大多數(shù)操作系統(tǒng)都支持的。

redis,Nginx,Netty,Node.js 為什么這么香？

這些技術都是伴隨Linux內(nèi)核迭代中提供了高效處理網(wǎng)絡請求的系統(tǒng)調(diào)用而出現(xiàn)的。了解計算機底層的知識才能更深刻地理解I/O，知其然，更要知其所以然。與君共勉！

文章來源：宜信技術學院 & 宜信支付結算團隊技術分享第8期-宜信支付結算部支付研發(fā)團隊高級工程師周勝帥《從操作系統(tǒng)層面理解Linux的網(wǎng)絡IO模型》

分享者：宜信支付結算部支付研發(fā)團隊高級工程師周勝帥

原文首發(fā)于支付結算團隊技術公號“野指針”