Netty源码分析系列--14. TCP拆包、粘包以及自定义通信协议

使用TCP在Netty中的问题

下面的例子传输ByteBuf数据类型，客户端循环10次发送hello world到服务端，服务端接收数据并打印：

public class ServerHandler extends SimpleChannelInboundHandler {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, ByteBuf msg) throws Exception {
        System.out.printf("调用服务handler");
    
        byte[] data = new byte[msg.readableBytes()];
    
        msg.readBytes(data);
    
        String text = new String(data, Charset.forName("utf-8"));
        System.out.println(text);
    }
}

public class ClientHandler extends SimpleChannelInboundHandler {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, ByteBuf msg) throws Exception {
        //NOOP
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        for (int i = 0; i < 10; i++) {
            ctx.writeAndFlush(Unpooled.copiedBuffer("hello world", Charset.forName("utf-8")));
        }
    }
}

运行以上代码，发现控制台的输出不一定打印10次数据，原因就在于TCP协议在传输时的粘包处理。

使用自定义协议

• 定义协议MyProtocol如下：协议包含头部head和数据本身body

public class MyProtocol {
    private int head;
    private byte[] body;

    public int getHead() {
        return head;
    }

    public void setHead(int head) {
        this.head = head;
    }

    public byte[] getBody() {
        return body;
    }

    public void setBody(byte[] body) {
        this.body = body;
    }
}

• 有了自定义协议后，就要有对应的编解码器。解码器MyProtocolDecoder继承前文中的ReplayingDeoder；而编码器MyProtocolEncoder继承Netty提供的MessageToByteEncoder。

/**
 * 将ByteBuf转换为MyProtocol，不需要状态管理，泛型为Void
 */
public class MyProtocolDecoder extends ReplayingDecoder {
    @Override
    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List