Add HBRequest.collateBody

Sources/Hummingbird/Server/Request.swift

@@ -126,6 +126,17 @@ public struct HBRequest: HBExtensible {
         return self.eventLoop.makeSucceededFuture(value)
     }
 
+    // Return new version of request with collated request body. If you want to process the
+    // request body in middleware you need to call this to ensure you have the full request
+    // body. Once this is called the request generated by this should be passed to the nextResponder
+    public func collateBody() -> EventLoopFuture<HBRequest> {
+        self.body.consumeBody(on: self.eventLoop).flatMapThrowing { buffer in
+            var request = self
+            request.body = .byteBuffer(buffer)
+            return request
+        }
+    }
+
     /// Store all the read-only values of the request in a class to avoid copying them
     /// everytime we pass the `HBRequest` struct about
     final class _Internal {

Tests/HummingbirdTests/ApplicationTests.swift

@@ -302,6 +302,29 @@ final class ApplicationTests: XCTestCase {
         }
     }
 
+    func testCollateBody() throws {
+        struct CollateMiddleware: HBMiddleware {
+            func apply(to request: HBRequest, next: HBResponder) -> EventLoopFuture<HBResponse> {
+                return request.collateBody().flatMap { request in
+                    request.logger.info("Buffer size: \(request.body.buffer!.readableBytes)")
+                    return next.respond(to: request)
+                }
+            }
+        }
+        let app = HBApplication(testing: .embedded)
+        app.middleware.add(CollateMiddleware())
+        app.router.put("/hello") { _ -> HTTPResponseStatus in
+            return .ok
+        }
+        try app.XCTStart()
+        defer { app.XCTStop() }
+
+        let buffer = self.randomBuffer(size: 512_000)
+        app.XCTExecute(uri: "/hello", method: .PUT, body: buffer) { response in
+            XCTAssertEqual(response.status, .ok)
+        }
+    }
+
     func testOptional() throws {
         let app = HBApplication(testing: .embedded)
         app.router