SnappyDirectBenchmark Class — netty Architecture

Source Code

microbench/src/main/java/io/netty/microbench/snappy/SnappyDirectBenchmark.java lines 41–108

@State(Scope.Benchmark)
@Fork(1)
@Threads(1)
@Warmup(iterations = 5)
@Measurement(iterations = 3)
public class SnappyDirectBenchmark extends AbstractMicrobenchmark {

    @Param({ "true", "false" })
    public boolean reuseHashTable;
    private ByteBuf buffer;
    private Snappy snappy;
    private ByteBuf in;
    private ByteBuf out;

    @Param({ "4096", "2048", "1024", "512", "256", "128" })
    private int bufferSizeInBytes;

    @AuxCounters(value = Type.OPERATIONS)
    @State(Scope.Thread)
    public static class AllocationMetrics {
        private long inputSize;

        private long outputSize;

        public long compressedRatio() {
            return inputSize / outputSize;
        }
    }

    @Setup
    public void setup() throws UnsupportedEncodingException {
        ByteBufAllocator allocator = UnpooledByteBufAllocator.DEFAULT;
        buffer = allocator.buffer(bufferSizeInBytes);

        if (reuseHashTable) {
            snappy = Snappy.withHashTableReuse();
        } else {
            snappy = new Snappy();
        }

        byte[] compressibleByteArray = new byte[buffer.writableBytes()];
        Arrays.fill(compressibleByteArray, (byte) 1);
        buffer.writeBytes(compressibleByteArray);

        in = Unpooled.wrappedBuffer(compressibleByteArray);
        out = Unpooled.directBuffer();
    }

    @Benchmark
    public ByteBuf encode(AllocationMetrics allocationMetrics) {
        int length = in.readableBytes();
        snappy.encode(in, out, length);
        in.resetReaderIndex();
        allocationMetrics.inputSize += length;
        allocationMetrics.outputSize += out.readableBytes();
        out.setIndex(0, 0);

        return out;
    }

    @TearDown(Level.Trial)
    public void teardown() {
        buffer.release();
        buffer = null;
        out.release();
        out = null;
    }
}