HugeDecompressIncomingHandler Class — netty Architecture

Source Code

codec-compression/src/test/java/io/netty/handler/codec/compression/AbstractIntegrationTest.java lines 236–266

    private static final class HugeDecompressIncomingHandler
            extends ChannelInboundHandlerAdapter implements ByteProcessor {
        final int memoryLimit;
        final MessageDigest digest;
        long total;

        HugeDecompressIncomingHandler(int memoryLimit, MessageDigest digest) {
            this.memoryLimit = memoryLimit;
            this.digest = digest;
        }

        @Override
        public void channelRead(ChannelHandlerContext ctx, Object msg) {
            ByteBuf buf = (ByteBuf) msg;
            total += buf.readableBytes();
            try {
                buf.forEachByte(this);
                PooledByteBufAllocator allocator = (PooledByteBufAllocator) ctx.alloc();
                assertThat(allocator.metric().usedHeapMemory()).isLessThan(memoryLimit);
                assertThat(allocator.metric().usedDirectMemory()).isLessThan(memoryLimit);
            } finally {
                buf.release();
            }
        }

        @Override
        public boolean process(byte value) throws Exception {
            digest.update(value);
            return true;
        }
    }