SimpleLeakAwareByteBuf Class — netty Architecture

Source Code

buffer/src/main/java/io/netty/buffer/SimpleLeakAwareByteBuf.java lines 27–227

class SimpleLeakAwareByteBuf extends WrappedByteBuf {

    /**
     * This object's is associated with the {@link ResourceLeakTracker}. When {@link ResourceLeakTracker#close(Object)}
     * is called this object will be used as the argument. It is also assumed that this object is used when
     * {@link ResourceLeakDetector#track(Object)} is called to create {@link #leak}.
     */
    private final ByteBuf trackedByteBuf;
    final ResourceLeakTracker<ByteBuf> leak;

    SimpleLeakAwareByteBuf(ByteBuf wrapped, ByteBuf trackedByteBuf, ResourceLeakTracker<ByteBuf> leak) {
        super(wrapped);
        this.trackedByteBuf = ObjectUtil.checkNotNull(trackedByteBuf, "trackedByteBuf");
        this.leak = ObjectUtil.checkNotNull(leak, "leak");
    }

    SimpleLeakAwareByteBuf(ByteBuf wrapped, ResourceLeakTracker<ByteBuf> leak) {
        this(wrapped, wrapped, leak);
    }

    @Override
    public ByteBuf slice() {
        return newSharedLeakAwareByteBuf(super.slice());
    }

    @Override
    public ByteBuf retainedSlice() {
        try {
            return unwrappedDerived(super.retainedSlice());
        } catch (IllegalReferenceCountException irce) {
            ThrowableUtil.addSuppressed(irce, leak.getCloseStackTraceIfAny());
            throw irce;
        }
    }

    @Override
    public ByteBuf retainedSlice(int index, int length) {
        try {
            return unwrappedDerived(super.retainedSlice(index, length));
        } catch (IllegalReferenceCountException irce) {
            ThrowableUtil.addSuppressed(irce, leak.getCloseStackTraceIfAny());
            throw irce;
        }
    }

    @Override
    public ByteBuf retainedDuplicate() {
        try {
            return unwrappedDerived(super.retainedDuplicate());
        } catch (IllegalReferenceCountException irce) {
            ThrowableUtil.addSuppressed(irce, leak.getCloseStackTraceIfAny());
            throw irce;
        }
    }

    @Override
    public ByteBuf readRetainedSlice(int length) {
        try {
            return unwrappedDerived(super.readRetainedSlice(length));
        } catch (IllegalReferenceCountException irce) {
            ThrowableUtil.addSuppressed(irce, leak.getCloseStackTraceIfAny());
            throw irce;
        }
    }

    @Override
    public ByteBuf slice(int index, int length) {
        return newSharedLeakAwareByteBuf(super.slice(index, length));
    }

    @Override
    public ByteBuf duplicate() {
        return newSharedLeakAwareByteBuf(super.duplicate());
    }

    @Override
    public ByteBuf readSlice(int length) {
        return newSharedLeakAwareByteBuf(super.readSlice(length));
    }

    @Override