Azure function improvement

Author: loicmathieu

extensions/azure-functions-http/runtime/src/main/java/io/quarkus/azure/functions/resteasy/runtime/BaseFunction.java

@@ -1,6 +1,7 @@
 package io.quarkus.azure.functions.resteasy.runtime;
 
 import java.io.ByteArrayOutputStream;
+import java.io.IOException;
 import java.io.PrintWriter;
 import java.io.StringWriter;
 import java.nio.channels.Channels;
@@ -32,11 +33,13 @@
 import io.quarkus.vertx.http.runtime.VertxHttpRecorder;
 
 public class BaseFunction {
-    private static final Logger log = Logger.getLogger("io.quarkus.azure");
+    private static final Logger LOG = Logger.getLogger("io.quarkus.azure");
 
     protected static final String deploymentStatus;
     protected static boolean started = false;
 
+    private static final int BUFFER_SIZE = 8096;
+
     static {
         StringWriter error = new StringWriter();
         PrintWriter errorWriter = new PrintWriter(error, true);
@@ -106,7 +109,7 @@ protected HttpResponseMessage nettyDispatch(HttpRequestMessage<Optional<byte[]>>
 
     private ByteArrayOutputStream createByteStream() {
         ByteArrayOutputStream baos;
-        baos = new ByteArrayOutputStream(500);
+        baos = new ByteArrayOutputStream(BUFFER_SIZE);
         return baos;
     }
 
@@ -136,7 +139,6 @@ public void handleMessage(Object msg) {
                 if (msg instanceof HttpContent) {
                     HttpContent content = (HttpContent) msg;
                     if (baos == null) {
-                        // todo what is right size?
                         baos = createByteStream();
                     }
                     int readable = content.content().readableBytes();
@@ -156,6 +158,11 @@ public void handleMessage(Object msg) {
                 }
                 if (msg instanceof LastHttpContent) {
                     responseBuilder.body(baos.toByteArray());
+                    try {
+                        baos.close();
+                    } catch (IOException e) {
+                        LOG.warn("Unable to close the ByteArrayOutputStream", e);
+                    }
                     future.complete(responseBuilder.build());
                 }
             } catch (Throwable ex) {