init

Author: azkrishpy

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

@@ -49,6 +49,21 @@ public CompletableFuture<HttpStream> acquireStream(HttpRequest request,
         return this.acquireStream((HttpRequestBase) request, streamHandler);
     }
 
+    /**
+     * Request an HTTP/1.1 HttpStream from StreamManager.
+     *
+     * @param request         HttpRequest. The Request to make to the Server.
+     * @param streamHandler   HttpStreamBaseResponseHandler. The Stream Handler to be called from the Native EventLoop
+     * @param useManualDataWrites A boolean variable to signal that body will be streamed using async writes.
+     * @return A future for a HttpStream that will be completed when the stream is
+     *         acquired.
+     * @throws CrtRuntimeException Exception happens from acquiring stream.
+     */
+    public CompletableFuture<HttpStream> acquireStream(HttpRequest request,
+            HttpStreamBaseResponseHandler streamHandler, boolean useManualDataWrites) {
+        return this.acquireStream((HttpRequestBase) request, streamHandler, useManualDataWrites);
+    }
+
     /**
      * Request an HTTP/1.1 HttpStream from StreamManager.
      *
@@ -60,6 +75,21 @@ public CompletableFuture<HttpStream> acquireStream(HttpRequest request,
      */
     public CompletableFuture<HttpStream> acquireStream(HttpRequestBase request,
             HttpStreamBaseResponseHandler streamHandler) {
+        return this.acquireStream(request, streamHandler, false); // overloading to ensure backward-compatibility
+    }
+
+    /**
+     * Request an HTTP/1.1 HttpStream from StreamManager.
+     *
+     * @param request         HttpRequestBase. The Request to make to the Server.
+     * @param streamHandler   HttpStreamBaseResponseHandler. The Stream Handler to be called from the Native EventLoop
+     * @param useManualDataWrites A boolean variable to signal that body will be streamed using async writes.
+     * @return A future for a HttpStream that will be completed when the stream is
+     *         acquired.
+     * @throws CrtRuntimeException Exception happens from acquiring stream.
+     */
+    public CompletableFuture<HttpStream> acquireStream(HttpRequestBase request,
+            HttpStreamBaseResponseHandler streamHandler, boolean usingManualDataWrites) {
         CompletableFuture<HttpStream> completionFuture = new CompletableFuture<>();
         HttpClientConnectionManager connManager = this.connectionManager;
 

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

@@ -55,7 +55,27 @@ public CompletableFuture<HttpStreamBase> acquireStream(HttpRequestBase request,
             return this.h2StreamManager.acquireStream(request, streamHandler)
                     .thenApply(stream -> (HttpStreamBase) stream);
         } else {
-            return this.h1StreamManager.acquireStream(request, streamHandler)
+            return this.h1StreamManager.acquireStream(request, streamHandler, false)
+                    .thenApply(stream -> (HttpStreamBase) stream);
+        }
+    }
+
+    /**
+     * Request an HttpStream from StreamManager.
+     *
+     * @param request         HttpRequestBase. The Request to make to the Server.
+     * @param streamHandler   HttpStreamBaseResponseHandler. The Stream Handler to be called from the Native EventLoop
+     * @param useManualDataWrites A boolean variable to signal that body will be streamed using async writes.
+     * @return A future for a HttpStreamBase that will be completed when the stream is
+     *         acquired.
+     */
+    public CompletableFuture<HttpStreamBase> acquireStream(HttpRequestBase request,
+            HttpStreamBaseResponseHandler streamHandler, boolean useManualDataWrites) {
+        if (this.h2StreamManager != null) {
+            return this.h2StreamManager.acquireStream(request, streamHandler)
+                    .thenApply(stream -> (HttpStreamBase) stream);
+        } else {
+            return this.h1StreamManager.acquireStream(request, streamHandler, useManualDataWrites)
                     .thenApply(stream -> (HttpStreamBase) stream);
         }
     }