Make execution strategies configurable (#281)

Author: fehnomenal

graphql-jpa-query-boot-starter/src/main/java/com/introproventures/graphql/jpa/query/boot/autoconfigure/GraphQLJpaQueryAutoConfiguration.java

@@ -41,6 +41,7 @@
 
 import graphql.GraphQL;
 import graphql.GraphQLContext;
+import graphql.execution.ExecutionStrategy;
 import graphql.execution.instrumentation.Instrumentation;
 import graphql.schema.GraphQLSchema;
 import graphql.schema.visibility.GraphqlFieldVisibility;
@@ -80,13 +81,19 @@ public GraphQLSchemaConfigurer graphQLJpaQuerySchemaConfigurer(GraphQLSchemaBuil
         public GraphQLExecutorContextFactory graphQLJpaExecutorContextFactory(ObjectProvider<GraphQLExecutionInputFactory> graphQLExecutionInputFactory,
                                                                            ObjectProvider<Supplier<GraphqlFieldVisibility>> graphqlFieldVisibility,
                                                                            ObjectProvider<Supplier<Instrumentation>> instrumentation,
-                                                                           ObjectProvider<Supplier<GraphQLContext>> graphqlContext) {
+                                                                           ObjectProvider<Supplier<GraphQLContext>> graphqlContext,
+                                                                              ObjectProvider<Supplier<ExecutionStrategy>> queryExecutionStrategy,
+                                                                              ObjectProvider<Supplier<ExecutionStrategy>> mutationExecutionStrategy,
+                                                                              ObjectProvider<Supplier<ExecutionStrategy>> subscriptionExecutionStrategy) {
             GraphQLJpaExecutorContextFactory bean = new GraphQLJpaExecutorContextFactory();
 
             graphQLExecutionInputFactory.ifAvailable(bean::withExecutionInputFactory);
             graphqlFieldVisibility.ifAvailable(bean::withGraphqlFieldVisibility);
             instrumentation.ifAvailable(bean::withInstrumentation);
             graphqlContext.ifAvailable(bean::withGraphqlContext);
+            queryExecutionStrategy.ifAvailable(bean::withQueryExecutionStrategy);
+            mutationExecutionStrategy.ifAvailable(bean::withMutationExecutionStrategy);
+            subscriptionExecutionStrategy.ifAvailable(bean::withSubscriptionExecutionStrategy);
 
             return bean;
         }

graphql-jpa-query-schema/src/main/java/com/introproventures/graphql/jpa/query/schema/impl/GraphQLJpaExecutorContext.java

@@ -29,6 +29,7 @@
 import graphql.ExecutionInput;
 import graphql.GraphQL;
 import graphql.GraphQLContext;
+import graphql.execution.ExecutionStrategy;
 import graphql.execution.instrumentation.ChainedInstrumentation;
 import graphql.execution.instrumentation.Instrumentation;
 import graphql.execution.instrumentation.dataloader.DataLoaderDispatcherInstrumentation;
@@ -48,6 +49,9 @@ public class GraphQLJpaExecutorContext implements GraphQLExecutorContext {
     private final Supplier<GraphQLContext> graphqlContext;
     private final Supplier<DataLoaderDispatcherInstrumentationOptions> dataLoaderDispatcherInstrumentationOptions;
     private final Supplier<DataLoaderRegistry> dataLoaderRegistry;
+    private final Supplier<ExecutionStrategy> queryExecutionStrategy;
+    private final Supplier<ExecutionStrategy> mutationExecutionStrategy;
+    private final Supplier<ExecutionStrategy> subscriptionExecutionStrategy;
 
     private GraphQLJpaExecutorContext(Builder builder) {
         this.graphQLSchema = builder.graphQLSchema;
@@ -57,6 +61,9 @@ private GraphQLJpaExecutorContext(Builder builder) {
         this.graphqlContext = builder.graphqlContext;
         this.dataLoaderDispatcherInstrumentationOptions = builder.dataLoaderDispatcherInstrumentationOptions;
         this.dataLoaderRegistry = builder.dataLoaderRegistry;
+        this.queryExecutionStrategy = builder.queryExecutionStrategy;
+        this.mutationExecutionStrategy = builder.mutationExecutionStrategy;
+        this.subscriptionExecutionStrategy = builder.subscriptionExecutionStrategy;
     }
 
     @Override
@@ -74,7 +81,10 @@ public GraphQL.Builder newGraphQL() {
         Instrumentation instrumentation = newIstrumentation();
 
         return GraphQL.newGraphQL(getGraphQLSchema())
-                      .instrumentation(instrumentation);
+                      .instrumentation(instrumentation)
+                      .queryExecutionStrategy(queryExecutionStrategy.get())
+                      .mutationExecutionStrategy(mutationExecutionStrategy.get())
+                      .subscriptionExecutionStrategy(subscriptionExecutionStrategy.get());
     }
 
     public GraphQLContext newGraphQLContext(DataLoaderRegistry dataLoaderRegistry) {
@@ -135,6 +145,12 @@ public interface IBuildStage {
 
         public IBuildStage dataLoaderRegistry(Supplier<DataLoaderRegistry> dataLoaderRegistry);
 
+        public IBuildStage queryExecutionStrategy(Supplier<ExecutionStrategy> queryExecutionStrategy);
+
+        public IBuildStage mutationExecutionStrategy(Supplier<ExecutionStrategy> mutationExecutionStrategy);
+
+        public IBuildStage subscriptionExecutionStrategy(Supplier<ExecutionStrategy> subscriptionExecutionStrategy);
+
         public GraphQLJpaExecutorContext build();
     }
 
@@ -150,6 +166,9 @@ public static final class Builder implements IGraphQLSchemaStage, IBuildStage {
         private Supplier<GraphQLContext> graphqlContext;
         private Supplier<DataLoaderDispatcherInstrumentationOptions> dataLoaderDispatcherInstrumentationOptions;
         private Supplier<DataLoaderRegistry> dataLoaderRegistry;
+        private Supplier<ExecutionStrategy> queryExecutionStrategy;
+        private Supplier<ExecutionStrategy> mutationExecutionStrategy;
+        private Supplier<ExecutionStrategy> subscriptionExecutionStrategy;
 
         private Builder() {
         }
@@ -198,6 +217,27 @@ public IBuildStage dataLoaderRegistry(Supplier<DataLoaderRegistry> dataLoaderReg
             return this;
         }
 
+        @Override
+        public IBuildStage queryExecutionStrategy(Supplier<ExecutionStrategy> queryExecutionStrategy) {
+            this.queryExecutionStrategy = queryExecutionStrategy;
+
+            return this;
+        }
+
+        @Override
+        public IBuildStage mutationExecutionStrategy(Supplier<ExecutionStrategy> mutationExecutionStrategy) {
+            this.mutationExecutionStrategy = mutationExecutionStrategy;
+
+            return this;
+        }
+
+        @Override
+        public IBuildStage subscriptionExecutionStrategy(Supplier<ExecutionStrategy> subscriptionExecutionStrategy) {
+            this.subscriptionExecutionStrategy = subscriptionExecutionStrategy;
+
+            return this;
+        }
+
         @Override
         public GraphQLJpaExecutorContext build() {
             return new GraphQLJpaExecutorContext(this);

graphql-jpa-query-schema/src/main/java/com/introproventures/graphql/jpa/query/schema/impl/GraphQLJpaExecutorContextFactory.java

@@ -27,6 +27,10 @@
 import com.introproventures.graphql.jpa.query.schema.GraphQLExecutorContext;
 import com.introproventures.graphql.jpa.query.schema.GraphQLExecutorContextFactory;
 import graphql.GraphQLContext;
+import graphql.execution.AsyncExecutionStrategy;
+import graphql.execution.AsyncSerialExecutionStrategy;
+import graphql.execution.ExecutionStrategy;
+import graphql.execution.SubscriptionExecutionStrategy;
 import graphql.execution.instrumentation.Instrumentation;
 import graphql.execution.instrumentation.SimpleInstrumentation;
 import graphql.execution.instrumentation.dataloader.DataLoaderDispatcherInstrumentationOptions;
@@ -57,6 +61,10 @@ public class GraphQLJpaExecutorContextFactory implements GraphQLExecutorContextF
         return BatchLoaderRegistry.newDataLoaderRegistry(options);
     };
 
+    private Supplier<ExecutionStrategy> queryExecutionStrategy = AsyncExecutionStrategy::new;
+    private Supplier<ExecutionStrategy> mutationExecutionStrategy = AsyncSerialExecutionStrategy::new;
+    private Supplier<ExecutionStrategy> subscriptionExecutionStrategy = SubscriptionExecutionStrategy::new;
+
 
     public GraphQLJpaExecutorContextFactory() {
     }
@@ -71,6 +79,9 @@ public GraphQLExecutorContext newExecutorContext(GraphQLSchema graphQLSchema) {
                                         .graphqlContext(graphqlContext)
                                         .dataLoaderDispatcherInstrumentationOptions(dataLoaderDispatcherInstrumentationOptions)
                                         .dataLoaderRegistry(dataLoaderRegistry)
+                                        .queryExecutionStrategy(queryExecutionStrategy)
+                                        .mutationExecutionStrategy(mutationExecutionStrategy)
+                                        .subscriptionExecutionStrategy(subscriptionExecutionStrategy)
                                         .build();
     }
 
@@ -101,6 +112,22 @@ public GraphQLJpaExecutorContextFactory withDataLoaderDispatcherInstrumentationO
         return this;
     }
 
+    public GraphQLJpaExecutorContextFactory withQueryExecutionStrategy(Supplier<ExecutionStrategy> queryExecutionStrategy) {
+        this.queryExecutionStrategy = queryExecutionStrategy;
+        return this;
+    }
+
+    public GraphQLJpaExecutorContextFactory withMutationExecutionStrategy(Supplier<ExecutionStrategy> mutationExecutionStrategy) {
+        this.mutationExecutionStrategy = mutationExecutionStrategy;
+        return this;
+    }
+
+    public GraphQLJpaExecutorContextFactory withSubscriptionExecutionStrategy(Supplier<ExecutionStrategy> subscriptionExecutionStrategy) {
+        this.subscriptionExecutionStrategy = subscriptionExecutionStrategy;
+        return this;
+    }
+
+
     public GraphQLExecutionInputFactory getExecutionInputFactory() {
         return executionInputFactory;
     }