Java NIO 開發網路協定框架設計：訊息邊界處理與可重用元件架構

前言

Java NIO (New I/O) 自 J2SE 1.4 引入 Buffer、Channel、Selector 等核心概念，為高效能網路程式設計提供新範式。本文整理開發自訂網路協定的實戰經驗，提出一套可重用的協定開發框架，核心解決訊息邊界問題與Buffer 操作封裝。

協定設計核心：訊息邊界處理

Socket 傳輸的是位元組流，單次 send() 的資料可能分多次 recv() 接收，或一次 recv() 收到多個封包。應用層協定必須自行解決邊界問題，主流兩種方案：

1. 分隔符號法：如 HTTP 以雙 \n 結束訊息
2. 固定長度表頭法：每個封包前置固定長度 Header，內含訊息長度與型別 (本框架採用)

框架核心介面與類別

1. MessageHeader 介面 - 訊息表頭契約

public interface MessageHeader {
    int getMessageType();           // 訊息型別識別碼
    int getMessageLength();         // 訊息總長度 (含 Header)
    void buildFromBuffer(ByteBuffer buffer);  // 從 Buffer 解析 Header
    void appendToBuffer(ByteBuffer buffer);   // 將 Header 序列化至 Buffer
}

2. Message 介面 - 完整訊息封裝

public interface Message {
    void setHeader(MessageHeader header);           // 設定 Header (Channel 接收時呼叫)
    MessageHeader getHeader();                      // 取得 Header
    void buildBodyFromBuffer(ByteBuffer buffer);    // 從 Buffer 解析 Body
    void appendBodyToBuffer(ByteBuffer buffer);     // 將 Body 序列化至 Buffer
}

3. MessageFactory 介面 - 抽象工廠模式

public interface MessageFactory {
    int getMessageHeaderLength();           // 固定 Header 位元組長度
    MessageHeader createMessageHeader();    // 建立 Header 實例
    Message createMessage(int type);        // 依型別建立 Message 實例
}

4. MessageChannel 類別 - 收發核心邏輯 (封裝 Buffer 操作)

class MessageChannel {
    public MessageChannel(int receiveBufferSize, int sendBufferSize, 
                          SocketChannel sc, MessageFactory mf);
    
    public Message receive() throws IOException, MessageFormatException;
    public void send(Message message) throws IOException;
}

receive() 流程：

1. 依 MessageFactory.getMessageHeaderLength() 讀取 Header 位元組
2. 呼叫 createMessageHeader() 建立物件，buildFromBuffer() 解析
3. 依 Header 中長度讀取 Body 位元組
4. 呼叫 createMessage(type) 建立物件，buildBodyFromBuffer() 解析
5. 呼叫 setHeader() 關聯，回傳完整 Message
6. 任何步驟失敗拋出 MessageFormatException

5. BufferUtil 工具類 - 字串序列化支援

class BufferUtil {
    // 固定長度字串
    static String getString(ByteBuffer buffer, int length);
    static void putString(ByteBuffer buffer, String str, int length);
    // C 風格字串 (以 \0 結束，不足補 0)
    static String getCString(ByteBuffer buffer, int maxLength);
    static void putCString(ByteBuffer buffer, String str, int maxLength);
    // 變長字串：長度前綴 (1/2/4 位元組)
    static String getVarStringByteLength(ByteBuffer buffer);
    static String getVarStringShortLength(ByteBuffer buffer);
    static String getVarStringIntLength(ByteBuffer buffer);
    static void putVarStringByteLength(ByteBuffer buffer, String str);
    static void putVarStringShortLength(ByteBuffer buffer, String str);
    static void putVarStringIntLength(ByteBuffer buffer, String str);
}

註：目前未處理字元編碼 (Charset)，實務應擴充支援 UTF-8/Big5 等。

6. MessageFormatException - 協定格式異常

封裝協定解析錯誤：Header 長度不符、未知訊息型別、Body 長度超過緩衝區等。

框架使用範例

// 伺服器端啟動
ServerSocketChannel ssc = ServerSocketChannel.open();
ssc.socket().bind(new InetSocketAddress(8888));
ssc.configureBlocking(false);

Selector selector = Selector.open();
ssc.register(selector, SelectionKey.OP_ACCEPT);

MyMessageFactory factory = new MyMessageFactory();  // 實作 MessageFactory

while (selector.select() > 0) {
    for (SelectionKey key : selector.selectedKeys()) {
        if (key.isAcceptable()) {
            SocketChannel sc = ssc.accept();
            sc.configureBlocking(false);
            MessageChannel mc = new MessageChannel(8192, 8192, sc, factory);
            key.attach(mc);
            key.interestOps(SelectionKey.OP_READ);
        } else if (key.isReadable()) {
            MessageChannel mc = (MessageChannel) key.attachment();
            try {
                Message msg = mc.receive();
                handleMessage(msg);  // 業務邏輯處理
            } catch (MessageFormatException e) {
                // 協定錯誤處理：關閉連線、記錄日誌
            }
        }
    }
    selector.selectedKeys().clear();
}

設計優勢與擴充建議

建議擴充方向：加入 TLS/SSL 支援 (SSLEngine)、訊息壓縮、心跳機制、協定版本協商、序列化框架整合 (Protobuf/MessagePack)。