allow customization of POST stream strategy

alexsteinerde · Aug 16, 2021 · 92ffc6a · 92ffc6a
1 parent 0ecbc8e
commit 92ffc6a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/GraphQLKit/Graphiti+Router.swift b/Sources/GraphQLKit/Graphiti+Router.swift
@@ -3,8 +3,8 @@ import Graphiti
 import GraphQL
 
 extension RoutesBuilder {
-    public func register<RootType>(graphQLSchema schema: Schema<RootType, Request>, withResolver rootAPI: RootType, at path: PathComponent="graphql") {
-        self.post(path) { (request) -> EventLoopFuture<Response> in
+    public func register<RootType>(graphQLSchema schema: Schema<RootType, Request>, withResolver rootAPI: RootType, at path: PathComponent="graphql", postBodyStreamStrategy: HTTPBodyStreamStrategy = .collect) {
+        self.on(.POST, path, body: postBodyStreamStrategy) { (request) -> EventLoopFuture<Response> in
             try request.resolveByBody(graphQLSchema: schema, with: rootAPI)
                 .flatMap({
                     $0.encodeResponse(status: .ok, for: request)