netty: fix outputstream with new buffer size

Author: jknack

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

@@ -23,25 +23,20 @@ public class NettyOutputStream extends OutputStream {
   private final ChannelHandlerContext context;
   private final ChannelFutureListener closeListener;
   private HttpResponse headers;
-  private AtomicBoolean closed = new AtomicBoolean(false);
+  private final AtomicBoolean closed = new AtomicBoolean(false);
 
   public NettyOutputStream(
       NettyContext ctx, ChannelHandlerContext context, int bufferSize, HttpResponse headers) {
     this.ctx = ctx;
-    this.buffer = context.alloc().buffer(0, bufferSize);
+    this.buffer = context.alloc().heapBuffer(bufferSize, bufferSize);
     this.context = context;
     this.headers = headers;
     this.closeListener = ctx;
   }
 
   @Override
   public void write(int b) {
-    writeHeaders();
-
-    if (buffer.maxWritableBytes() < 1) {
-      flush(null, null);
-    }
-    buffer.writeByte(b);
+    write(new byte[] {(byte) b}, 0, 1);
   }
 
   @Override

modules/jooby-netty/src/test/java/io/jooby/internal/netty/Issue3554.java

@@ -29,7 +29,7 @@ public void shouldCloseOutputStreamOnce() throws IOException {
     when(buffer.readableBytes()).thenReturn(0);
 
     var bufferAllocator = mock(ByteBufAllocator.class);
-    when(bufferAllocator.buffer(0, 1024)).thenReturn(buffer);
+    when(bufferAllocator.heapBuffer(1024, 1024)).thenReturn(buffer);
 
     var future = mock(ChannelFuture.class);
     var channelContext = mock(ChannelHandlerContext.class);