Merge pull request #9 from jaredh159/body-stream-strategy

allow customization of POST stream strategy
alexsteinerde · Aug 20, 2021 · c0cc82e · c0cc82e
2 parents 0ecbc8e + 92ffc6a
commit c0cc82e
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)