add ExecutionInputCustomizer

graphql-java-spring-webflux/src/main/java/graphql/spring/web/reactive/ExecutionInputCustomizer.java

@@ -0,0 +1,13 @@
+package graphql.spring.web.reactive;
+
+import graphql.ExecutionInput;
+import graphql.PublicApi;
+import org.springframework.web.server.ServerWebExchange;
+import reactor.core.publisher.Mono;
+
+@PublicApi
+public interface ExecutionInputCustomizer {
+
+    Mono<ExecutionInput> customizeExecutionInput(ExecutionInput executionInput, ServerWebExchange webRequest);
+
+}

graphql-java-spring-webflux/src/main/java/graphql/spring/web/reactive/components/DefaultExecutionInputCustomizer.java

@@ -0,0 +1,18 @@
+package graphql.spring.web.reactive.components;
+
+import graphql.ExecutionInput;
+import graphql.Internal;
+import graphql.spring.web.reactive.ExecutionInputCustomizer;
+import org.springframework.stereotype.Component;
+import org.springframework.web.server.ServerWebExchange;
+import reactor.core.publisher.Mono;
+
+@Component
+@Internal
+public class DefaultExecutionInputCustomizer implements ExecutionInputCustomizer {
+
+    @Override
+    public Mono<ExecutionInput> customizeExecutionInput(ExecutionInput executionInput, ServerWebExchange webRequest) {
+        return Mono.just(executionInput);
+    }
+}

graphql-java-spring-webflux/src/main/java/graphql/spring/web/reactive/components/DefaultGraphQLInvocation.java

@@ -4,6 +4,7 @@
 import graphql.ExecutionResult;
 import graphql.GraphQL;
 import graphql.Internal;
+import graphql.spring.web.reactive.ExecutionInputCustomizer;
 import graphql.spring.web.reactive.GraphQLInvocation;
 import graphql.spring.web.reactive.GraphQLInvocationData;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -18,14 +19,18 @@ public class DefaultGraphQLInvocation implements GraphQLInvocation {
     @Autowired
     private GraphQL graphQL;
 
+    @Autowired
+    ExecutionInputCustomizer executionInputCustomizer;
+
     @Override
     public Mono<ExecutionResult> invoke(GraphQLInvocationData invocationData, ServerWebExchange serverWebExchange) {
         ExecutionInput executionInput = ExecutionInput.newExecutionInput()
                 .query(invocationData.getQuery())
                 .operationName(invocationData.getOperationName())
                 .variables(invocationData.getVariables())
                 .build();
-        return Mono.fromCompletionStage(graphQL.executeAsync(executionInput));
+        Mono<ExecutionInput> customizedExecutionInputMono = executionInputCustomizer.customizeExecutionInput(executionInput, serverWebExchange);
+        return customizedExecutionInputMono.flatMap(customizedExecutionInput -> Mono.fromCompletionStage(graphQL.executeAsync(customizedExecutionInput)));
     }
 
 }

graphql-java-spring-webmvc/src/main/java/graphql/spring/web/servlet/ExecutionInputCustomizer.java

@@ -0,0 +1,14 @@
+package graphql.spring.web.servlet;
+
+import graphql.ExecutionInput;
+import graphql.PublicApi;
+import org.springframework.web.context.request.WebRequest;
+
+import java.util.concurrent.CompletableFuture;
+
+@PublicApi
+public interface ExecutionInputCustomizer {
+
+    CompletableFuture<ExecutionInput> customizeExecutionInput(ExecutionInput executionInput, WebRequest webRequest);
+
+}

graphql-java-spring-webmvc/src/main/java/graphql/spring/web/servlet/components/DefaultExecutionInputCustomizer.java

@@ -0,0 +1,19 @@
+package graphql.spring.web.servlet.components;
+
+import graphql.ExecutionInput;
+import graphql.Internal;
+import graphql.spring.web.servlet.ExecutionInputCustomizer;
+import org.springframework.stereotype.Component;
+import org.springframework.web.context.request.WebRequest;
+
+import java.util.concurrent.CompletableFuture;
+
+@Component
+@Internal
+public class DefaultExecutionInputCustomizer implements ExecutionInputCustomizer {
+
+    @Override
+    public CompletableFuture<ExecutionInput> customizeExecutionInput(ExecutionInput executionInput, WebRequest webRequest) {
+        return CompletableFuture.completedFuture(executionInput);
+    }
+}

graphql-java-spring-webmvc/src/main/java/graphql/spring/web/servlet/components/DefaultGraphQLInvocation.java

@@ -4,6 +4,7 @@
 import graphql.ExecutionResult;
 import graphql.GraphQL;
 import graphql.Internal;
+import graphql.spring.web.servlet.ExecutionInputCustomizer;
 import graphql.spring.web.servlet.GraphQLInvocation;
 import graphql.spring.web.servlet.GraphQLInvocationData;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -19,14 +20,18 @@ public class DefaultGraphQLInvocation implements GraphQLInvocation {
     @Autowired
     private GraphQL graphQL;
 
+    @Autowired
+    ExecutionInputCustomizer executionInputCustomizer;
+
     @Override
     public CompletableFuture<ExecutionResult> invoke(GraphQLInvocationData invocationData, WebRequest webRequest) {
         ExecutionInput executionInput = ExecutionInput.newExecutionInput()
                 .query(invocationData.getQuery())
                 .operationName(invocationData.getOperationName())
                 .variables(invocationData.getVariables())
                 .build();
-        return graphQL.executeAsync(executionInput);
+        CompletableFuture<ExecutionInput> customizedExecutionInput = executionInputCustomizer.customizeExecutionInput(executionInput, webRequest);
+        return customizedExecutionInput.thenCompose(graphQL::executeAsync);
     }
 
 }