allow customization of execution for subscription events and export the default

Author: yaacovCR

src/execution/execute.ts

@@ -125,6 +125,10 @@ export interface ExecutionInfo extends DocumentInfo {
   fieldResolver: GraphQLFieldResolver<any, any>;
   typeResolver: GraphQLTypeResolver<any, any>;
   subscribeFieldResolver: GraphQLFieldResolver<any, any>;
+  subscriptionEventExecutor: (
+    exeInfo: ExecutionInfo,
+    payload: unknown,
+  ) => PromiseOrValue<ExecutionResult>;
 }
 
 /**
@@ -169,6 +173,12 @@ export interface ExecutionArgs {
   fieldResolver?: Maybe<GraphQLFieldResolver<any, any>>;
   typeResolver?: Maybe<GraphQLTypeResolver<any, any>>;
   subscribeFieldResolver?: Maybe<GraphQLFieldResolver<any, any>>;
+  subscriptionEventExecutor?: Maybe<
+    (
+      exeInfo: ExecutionInfo,
+      payload: unknown,
+    ) => PromiseOrValue<ExecutionResult>
+  >;
 }
 
 type FieldsExecutor = (
@@ -329,6 +339,7 @@ export function buildExecutionInfo(
     fieldResolver,
     typeResolver,
     subscribeFieldResolver,
+    subscriptionEventExecutor,
   } = args;
 
   // If the schema used for execution is invalid, throw an error.
@@ -362,6 +373,8 @@ export function buildExecutionInfo(
     fieldResolver: fieldResolver ?? defaultFieldResolver,
     typeResolver: typeResolver ?? defaultTypeResolver,
     subscribeFieldResolver: subscribeFieldResolver ?? defaultFieldResolver,
+    subscriptionEventExecutor:
+      subscriptionEventExecutor ?? defaultSubscriptionEventExecutor,
   };
 }
 
@@ -1244,13 +1257,19 @@ function mapSourceToResponse(
   // "ExecuteSubscriptionEvent" algorithm, as it is nearly identical to the
   // "ExecuteQuery" algorithm, for which `execute` is also used.
   return mapAsyncIterator(resultOrStream, (payload: unknown) =>
-    executeQuery({
-      ...exeInfo,
-      rootValue: payload,
-    }),
+    exeInfo.subscriptionEventExecutor(exeInfo, payload),
   );
 }
 
+export const defaultSubscriptionEventExecutor = (
+  exeInfo: ExecutionInfo,
+  payload: unknown,
+): PromiseOrValue<ExecutionResult> =>
+  executeQuery({
+    ...exeInfo,
+    rootValue: payload,
+  });
+
 /**
  * Implements the "CreateSourceEventStream" algorithm described in the
  * GraphQL specification, resolving the subscription source event stream.

src/execution/index.ts

@@ -5,6 +5,7 @@ export {
   executeSync,
   defaultFieldResolver,
   defaultTypeResolver,
+  defaultSubscriptionEventExecutor,
 } from './execute';
 
 export type {

src/index.ts

@@ -315,6 +315,7 @@ export {
   executeSync,
   defaultFieldResolver,
   defaultTypeResolver,
+  defaultSubscriptionEventExecutor,
   responsePathAsArray,
   getArgumentValues,
   getVariableValues,