Added support for setting context object in execution input

README.md

@@ -111,10 +111,11 @@ The following properties are currently available:
 
 The following Beans can be overridden by providing a different implementation. 
 
-| Interface | Description | 
-| --- | --- | 
-| GraphQLInvocation | Executes one request. The default impl just calls the provided `GraphQL` bean.|
-| ExecutionResultHandler | Takes a `ExecutionResult` and sends the result back to the client. The default impl returns `ExecutionResult.toSpecification()` as json. |
+| Interface | Description | Default Implementation |
+| --- | --- | --- |
+| `GraphQLInvocation` | Executes one request. The default impl just calls the provided `GraphQL` bean. | `DefaultGraphQLInvocation` |
+| `GraphQLContextBuilder` | Builds the context `Object` passed to the `ExecutionInput`. Used by the `DefaultGraphQLInvocation`. | None |
+| `ExecutionResultHandler` | Takes a `ExecutionResult` and sends the result back to the client. The default impl returns `ExecutionResult.toSpecification()` as json. | `DefaultExecutionResultHandler` |
 
 
 

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

@@ -0,0 +1,10 @@
+package graphql.spring.web.reactive;
+
+import graphql.PublicApi;
+import org.springframework.web.server.ServerWebExchange;
+
+@PublicApi
+public interface GraphQLContextBuilder {
+
+    Object build(ServerWebExchange webRequest);
+}

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.GraphQLContextBuilder;
 import graphql.spring.web.reactive.GraphQLInvocation;
 import graphql.spring.web.reactive.GraphQLInvocationData;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -18,9 +19,14 @@ public class DefaultGraphQLInvocation implements GraphQLInvocation {
     @Autowired
     private GraphQL graphQL;
 
+    @Autowired(required = false)
+    private GraphQLContextBuilder contextBuilder;
+
     @Override
     public Mono<ExecutionResult> invoke(GraphQLInvocationData invocationData, ServerWebExchange serverWebExchange) {
+        Object context = contextBuilder != null ? contextBuilder.build(serverWebExchange) : null;
         ExecutionInput executionInput = ExecutionInput.newExecutionInput()
+                .context(context)
                 .query(invocationData.getQuery())
                 .operationName(invocationData.getOperationName())
                 .variables(invocationData.getVariables())

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

@@ -0,0 +1,10 @@
+package graphql.spring.web.servlet;
+
+import graphql.PublicApi;
+import org.springframework.web.context.request.WebRequest;
+
+@PublicApi
+public interface GraphQLContextBuilder {
+
+    Object build(WebRequest webRequest);
+}

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.GraphQLContextBuilder;
 import graphql.spring.web.servlet.GraphQLInvocation;
 import graphql.spring.web.servlet.GraphQLInvocationData;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -19,9 +20,14 @@ public class DefaultGraphQLInvocation implements GraphQLInvocation {
     @Autowired
     private GraphQL graphQL;
 
+    @Autowired(required = false)
+    private GraphQLContextBuilder contextBuilder;
+
     @Override
     public CompletableFuture<ExecutionResult> invoke(GraphQLInvocationData invocationData, WebRequest webRequest) {
+        Object context = contextBuilder != null ? contextBuilder.build(webRequest) : null;
         ExecutionInput executionInput = ExecutionInput.newExecutionInput()
+                .context(context)
                 .query(invocationData.getQuery())
                 .operationName(invocationData.getOperationName())
                 .variables(invocationData.getVariables())

graphql-java-spring.iml

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module external.linked.project.id="graphql-java-spring" external.linked.project.path="$MODULE_DIR$" external.root.project.path="$MODULE_DIR$" external.system.id="GRADLE" external.system.module.group="" external.system.module.version="unspecified" type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/.gradle" />
+      <excludeFolder url="file://$MODULE_DIR$/build" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>