diff --git a/graphql-kickstart-spring-webflux/src/main/java/graphql/kickstart/spring/webflux/GraphQLController.java b/graphql-kickstart-spring-webflux/src/main/java/graphql/kickstart/spring/webflux/GraphQLController.java index fbb78e3b..f4b736bc 100644 --- a/graphql-kickstart-spring-webflux/src/main/java/graphql/kickstart/spring/webflux/GraphQLController.java +++ b/graphql-kickstart-spring-webflux/src/main/java/graphql/kickstart/spring/webflux/GraphQLController.java @@ -14,13 +14,14 @@ @RestController public class GraphQLController extends AbstractGraphQLController { - + private final GraphQLObjectMapper objectMapper; private final GraphQLInvoker graphQLInvoker; private final GraphQLSpringInvocationInputFactory invocationInputFactory; public GraphQLController(GraphQLObjectMapper objectMapper, GraphQLInvoker graphQLInvoker, GraphQLSpringInvocationInputFactory invocationInputFactory) { super(objectMapper); + this.objectMapper = objectMapper; this.graphQLInvoker = graphQLInvoker; this.invocationInputFactory = invocationInputFactory; } @@ -33,7 +34,7 @@ protected Object executeRequest( GraphQLSingleInvocationInput invocationInput = invocationInputFactory .create(new GraphQLRequest(query, variables, operationName), serverWebExchange); Mono executionResult = Mono.fromCompletionStage(graphQLInvoker.executeAsync(invocationInput)); - return executionResult.map(ExecutionResult::toSpecification); + return executionResult.map(objectMapper::createResultFromExecutionResult); } }