From 2e36ac45944299676f0df994443b2c973b008b47 Mon Sep 17 00:00:00 2001
From: Norman Maurer <norman_maurer@apple.com>
Date: Mon, 26 Oct 2015 11:09:47 +0100
Subject: [PATCH] Add benchmark for HeapByteBuf implementations.

Motivation:

To prove one implementation is faster as the other we should have a benchmark.

Modifications:

Add benchmark which benchmarks the unsafe and non-unsafe implementation of HeapByteBuf.

Result:

Able to compare speed of implementations easily.
---
 .../buffer/HeapByteBufBenchmark.java          | 103 ++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 microbench/src/test/java/io/netty/microbench/buffer/HeapByteBufBenchmark.java

diff --git a/microbench/src/test/java/io/netty/microbench/buffer/HeapByteBufBenchmark.java b/microbench/src/test/java/io/netty/microbench/buffer/HeapByteBufBenchmark.java
new file mode 100644
index 0000000000..baf2d0d01d
--- /dev/null
+++ b/microbench/src/test/java/io/netty/microbench/buffer/HeapByteBufBenchmark.java
@@ -0,0 +1,103 @@
+/*
+ * Copyright 2015 The Netty Project
+ *
+ * The Netty Project licenses this file to you under the Apache License,
+ * version 2.0 (the "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at:
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+package io.netty.microbench.buffer;
+
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.ByteBufAllocator;
+import io.netty.microbench.util.AbstractMicrobenchmark;
+import org.openjdk.jmh.annotations.Benchmark;
+import org.openjdk.jmh.annotations.Setup;
+import org.openjdk.jmh.annotations.TearDown;
+
+import java.lang.reflect.Constructor;
+
+public class HeapByteBufBenchmark extends AbstractMicrobenchmark {
+
+    private ByteBuf unsafeBuffer;
+    private ByteBuf buffer;
+
+    private static ByteBuf newBuffer(String classname) throws Exception {
+        // Using reflection to workound package-private implementations.
+        Class<?> clazz = Class.forName(classname);
+        Constructor<?> constructor = clazz.getDeclaredConstructor(ByteBufAllocator.class, int.class, int.class);
+        constructor.setAccessible(true);
+        return (ByteBuf) constructor.newInstance(ByteBufAllocator.DEFAULT, 8, Integer.MAX_VALUE);
+    }
+
+    @Setup
+    public void setup() throws Exception {
+        unsafeBuffer = newBuffer("io.netty.buffer.UnpooledUnsafeHeapByteBuf");
+        buffer = newBuffer("io.netty.buffer.UnpooledHeapByteBuf");
+        unsafeBuffer.writeLong(1L);
+        buffer.writeLong(1L);
+    }
+
+    @TearDown
+    public void destroy() {
+        unsafeBuffer.release();
+        buffer.release();
+    }
+
+    @Benchmark
+    public byte getByteUnsafe() {
+        return unsafeBuffer.getByte(0);
+    }
+
+    @Benchmark
+    public short getByte() {
+        return buffer.getByte(0);
+    }
+
+    @Benchmark
+    public short getShortUnsafe() {
+        return unsafeBuffer.getShort(0);
+    }
+
+    @Benchmark
+    public short getShort() {
+        return buffer.getShort(0);
+    }
+
+    @Benchmark
+    public int getMediumUnsafe() {
+        return unsafeBuffer.getMedium(0);
+    }
+
+    @Benchmark
+    public int getMedium() {
+        return buffer.getMedium(0);
+    }
+
+    @Benchmark
+    public int getIntUnsafe() {
+        return unsafeBuffer.getInt(0);
+    }
+
+    @Benchmark
+    public int getInt() {
+        return buffer.getInt(0);
+    }
+
+    @Benchmark
+    public long getLongUnsafe() {
+        return unsafeBuffer.getLong(0);
+    }
+
+    @Benchmark
+    public long getLong() {
+        return buffer.getLong(0);
+    }
+}