如何安全释放netty缓冲区?

时间:2016-03-22 11:13:39

标签: java buffer netty

我正在使用Netty 4.0并且我已经阅读了描述正常行为的Reference counted objects guide,但是当发生异常时我不知道如何释放ByteBuf

例如:

protected void encode(ChannelHandlerContext ctx, Object msg, List<Object> out) throws Exception {
    ByteBuf buf = null;
    if (msg instanceof HttpMessage) {
        if (state != ST_INIT) {
            throw new IllegalStateException("unexpected message type: " + StringUtil.simpleClassName(msg));
        }

        @SuppressWarnings({ "unchecked", "CastConflictsWithInstanceof" })
        H m = (H) msg;

        buf = ctx.alloc().buffer();
        // Encode the message.
        encodeInitialLine(buf, m);//
        encodeHeaders(m.headers(), buf);
        buf.writeBytes(CRLF);
        state = HttpHeaders.isTransferEncodingChunked(m) ? ST_CONTENT_CHUNK : ST_CONTENT_NON_CHUNK;
    }

    // Bypass the encoder in case of an empty buffer, so that the following idiom works:
    //
    //     ch.write(Unpooled.EMPTY_BUFFER).addListener(ChannelFutureListener.CLOSE);
    //
    // See https://github.com/netty/netty/issues/2983 for more information.

    if (msg instanceof ByteBuf && !((ByteBuf) msg).isReadable()) {
        out.add(EMPTY_BUFFER);
        return;
    }

    if (msg instanceof HttpContent || msg instanceof ByteBuf || msg instanceof FileRegion) {

        if (state == ST_INIT) {
            throw new IllegalStateException("unexpected message type: " + StringUtil.simpleClassName(msg));
        }

        final long contentLength = contentLength(msg);
        if (state == ST_CONTENT_NON_CHUNK) {
            if (contentLength > 0) {
                if (buf != null && buf.writableBytes() >= contentLength && msg instanceof HttpContent) {
                    // merge into other buffer for performance reasons
                    buf.writeBytes(((HttpContent) msg).content());
                    out.add(buf);
                } else {
                    if (buf != null) {
                        out.add(buf);
                    }
                    out.add(encodeAndRetain(msg));
                }
            } else {
                if (buf != null) {
                    out.add(buf);
                } else {
                    // Need to produce some output otherwise an
                    // IllegalStateException will be thrown
                    out.add(EMPTY_BUFFER);
                }
            }

            if (msg instanceof LastHttpContent) {
                state = ST_INIT;
            }
        } else if (state == ST_CONTENT_CHUNK) {
            if (buf != null) {
                out.add(buf);
            }
            encodeChunkedContent(ctx, msg, contentLength, out);
        } else {
            throw new Error();
        }
    } else {
        if (buf != null) {
            out.add(buf);
        }
    }
}

encodeInitialLine(buf, m);引发异常时,如何发布buf

1 个答案:

答案 0 :(得分:1)

你应该将整个方法包装在try-finally块中,在finally块中,你应该检查buf是否为null。如果它不为空,请致电buf.release()

protected void encode(ChannelHandlerContext ctx, Object msg, List<Object> out) throws Exception {
    ByteBuf buf = null;
    try {
        ...
    } finally {
        if (buf != null) {
            buf.release();
        }
    }
}