【源码】ByteToMessageDecoder对比自定义实现

【源码】ByteToMessageDecoder对比自定义实现

前言

在上一篇随笔中，我们探讨了如何实现一套自定义通信协议，其中涉及到的粘包和拆包处理最初是完全自定义实现的，后来则改为了继承 ByteToMessageDecoder 来简化处理。

本篇将重点讨论这两种实现方式在缓存管理上的主要区别，并深入分析其中的不同之处以及值得借鉴的经验和技巧。

代码回顾

1）完全自定义实现

无缓存的情况

• 反复从ByteBuf中提取完整的消息
• 剩余的残缺消息写入缓存（会进行数据拷贝）

有缓存的情况

• 将新收到的数据接入缓存
• 反复从缓存中提取完整消息
• 释放缓存内读取过的数据（会进行数据移动，导致拷贝）

public class EchoServerHandler extends ChannelInboundHandlerAdapter { private static final int HEADER_LENGTH = 4; //消息头部长度 private ByteBuf buffer = Unpooled.buffer(1024); //缓存残缺消息 @Override public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { ByteBuf income = (ByteBuf) msg;

    //上一次有缓存存在，则本数据包不是消息头开头，
    if(buffer.readableBytes() > 0) {
        //进行必要的扩容，下面的readBytes不会自动扩容
        buffer.ensureWritable(income.readableBytes()); 
        income.readBytes(buffer, income.readableBytes());

        readMsgFromBuffer(buffer);

        //剩下一点残缺消息
        if(buffer.readableBytes() > 0) {
            //保留剩下的数据，重置读索引为0
            System.out.println("缓存剩余字节："+buffer.readableBytes());
            buffer.discardReadBytes();
        } else { //刚刚好，则清空数据
            buffer.clear();
        }
    } else {
        readMsgFromBuffer(income);

        //剩下的数据全部写入缓存
        if (income.readableBytes() >0) {
            System.out.println("剩余字节:"+income.readableBytes());
            income.readBytes(buffer, income.readableBytes());
        }
    }

}

//从字节数组中读取完整的消息
private void readMsgFromBuffer(ByteBuf byteBuf) {
    //剩余可读消息是否包含一个消息头
    while(byteBuf.readableBytes() >= HEADER\_LENGTH) {
        byteBuf.markReaderIndex(); //由于可能读不到完整的消息，所以读之前先标记索引位置，方便重置
        //读取消息头
        byte\[\] headerBytes = new byte\[4\];
        byteBuf.readBytes(headerBytes);
        //获取类型
        int type = headerBytes\[0\] & 0xFF;
        //获取消息体长度
        int bodyLength = ((headerBytes\[1\] & 0xFF) << 16) |
                ((headerBytes\[2\] & 0xFF) << 8) |
                (headerBytes\[3\] & 0xFF);

        //不包含请求体
        if (byteBuf.readableBytes() < bodyLength) {
            byteBuf.resetReaderIndex(); //重置读索引到当前消息头位置
            break;
        }

        // 完整消息体已经接收，处理消息
        byte\[\] body = new byte\[bodyLength\];
        byteBuf.readBytes(body);
        //System.out.println("type:"+type+"||length:"+bodyLength+"||body:"+new String(body, CharsetUtil.UTF\_8));
        if(type == 1) {
            try {
                HelloRequest request \= HelloRequest.parseFrom(body);
                System.out.println("收到消息:"+request.toString());
            } catch (Exception e) {
                System.out.println("解析失败："+new String(body, CharsetUtil.UTF\_8));
            }
        } else {
            System.out.println("消息类型未知："+type);
        }

    }
}

....

}

2）继承ByteToMessageDecoder的实现

使用ByteToMessageDecoder后，数据的解码变得更加简化。只需检查缓冲区是否有足够的数据来提取一个/多个完整的消息。

如果数据不足，解码过程就会结束，无需额外管理缓存。

public class MessageDecoder extends ByteToMessageDecoder { private static final int HEADER_LENGTH = 4; //消息头部长度 @Override protected void decode(ChannelHandlerContext ctx, ByteBuf in, List

转载声明
本文内容出自网络，非原创作品。由于无法确认原始来源和作者信息，在此对原作者表示感谢。
如涉及版权问题，请联系 [联系邮箱]，我们将及时处理。