From 32c5b0d05017880439503e3e4a325f9a40611269 Mon Sep 17 00:00:00 2001
From: Yaacov Rydzinski <yaacovCR@gmail.com>
Date: Tue, 10 Dec 2024 05:06:02 +0200
Subject: [PATCH] move subscription `@defer` check out of collectFields (#4308)

this way, if (when?) we re-enable incemental delivery support with subscriptions, the signature of collect fields will not need to change
---
 src/execution/collectFields.ts | 15 ---------------
 src/execution/execute.ts       | 16 +++++++++++++---
 2 files changed, 13 insertions(+), 18 deletions(-)

diff --git a/src/execution/collectFields.ts b/src/execution/collectFields.ts
index 84d4db1595..3fbadaeafe 100644
--- a/src/execution/collectFields.ts
+++ b/src/execution/collectFields.ts
@@ -1,5 +1,4 @@
 import { AccumulatorMap } from '../jsutils/AccumulatorMap.js';
-import { invariant } from '../jsutils/invariant.js';
 import type { ObjMap } from '../jsutils/ObjMap.js';
 
 import type {
@@ -10,7 +9,6 @@ import type {
   OperationDefinitionNode,
   SelectionSetNode,
 } from '../language/ast.js';
-import { OperationTypeNode } from '../language/ast.js';
 import { Kind } from '../language/kinds.js';
 
 import type { GraphQLObjectType } from '../type/definition.js';
@@ -52,7 +50,6 @@ interface CollectFieldsContext {
   schema: GraphQLSchema;
   fragments: ObjMap<FragmentDetails>;
   variableValues: VariableValues;
-  operation: OperationDefinitionNode;
   runtimeType: GraphQLObjectType;
   visitedFragmentNames: Set<string>;
   hideSuggestions: boolean;
@@ -86,7 +83,6 @@ export function collectFields(
     fragments,
     variableValues,
     runtimeType,
-    operation,
     visitedFragmentNames: new Set(),
     hideSuggestions,
   };
@@ -115,7 +111,6 @@ export function collectSubfields(
   schema: GraphQLSchema,
   fragments: ObjMap<FragmentDetails>,
   variableValues: VariableValues,
-  operation: OperationDefinitionNode,
   returnType: GraphQLObjectType,
   fieldDetailsList: FieldDetailsList,
   hideSuggestions: boolean,
@@ -128,7 +123,6 @@ export function collectSubfields(
     fragments,
     variableValues,
     runtimeType: returnType,
-    operation,
     visitedFragmentNames: new Set(),
     hideSuggestions,
   };
@@ -170,7 +164,6 @@ function collectFieldsImpl(
     fragments,
     variableValues,
     runtimeType,
-    operation,
     visitedFragmentNames,
     hideSuggestions,
   } = context;
@@ -203,7 +196,6 @@ function collectFieldsImpl(
         }
 
         const newDeferUsage = getDeferUsage(
-          operation,
           variableValues,
           fragmentVariableValues,
           selection,
@@ -237,7 +229,6 @@ function collectFieldsImpl(
         const fragName = selection.name.value;
 
         const newDeferUsage = getDeferUsage(
-          operation,
           variableValues,
           fragmentVariableValues,
           selection,
@@ -309,7 +300,6 @@ function collectFieldsImpl(
  * not disabled by the "if" argument.
  */
 function getDeferUsage(
-  operation: OperationDefinitionNode,
   variableValues: VariableValues,
   fragmentVariableValues: VariableValues | undefined,
   node: FragmentSpreadNode | InlineFragmentNode,
@@ -330,11 +320,6 @@ function getDeferUsage(
     return;
   }
 
-  invariant(
-    operation.operation !== OperationTypeNode.SUBSCRIPTION,
-    '`@defer` directive not supported on subscription operations. Disable `@defer` by setting the `if` argument to `false`.',
-  );
-
   return {
     label: typeof defer.label === 'string' ? defer.label : undefined,
     parentDeferUsage,
diff --git a/src/execution/execute.ts b/src/execution/execute.ts
index 00f0e7b1ae..4604cb4c6f 100644
--- a/src/execution/execute.ts
+++ b/src/execution/execute.ts
@@ -103,13 +103,12 @@ const collectSubfields = memoize3(
     returnType: GraphQLObjectType,
     fieldDetailsList: FieldDetailsList,
   ) => {
-    const { schema, fragments, operation, variableValues, hideSuggestions } =
+    const { schema, fragments, variableValues, hideSuggestions } =
       validatedExecutionArgs;
     return _collectSubfields(
       schema,
       fragments,
       variableValues,
-      operation,
       returnType,
       fieldDetailsList,
       hideSuggestions,
@@ -1894,13 +1893,24 @@ function collectAndExecuteSubfields(
   incrementalContext: IncrementalContext | undefined,
   deferMap: ReadonlyMap<DeferUsage, DeferredFragmentRecord> | undefined,
 ): PromiseOrValue<GraphQLWrappedResult<ObjMap<unknown>>> {
+  const validatedExecutionArgs = exeContext.validatedExecutionArgs;
+
   // Collect sub-fields to execute to complete this value.
   const collectedSubfields = collectSubfields(
-    exeContext.validatedExecutionArgs,
+    validatedExecutionArgs,
     returnType,
     fieldDetailsList,
   );
   const { groupedFieldSet, newDeferUsages } = collectedSubfields;
+
+  if (newDeferUsages.length > 0) {
+    invariant(
+      validatedExecutionArgs.operation.operation !==
+        OperationTypeNode.SUBSCRIPTION,
+      '`@defer` directive not supported on subscription operations. Disable `@defer` by setting the `if` argument to `false`.',
+    );
+  }
+
   return executeSubExecutionPlan(
     exeContext,
     returnType,