Netty（三）TCP粘包拆包處理

tcp是一個“流”的協議，一個完整的包可能會被TCP拆分成多個包進行發送，也可能把小的封裝成一個大的數據包發送，這就是所謂的TCP粘包和拆包問題。

粘包、拆包問題說明

假設客戶端分別發送數據包D1和D2給服務端，由於服務端一次性讀取到的字節數是不確定的，所以可能存在以下4種情況。

• 1.服務端分2次讀取到了兩個獨立的包，分別是D1,D2,沒有粘包和拆包；
• 2.服務端一次性接收了兩個包，D1和D2粘在一起了，被成為TCP粘包;
• 3.服務端分2次讀取到了兩個數據包，第一次讀取到了完整的D1和D2包的部分內容,第二次讀取到了D2包的剩余內容，這被稱為拆包；
• 4.服務端分2次讀取到了兩個數據包，第一次讀取到了部分D1，第二次讀取D1剩余的部分和完整的D2包；

如果此時服務端TCP接收滑動窗非常小,而數據包D1和D2都很大，很有可能發送第五種可能，即服務端多次才能把D1和D2接收完全，期間多次發生拆包情況。（TCP接收滑動窗：是接收端的大小，隨着流量大小而變化，如果我的解釋還不明確，請讀者自行百度，或者查閱《計算機網絡》、《TCP/IP》中TCP的內容）

粘包問題的解決策略

由於底層的TCP無法理解上層的業務邏輯，所以在底層是無法確保數據包不被拆分和重組的，這個問題只能通過上層的應用協議棧設計來解決，根據業界的主流協議的解決方案，歸納如下：

• 1.消息定長，例如每個報文的大小為固定長度200字節,如果不夠，空位補空格；
• 2.在包尾增加回車換行符進行分割，例如FTP協議；
• 3.將消息分為消息頭和消息體，消息頭中包含表示消息總長度（或者消息體長度）的字段，通常設計思路是消息頭的第一個字段用int來表示消息的總長度；（我之前linux C開發，就用的這種）。
• 4.更復雜的應用層協議；

 

為了解決TCP粘包拆包的問題，Netty默認提供了多種編碼器來處理，以下通過代碼來說明；

服務端：

public class TimeServer {
/* --------      和Netty入門章節一樣的代碼     -------- */
    private class ChildChannelHandler extends ChannelInitializer<SocketChannel>{
        @Override
        protected  void initChannel(SocketChannel arg0)throws Exception{
            // 增加 LineBasedFrameDecoder 和StringDecoder編碼器
            arg0.pipeline().addLast(new LineBasedFrameDecoder(1024));
            arg0.pipeline().addLast(new StringDecoder());
            arg0.pipeline().addLast(new TimeServerHandler());
        }
    }

/* --------      和Netty入門章節一樣的代碼     -------- */
}

public class TimeServerHandler extends ChannelHandlerAdapter{

    private int counter;

    // 用於網絡的讀寫操作
    @Override
    public void channelRead(ChannelHandlerContext ctx,Object msg)
            throws Exception{

        String body = (String)msg;
        System.out.println("the time server order : " + body+";the counter is :"+ (++counter));

        String currentTime = "QUERY TIME ORDER".equalsIgnoreCase(body)?new Date(
                System.currentTimeMillis()).toString():"BAD ORDER";
        currentTime +=System.getProperty("line.separator");   // System.getProperty("line.separator")，獲取/n的作用
        ByteBuf resp = Unpooled.copiedBuffer(currentTime.getBytes());
        ctx.writeAndFlush(resp);

        // 當客戶端和服務端建立tcp成功之后，Netty的NIO線程會調用channelActive
        // 發送查詢時間的指令給服務端。
        // 調用ChannelHandlerContext的writeAndFlush方法，將請求消息發送給服務端
        // 當服務端應答時，channelRead方法被調用
    }

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

客服端

public class TimeClient {
    public void connect(String host,int port)throws Exception{
        // 配置服務端的NIO線程組
        EventLoopGroup group = new NioEventLoopGroup();

        try {
            // Bootstrap 類，是啟動NIO服務器的輔助啟動類
            Bootstrap b = new Bootstrap();
            b.group(group).channel(NioSocketChannel.class)
                    .option(ChannelOption.TCP_NODELAY,true)
                    .handler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        public void initChannel(SocketChannel ch)
                                throws Exception{
                            // 增加 LineBasedFrameDecoder 和StringDecoder編碼器
                            ch.pipeline().addLast(new LineBasedFrameDecoder(1024));
                            ch.pipeline().addLast(new StringDecoder());
                            ch.pipeline().addLast(new TimeClientHandler());
                        }
                    });

            // 發起異步連接操作
            ChannelFuture f= b.connect(host,port).sync();

            // 等待客服端鏈路關閉
            f.channel().closeFuture().sync();
        }finally {
            group.shutdownGracefully();
        }
    }

    public static void main(String[]args)throws Exception{
        int port = 8080;
        if(args!=null && args.length>0){
            try {
                port = Integer.valueOf(args[0]);
            }
            catch (NumberFormatException ex){}
        }
        new TimeClient().connect("127.0.0.1",port);
    }
}

以上的代碼，主要就是增加 LineBasedFrameDecoder 和StringDecoder編碼器來處理粘包、拆包問題。全部項目代碼，源碼在src/main/java/NettyStickyPacket下，分為客戶端和服務端，他們的代碼基本和Netty入門章節的代碼類似，只是增加了解碼器。

 

LineBasedFrameDecoder 和StringDecoder編碼器說明

LineBasedFrameDecoder 工作原理：依次編譯bytebuf中的可讀字符，判斷看是否有“\n”或者“\r\n”，如果有，就以此位置為結束位置，從可讀索引到結束位置區間的字節就組成了一行。它是以換行符為結束標志的解碼器，支持攜帶結束符或者不攜帶結束符兩種解碼方式，同時支持單行的最大長度。如果連續讀取到最大長度后，仍然沒有發現換行符，就會拋出異常，同時忽略掉之前讀到的異常碼流。

StringDecoder功能：將接收到的對象轉成字符串，然后繼續調用后面的handler。 LineBasedFrameDecoder + StringDecoder組成就是按行切換的文本解碼器，它被設計用來支持TCP的粘包和拆包。

可能讀者會提出新的疑問：如果發送的消息不是以換行符結束怎么辦？或者沒有回車換行符，靠消息頭中的長度字段來分包怎么辦？是不是需要自己寫半包解碼器？答案是否定的，Netty提供了多種支持TCP粘包/拆包的解碼器。

關於“分隔符解碼器”，下一章單獨講解說明。

 

源碼下載

GitHub地址：https://github.com/orange1438/Netty_Course

 

作者：orange1438
出處：http://www.cnblogs.com/orange1438/
本文版權歸作者和博客園共有，歡迎轉載，但未經作者同意必須保留此段聲明，且在文章頁面明顯位置給出原文連接，否則保留追究法律責任的權利。