netty: starts of custom byte buf

Author: jknack

modules/jooby-netty/src/main/java/io/jooby/internal/netty/NettyOutputByteArrayStaticNew.java

@@ -0,0 +1,41 @@
+/*
+ * Jooby https://jooby.io
+ * Apache License Version 2.0 https://jooby.io/LICENSE.txt
+ * Copyright 2014 Edgar Espina
+ */
+package io.jooby.internal.netty;
+
+import edu.umd.cs.findbugs.annotations.NonNull;
+import io.jooby.Context;
+import io.netty.buffer.ByteBuf;
+
+public class NettyOutputByteArrayStaticNew implements NettyByteBufRef {
+  private final ByteBuf buf;
+  private final NettyString contentLength;
+  private final int length;
+
+  protected NettyOutputByteArrayStaticNew(byte[] bytes, int offset, int len) {
+    this.length = len - offset;
+    this.buf = new NettyUnsafeHeapByteBuf(this.length, this.length);
+    this.buf.writeBytes(bytes, offset, length);
+    this.contentLength = new NettyString(Integer.toString(size()));
+  }
+
+  @Override
+  public int size() {
+    return length;
+  }
+
+  @NonNull public ByteBuf byteBuf() {
+    return buf.slice(0, length);
+  }
+
+  @Override
+  public void send(Context ctx) {
+    if (ctx.getClass() == NettyContext.class) {
+      ((NettyContext) ctx).send(buf.slice(), contentLength);
+    } else {
+      ctx.send(asByteBuffer());
+    }
+  }
+}

modules/jooby-netty/src/main/java/io/jooby/internal/netty/NettyUnsafeHeapByteBuf.java

@@ -0,0 +1,58 @@
+/*
+ * Jooby https://jooby.io
+ * Apache License Version 2.0 https://jooby.io/LICENSE.txt
+ * Copyright 2014 Edgar Espina
+ */
+package io.jooby.internal.netty;
+
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.UnpooledByteBufAllocator;
+import io.netty.buffer.UnpooledUnsafeHeapByteBuf;
+
+/** An un-releasable, un-pooled, un-instrumented, un-safe heap {@code ByteBuf}. */
+public final class NettyUnsafeHeapByteBuf extends UnpooledUnsafeHeapByteBuf {
+
+  private static final byte[] EMPTY = new byte[0];
+
+  public NettyUnsafeHeapByteBuf(int initialCapacity, int maxCapacity) {
+    super(UnpooledByteBufAllocator.DEFAULT, initialCapacity, maxCapacity);
+  }
+
+  @Override
+  protected byte[] allocateArray(int initialCapacity) {
+    if (initialCapacity == 0) {
+      return EMPTY;
+    }
+    return super.allocateArray(initialCapacity);
+  }
+
+  @Override
+  public ByteBuf retain(int increment) {
+    return this;
+  }
+
+  @Override
+  public ByteBuf retain() {
+    return this;
+  }
+
+  @Override
+  public ByteBuf touch() {
+    return this;
+  }
+
+  @Override
+  public ByteBuf touch(Object hint) {
+    return this;
+  }
+
+  @Override
+  public boolean release() {
+    return false;
+  }
+
+  @Override
+  public boolean release(int decrement) {
+    return false;
+  }
+}

modules/jooby-netty/src/test/java/io/jooby/netty/ByteBufBench.java

@@ -5,11 +5,13 @@
  */
 package io.jooby.netty;
 
+import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
 import java.util.concurrent.TimeUnit;
 
 import org.openjdk.jmh.annotations.*;
 
+import io.netty.buffer.ByteBuf;
 import io.netty.buffer.Unpooled;
 
 @Fork(5)
@@ -20,10 +22,16 @@
 @State(Scope.Benchmark)
 public class ByteBufBench {
   private String string;
+  private byte[] bytes;
+  private ByteBuf buf;
+  private ByteBuffer buffer;
 
   @Setup
   public void setup() {
     string = "Hello World!";
+    bytes = string.getBytes(StandardCharsets.UTF_8);
+    buf = Unpooled.wrappedBuffer(bytes);
+    buffer = ByteBuffer.wrap(bytes);
   }
 
   @Benchmark
@@ -41,8 +49,28 @@ public void stringUtf8Bytes() {
     Unpooled.wrappedBuffer(string.getBytes(StandardCharsets.UTF_8));
   }
 
+  @Benchmark
+  public void byteBufWrapper() {
+    Unpooled.wrappedBuffer(bytes);
+  }
+
+  @Benchmark
+  public void byteBufferWrapper() {
+    ByteBuffer.wrap(bytes);
+  }
+
   @Benchmark
   public void stringUSAsciiBytes() {
     Unpooled.wrappedBuffer(string.getBytes(StandardCharsets.US_ASCII));
   }
+
+  @Benchmark
+  public void bufferSlice() {
+    buffer.slice();
+  }
+
+  @Benchmark
+  public void byteBufSlice() {
+    buf.slice();
+  }
 }