Java NIO：非阻塞 IO 的核心原理与案例

一、核心原理

传统阻塞式 IO 中，线程在读写数据时会挂起等待，导致资源浪费。Java NIO（New I/O）采用非阻塞模式，核心是通过事件驱动机制实现高效 I/O 处理：

1. 通道（Channel）

   • 双向数据传输管道（如 SocketChannel、FileChannel）
   • 支持异步读写，线程无需阻塞等待

2. 缓冲区（Buffer）

   • 数据中转容器（如 ByteBuffer）
   • 读写操作通过 flip()、clear() 切换模式

3. 选择器（Selector）

   • 核心事件调度器
   • 通过 select() 监听多个通道的 I/O 事件（如连接、读、写）
   • 事件触发时返回 SelectionKey 集合

非阻塞流程：

graph LR
A[线程] --> B[Selector.select()]
B --> C{事件就绪?}
C -- 是 --> D[处理SelectionKey]
C -- 否 --> B
D --> E[读写Buffer]
E --> B

单线程可处理数千连接，避免线程阻塞和上下文切换开销。

---

二、关键代码案例：NIO 回声服务器

import java.nio.*;
import java.nio.channels.*;
import java.net.*;
import java.util.*;

public class NioEchoServer {
    public static void main(String[] args) throws Exception {
        Selector selector = Selector.open();
        ServerSocketChannel serverChannel = ServerSocketChannel.open();
        serverChannel.bind(new InetSocketAddress(8080));
        serverChannel.configureBlocking(false);  // 非阻塞模式
        serverChannel.register(selector, SelectionKey.OP_ACCEPT);  // 注册连接事件

        System.out.println("Server started on port 8080");

        while (true) {
            selector.select();  // 阻塞直到事件就绪
            Set<SelectionKey> keys = selector.selectedKeys();
            Iterator<SelectionKey> iter = keys.iterator();

            while (iter.hasNext()) {
                SelectionKey key = iter.next();
                iter.remove();

                if (key.isAcceptable()) {  // 处理新连接
                    ServerSocketChannel server = (ServerSocketChannel) key.channel();
                    SocketChannel client = server.accept();
                    client.configureBlocking(false);
                    client.register(selector, SelectionKey.OP_READ);
                    System.out.println("Client connected: " + client.getRemoteAddress());
                }

                if (key.isReadable()) {  // 处理读事件
                    SocketChannel client = (SocketChannel) key.channel();
                    ByteBuffer buffer = ByteBuffer.allocate(256);
                    int bytesRead = client.read(buffer);
                    
                    if (bytesRead == -1) {  // 客户端断开
                        client.close();
                        continue;
                    }

                    buffer.flip();
                    byte[] data = new byte[buffer.remaining()];
                    buffer.get(data);
                    String message = new String(data);
                    System.out.println("Received: " + message);

                    // 回声响应
                    ByteBuffer response = ByteBuffer.wrap(("ECHO: " + message).getBytes());
                    client.write(response);
                }
            }
        }
    }
}

---

三、原理解析

1. 事件驱动模型

   • Selector 通过操作系统级轮询（如 epoll）监听事件
   • 时间复杂度 $O(1)$，与连接数无关

2. 零拷贝优化

   • FileChannel.transferTo() 直接在内核空间传输数据
   • 避免用户态与内核态的数据复制

3. 缓冲区复用

   • 通过 Buffer 池减少内存分配开销
   • 典型读写流程：

     buffer.clear();    // 准备写入数据
     channel.read(buffer);
     buffer.flip();     // 切换为读模式
     channel.write(buffer);
     

---

四、适用场景

• 高并发网络服务（如聊天服务器、API 网关）
• 大文件传输（利用零拷贝）
• 需要管理上万连接的场景（如物联网网关）

注意事项：NIO 编程复杂度较高，需注意资源泄漏（如未关闭 Channel）和边缘触发问题。建议结合 java.nio.channels.CompletionHandler 实现异步回调简化逻辑。