Java NIO 核心原理與 Selector 多路復用實戰詳解

前言

Java NIO (New I/O) 引入了通道與緩衝區模型，配合 Selector 多路復用機制，實現了單線程高效管理成千上萬連接的能力。本文深入剖析 Selector 工作流程、關鍵 API，並提供一個完整的非阻塞 TCP 服務端示例，幫助開發者快速掌握 NIO 核心編程範式。

核心概念

• Channel (通道)：雙向數據傳輸管道，如 ServerSocketChannel、SocketChannel，可配置為非阻塞模式。
• Buffer (緩衝區)：數據讀寫的容器，核心屬性：capacity、position、limit。
• Selector (選擇器)：多路復用核心，單線程輪詢多個 Channel 的就緒事件 (連接、讀、寫)。
• SelectionKey (選鍵)：Channel 向 Selector 註冊的憑證，包含感興趣的操作集 (interest set) 與就緒操作集 (ready set)。

Selector 工作流程

1. 創建 Selector.open()。
2. 創建 ServerSocketChannel，綁定端口，設置非阻塞 configureBlocking(false)。
3. 向 Selector 註冊 Channel 及關注事件：channel.register(selector, SelectionKey.OP_ACCEPT)。
4. 循環調用 selector.select() 阻塞等待事件就緒。
5. 獲取就緒鍵集 selector.selectedKeys()，遍歷處理：

while (true) {
    int ready = selector.select(); // 阻塞直到至少一個通道就緒
    if (ready == 0) continue;

    Set<SelectionKey> keys = selector.selectedKeys();
    Iterator<SelectionKey> it = keys.iterator();
    while (it.hasNext()) {
        SelectionKey key = it.next();
        it.remove(); // 必須移除，避免重複處理

        if (key.isAcceptable()) {
            // 接受新連接
            ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
            SocketChannel sc = ssc.accept();
            sc.configureBlocking(false);
            // 註冊讀/寫事件
            sc.register(selector, SelectionKey.OP_READ | SelectionKey.OP_WRITE);
        } else if (key.isReadable()) {
            // 讀取數據
            SocketChannel sc = (SocketChannel) key.channel();
            ByteBuffer buf = ByteBuffer.allocate(1024);
            int len = sc.read(buf);
            if (len > 0) {
                buf.flip();
                // 處理業務邏輯...
            } else if (len == -1) {
                key.cancel();
                sc.close();
            }
        } else if (key.isWritable()) {
            // 寫入數據邏輯
        }
    }
}

完整服務端示例

public class NioServer {
    public static void main(String[] args) throws IOException {
        Selector selector = Selector.open();
        ServerSocketChannel ssc = ServerSocketChannel.open();
        ssc.bind(new InetSocketAddress(8080));
        ssc.configureBlocking(false);
        ssc.register(selector, SelectionKey.OP_ACCEPT);

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

        while (true) {
            selector.select();
            Set<SelectionKey> keys = selector.selectedKeys();
            Iterator<SelectionKey> it = keys.iterator();
            while (it.hasNext()) {
                SelectionKey key = it.next();
                it.remove();

                if (key.isAcceptable()) {
                    accept(key);
                } else if (key.isReadable()) {
                    read(key);
                }
            }
        }
    }

    private static void accept(SelectionKey key) throws IOException {
        ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
        SocketChannel sc = ssc.accept();
        sc.configureBlocking(false);
        sc.register(key.selector(), SelectionKey.OP_READ, ByteBuffer.allocate(1024));
        System.out.println("Client connected: " + sc.getRemoteAddress());
    }

    private static void read(SelectionKey key) throws IOException {
        SocketChannel sc = (SocketChannel) key.channel();
        ByteBuffer buf = (ByteBuffer) key.attachment();
        buf.clear();
        int len = sc.read(buf);
        if (len > 0) {
            buf.flip();
            String msg = StandardCharsets.UTF_8.decode(buf).toString();
            System.out.println("Received: " + msg.trim());
            // 回傳確認
            sc.write(StandardCharsets.UTF_8.encode("ACK: " + msg));
        } else if (len == -1) {
            key.cancel();
            sc.close();
            System.out.println("Client disconnected");
        }
    }
}

關鍵注意點

1. 非阻塞模式必須：Channel 必須配置 configureBlocking(false) 才能註冊到 Selector。
2. selectedKeys() 必須手動移除：遍歷後調用 it.remove()，否則下次 select 仍會返回已處理的鍵，導致死循環或重複處理。
3. 附件傳遞緩衝區：註冊時附加 ByteBuffer 作為 attachment，避免頻繁分配內存。
4. 半包/粘包處理：NIO 為流式傳輸，需自行實現應用層協議 (如定長、分隔符、長度前綴) 進行消息邊界解析。
5. 寫事件註冊策略：OP_WRITE 常時就緒，建議僅在發送緩衝區有數據待發送時註冊，發送完畢後立即取消註冊，避免 CPU 空轉。

NIO vs BIO 對比

總結

Java NIO 的 Selector 機制是高性能網絡編程的基石。理解「註冊 -> 輪詢 -> 處理」三步曲，熟練掌握 SelectionKey 狀態流轉與 ByteBuffer 讀寫技巧，即可構建出穩定高效的網絡服務端。實際項目中，推薦使用 Netty 等成熟框架屏蔽底層細節，專注業務邏輯開發。