netty源码解析之netty解码器

1 ByteToMessageDecoder.java类的基本结构如下:
 
 
其中ByteToMessageDecoder类是累加器的基础类,其核心方法如下:
// ByteToMessageDecoder.java
@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    if (msg instanceof ByteBuf) {
        // 首先会从RECYCLE的stack中取一个对象 return RECYCLER.get();
        CodecOutputList out = CodecOutputList.newInstance();
        try {
            ByteBuf data = (ByteBuf) msg;
            first = cumulation == null;
            // 判断是否第一次读,第一次读直接赋值
            if (first) {
                cumulation = data;
            } else {
                // A点: 非第一次进行字节流的累加
                cumulation = cumulator.cumulate(ctx.alloc(), cumulation, data);
            }
            // B点:对累加的字节流进行解析,并解析到out中(B点会提供一个decode(ctx, in, out);给子类进行实现 --模板方法模式)
            callDecode(ctx, cumulation, out);
        } catch (DecoderException e) {
            throw e;
        } catch (Throwable t) {
            throw new DecoderException(t);
        } finally {
            if (cumulation != null && !cumulation.isReadable()) {
                numReads = 0;
                cumulation.release();
                cumulation = null;
           } else if (++ numReads >= discardAfterReads) {
                // We did enough reads already try to discard some bytes so we not risk to see a OOME.
                // See https://github.com/netty/netty/issues/4275
                numReads = 0;
                discardSomeReadBytes();
            }
 
            int size = out.size();
            decodeWasNull = !out.insertSinceRecycled();
            // C点:将字节流byteBuf向下传播
            fireChannelRead(ctx, out, size);
            out.recycle();
        }
    } else {
        ctx.fireChannelRead(msg);
    }
}
 
首先分析上面A点:非第一次进行字节流的累加 ,跟踪源码如下:
  • 分析当前对应的对象变量如下:private Cumulator cumulator = MERGE_CUMULATOR;
  • 继续跟踪源码如下:
// ByteToMessageDecoder.java
public static final Cumulator MERGE_CUMULATOR = new Cumulator() {
    @Override
    public ByteBuf cumulate(ByteBufAllocator alloc, ByteBuf cumulation, ByteBuf in) {
        ByteBuf buffer;
        // 判断当前是否还有空间可写
        if (cumulation.writerIndex() > cumulation.maxCapacity() - in.readableBytes() || cumulation.refCnt() > 1) {
            // 如果有就进行拓展
            buffer = expandCumulation(alloc, cumulation, in.readableBytes());
        } else {
            buffer = cumulation;
        }
        // 将新来的字节流加入到buf中,然后释放
        buffer.writeBytes(in);
        in.release();
        return buffer;
    }
};
拓展的方法如下:
// 总的来说就是拿到一个新的加上旧的空间,然后将旧的写到新申请的buf中,然后释放旧的
static ByteBuf expandCumulation(ByteBufAllocator alloc, ByteBuf cumulation, int readable) {
    ByteBuf oldCumulation = cumulation;
    cumulation = alloc.buffer(oldCumulation.readableBytes() + readable);
    cumulation.writeBytes(oldCumulation);
    oldCumulation.release();
    return cumulation;
}
 
 
 
 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章