prototype

Author: azkrishpy

include/aws/crt/http/HttpRequestResponse.h

@@ -60,6 +60,13 @@ namespace Aws
                  */
                 bool SetBody(const std::shared_ptr<Aws::Crt::Io::InputStream> &body) noexcept;
 
+                /**
+                 * Sets an async input stream as the message body
+                 * @param body the async input stream representing the message body
+                 * @return future<bool> indicating success/failure
+                 */
+                std::future<bool> SetBody(const std::shared_ptr<Aws::Crt::Io::AsyncInputStream> &body) noexcept;
+
                 /**
                  * Gets the number of headers contained in this request
                  * @return the number of headers contained in this request
@@ -101,6 +108,7 @@ namespace Aws
                 Allocator *m_allocator;
                 struct aws_http_message *m_message;
                 std::shared_ptr<Aws::Crt::Io::InputStream> m_bodyStream;
+                std::shared_ptr<Aws::Crt::Io::AsyncInputStream> m_asyncBodyStream;
             };
 
             /**

include/aws/crt/io/Stream.h

@@ -9,6 +9,8 @@
 #include <aws/crt/Types.h>
 #include <aws/io/stream.h>
 
+#include <future>
+
 namespace Aws
 {
     namespace Crt
@@ -192,6 +194,33 @@ namespace Aws
               private:
                 std::shared_ptr<Aws::Crt::Io::IStream> m_stream;
             };
+
+            /**
+             * Interface for asynchronous input streams.
+             * Used for async HTTP request bodies.
+             */
+            class AWS_CRT_CPP_API AsyncInputStream
+            {
+              public:
+                virtual ~AsyncInputStream() = default;
+
+                AsyncInputStream(const AsyncInputStream &) = delete;
+                AsyncInputStream &operator=(const AsyncInputStream &) = delete;
+                AsyncInputStream(AsyncInputStream &&) = delete;
+                AsyncInputStream &operator=(AsyncInputStream &&) = delete;
+
+                virtual bool IsValid() const noexcept = 0;
+
+              protected:
+                AsyncInputStream() = default;
+
+                /**
+                 * Asynchronously read into buffer.
+                 * @return future<bool> - true on success (including EOF/no data available), false on error
+                 */
+                virtual std::future<bool> ReadImpl(ByteBuf &buffer) noexcept = 0;
+            };
+
         } // namespace Io
     } // namespace Crt
 } // namespace Aws

source/http/HttpRequestResponse.cpp

@@ -57,12 +57,24 @@ namespace Aws
             bool HttpMessage::SetBody(const std::shared_ptr<Aws::Crt::Io::InputStream> &body) noexcept
             {
                 m_bodyStream = body;
+                m_asyncBodyStream = nullptr;
                 aws_http_message_set_body_stream(
                     m_message, m_bodyStream && *m_bodyStream ? m_bodyStream->GetUnderlyingStream() : nullptr);
 
                 return true;
             }
 
+            std::future<bool> HttpMessage::SetBody(const std::shared_ptr<Aws::Crt::Io::AsyncInputStream> &body) noexcept
+            {
+                m_asyncBodyStream = body;
+                m_bodyStream = nullptr;
+                aws_http_message_set_body_stream(m_message, nullptr);
+
+                std::promise<bool> promise;
+                promise.set_value(body == nullptr || body->IsValid());
+                return promise.get_future();
+            }
+
             size_t HttpMessage::GetHeaderCount() const noexcept
             {
                 return aws_http_message_get_header_count(m_message);