HttpFragmentedRequestDecoderBenchmark Class — netty Architecture

Source Code

microbench/src/main/java/io/netty/microbench/http/HttpFragmentedRequestDecoderBenchmark.java lines 51–126

@State(Scope.Benchmark)
@Warmup(iterations = 10, time = 500, timeUnit = TimeUnit.MILLISECONDS)
@Measurement(iterations = 10, time = 500, timeUnit = TimeUnit.MILLISECONDS)
public class HttpFragmentedRequestDecoderBenchmark extends AbstractMicrobenchmark {
    @Param({ "64", "128" })
    public int headerFragmentBytes;

    @Param({ "false", "true" })
    public boolean direct;

    @Param({ "false", "true" })
    public boolean pooled;

    @Param({ "true", "false"})
    public boolean validateHeaders;

    private EmbeddedChannel channel;

    private ByteBuf[] fragmentedRequest;

    private static ByteBuf[] stepsBuffers(ByteBufAllocator alloc, byte[] content, int fragmentSize, boolean direct) {
        // allocate a single big buffer and just slice it
        final int headerLength = content.length - CONTENT_LENGTH;
        final ArrayList<ByteBuf> bufs = new ArrayList<ByteBuf>();
        for (int a = 0; a < headerLength;) {
            int amount = fragmentSize;
            if (a + amount > headerLength) {
                amount = headerLength - a;
            }
            final ByteBuf buf = direct? alloc.directBuffer(amount, amount) : alloc.heapBuffer(amount, amount);
            buf.writeBytes(content, a, amount);
            bufs.add(buf);
            a += amount;
        }
        // don't split the content
        // Should produce HttpContent
        final ByteBuf buf = direct?
                alloc.directBuffer(CONTENT_LENGTH, CONTENT_LENGTH) :
                alloc.heapBuffer(CONTENT_LENGTH, CONTENT_LENGTH);
        buf.writeBytes(content, content.length - CONTENT_LENGTH, CONTENT_LENGTH);
        bufs.add(buf);
        return bufs.toArray(new ByteBuf[0]);
    }

    @Setup
    public void initPipeline() {
        final ByteBufAllocator allocator = pooled? PooledByteBufAllocator.DEFAULT : UnpooledByteBufAllocator.DEFAULT;
        fragmentedRequest = stepsBuffers(allocator, CONTENT_MIXED_DELIMITERS, headerFragmentBytes, direct);
        channel = new EmbeddedChannel(
                new HttpRequestDecoder(DEFAULT_MAX_INITIAL_LINE_LENGTH, DEFAULT_MAX_HEADER_SIZE, DEFAULT_MAX_CHUNK_SIZE,
                                       validateHeaders, DEFAULT_INITIAL_BUFFER_SIZE));
    }

    @TearDown
    public void releaseStepBuffers() {
        for (ByteBuf buf : fragmentedRequest) {
            buf.release();
        }
    }

    @Benchmark
    @CompilerControl(Mode.DONT_INLINE)
    public void testDecodeWholeRequestInMultipleStepsMixedDelimiters() {
        final EmbeddedChannel channel = this.channel;
        for (ByteBuf buf : this.fragmentedRequest) {
            buf.resetReaderIndex();
            buf.retain();
            channel.writeInbound(buf);
            final Queue<Object> decoded = channel.inboundMessages();
            Object o;
            while ((o = decoded.poll()) != null) {
                ReferenceCountUtil.release(o);
            }
        }
    }
}