add for h2

Author: azkrishpy

crt/aws-c-http

@@ -151,17 +151,48 @@ private Http2StreamManager(Http2StreamManagerOptions options) {
     public CompletableFuture<Http2Stream> acquireStream(Http2Request request,
             HttpStreamBaseResponseHandler streamHandler) {
 
-        return this.acquireStream((HttpRequestBase) request, streamHandler);
+        return this.acquireStream((HttpRequestBase) request, streamHandler, false);
+    }
+
+    /**
+     * Request a Http2Stream from StreamManager.
+     *
+     * @param request             The Request to make to the Server.
+     * @param streamHandler       The Stream Handler to be called from the Native
+     *                            EventLoop
+     * @param useManualDataWrites When {@code true}, request body data is provided via
+     *                            {@link HttpStreamBase#writeData} instead of from the
+     *                            request's {@link HttpRequestBodyStream}. See
+     *                            {@link Http2ClientConnection#makeRequest(HttpRequestBase, HttpStreamBaseResponseHandler, boolean)}
+     *                            for caveats on combining a body stream with manual writes.
+     * @return A future for a Http2Stream that will be completed when the stream is
+     *         acquired.
+     */
+    public CompletableFuture<Http2Stream> acquireStream(Http2Request request,
+            HttpStreamBaseResponseHandler streamHandler, boolean useManualDataWrites) {
+
+        return this.acquireStream((HttpRequestBase) request, streamHandler, useManualDataWrites);
     }
 
     public CompletableFuture<Http2Stream> acquireStream(HttpRequest request,
             HttpStreamBaseResponseHandler streamHandler) {
 
-        return this.acquireStream((HttpRequestBase) request, streamHandler);
+        return this.acquireStream((HttpRequestBase) request, streamHandler, false);
+    }
+
+    public CompletableFuture<Http2Stream> acquireStream(HttpRequest request,
+            HttpStreamBaseResponseHandler streamHandler, boolean useManualDataWrites) {
+
+        return this.acquireStream((HttpRequestBase) request, streamHandler, useManualDataWrites);
     }
 
     public CompletableFuture<Http2Stream> acquireStream(HttpRequestBase request,
         HttpStreamBaseResponseHandler streamHandler) {
+        return this.acquireStream(request, streamHandler, false);
+    }
+
+    public CompletableFuture<Http2Stream> acquireStream(HttpRequestBase request,
+        HttpStreamBaseResponseHandler streamHandler, boolean useManualDataWrites) {
         CompletableFuture<Http2Stream> completionFuture = new CompletableFuture<>();
         AsyncCallback acquireStreamCompleted = AsyncCallback.wrapFuture(completionFuture, null);
         if (isNull()) {
@@ -174,7 +205,8 @@ public CompletableFuture<Http2Stream> acquireStream(HttpRequestBase request,
                     request.marshalForJni(),
                     request.getBodyStream(),
                     new HttpStreamResponseHandlerNativeAdapter(streamHandler),
-                    acquireStreamCompleted);
+                    acquireStreamCompleted,
+                    useManualDataWrites);
         } catch (CrtRuntimeException ex) {
             completionFuture.completeExceptionally(ex);
         }
@@ -278,7 +310,8 @@ private static native void http2StreamManagerAcquireStream(long stream_manager,
             byte[] marshalledRequest,
             HttpRequestBodyStream bodyStream,
             HttpStreamResponseHandlerNativeAdapter responseHandler,
-            AsyncCallback completedCallback) throws CrtRuntimeException;
+            AsyncCallback completedCallback,
+            boolean useManualDataWrites) throws CrtRuntimeException;
 
     private static native HttpManagerMetrics http2StreamManagerFetchMetrics(long stream_manager) throws CrtRuntimeException;
 }

src/main/java/software/amazon/awssdk/crt/http/Http2StreamManager.java

@@ -72,7 +72,7 @@ public CompletableFuture<HttpStreamBase> acquireStream(HttpRequestBase request,
     public CompletableFuture<HttpStreamBase> acquireStream(HttpRequestBase request,
             HttpStreamBaseResponseHandler streamHandler, boolean useManualDataWrites) {
         if (this.h2StreamManager != null) {
-            return this.h2StreamManager.acquireStream(request, streamHandler)
+            return this.h2StreamManager.acquireStream(request, streamHandler, useManualDataWrites)
                     .thenApply(stream -> (HttpStreamBase) stream);
         } else {
             return this.h1StreamManager.acquireStream(request, streamHandler, useManualDataWrites)

src/main/java/software/amazon/awssdk/crt/http/HttpStreamManager.java

@@ -365,7 +365,8 @@ JNIEXPORT void JNICALL Java_software_amazon_awssdk_crt_http_Http2StreamManager_h
     jbyteArray marshalled_request,
     jobject jni_http_request_body_stream,
     jobject jni_http_response_callback_handler,
-    jobject java_async_callback) {
+    jobject java_async_callback,
+    jboolean jni_use_manual_data_writes) {
     (void)jni_class;
     aws_cache_jni_ids(env);
 
@@ -411,6 +412,7 @@ JNIEXPORT void JNICALL Java_software_amazon_awssdk_crt_http_Http2StreamManager_h
         .on_complete = aws_java_http_stream_on_stream_complete_fn,
         .on_destroy = aws_java_http_stream_on_stream_destroy_fn,
         .user_data = stream_binding,
+        .use_manual_data_writes = jni_use_manual_data_writes,
     };
 
     struct aws_allocator *allocator = aws_jni_get_allocator();