fix in execute as well

Author: JoviDeCroock

src/execution/__tests__/executor-test.ts

@@ -22,7 +22,6 @@ import { GraphQLBoolean, GraphQLInt, GraphQLString } from '../../type/scalars';
 import { GraphQLSchema } from '../../type/schema';
 
 import { execute, executeSync } from '../execute';
-import { print } from '../../language';
 
 describe('Execute: Handles basic execution tasks', () => {
   it('throws if no document is provided', () => {
@@ -81,7 +80,6 @@ describe('Execute: Handles basic execution tasks', () => {
       }),
     });
 
-
     const n = 10000;
     const fragments = Array.from(Array(n).keys()).reduce(
       (acc, next) =>
@@ -96,7 +94,6 @@ describe('Execute: Handles basic execution tasks', () => {
     const document = parse(`
       query {
         ...X${n}
-        __typename
       }
       ${fragments}
       fragment X0 on Query {

src/execution/collectFields.ts

@@ -21,6 +21,11 @@ import { typeFromAST } from '../utilities/typeFromAST';
 
 import { getDirectiveValues } from './values';
 
+interface FragmentEntry {
+  fragment: FragmentDefinitionNode;
+  runtimeType: GraphQLObjectType;
+}
+
 /**
  * Given a selectionSet, collects all of the fields and returns them.
  *
@@ -37,16 +42,34 @@ export function collectFields(
   runtimeType: GraphQLObjectType,
   selectionSet: SelectionSetNode,
 ): Map<string, ReadonlyArray<FieldNode>> {
+  const foundFragments: Array<FragmentEntry> = [];
   const fields = new Map();
+  const visited = new Set<string>();
   collectFieldsImpl(
     schema,
     fragments,
     variableValues,
     runtimeType,
     selectionSet,
     fields,
-    new Set(),
+    visited,
+    foundFragments,
   );
+
+  let entry;
+  while ((entry = foundFragments.pop()) !== undefined) {
+    collectFieldsImpl(
+      schema,
+      fragments,
+      variableValues,
+      entry.runtimeType,
+      entry.fragment.selectionSet,
+      fields,
+      visited,
+      foundFragments,
+    );
+  }
+
   return fields;
 }
 
@@ -68,6 +91,7 @@ export function collectSubfields(
   fieldNodes: ReadonlyArray<FieldNode>,
 ): Map<string, ReadonlyArray<FieldNode>> {
   const subFieldNodes = new Map();
+  const foundFragments: Array<FragmentEntry> = [];
   const visitedFragmentNames = new Set<string>();
   for (const node of fieldNodes) {
     if (node.selectionSet) {
@@ -79,9 +103,25 @@ export function collectSubfields(
         node.selectionSet,
         subFieldNodes,
         visitedFragmentNames,
+        foundFragments,
       );
     }
   }
+
+  let entry;
+  while ((entry = foundFragments.pop()) !== undefined) {
+    collectFieldsImpl(
+      schema,
+      fragments,
+      variableValues,
+      entry.runtimeType,
+      entry.fragment.selectionSet,
+      subFieldNodes,
+      visitedFragmentNames,
+      foundFragments,
+    );
+  }
+
   return subFieldNodes;
 }
 
@@ -93,6 +133,7 @@ function collectFieldsImpl(
   selectionSet: SelectionSetNode,
   fields: Map<string, Array<FieldNode>>,
   visitedFragmentNames: Set<string>,
+  foundFragments: Array<FragmentEntry>,
 ): void {
   for (const selection of selectionSet.selections) {
     switch (selection.kind) {
@@ -124,6 +165,7 @@ function collectFieldsImpl(
           selection.selectionSet,
           fields,
           visitedFragmentNames,
+          foundFragments,
         );
         break;
       }
@@ -143,15 +185,8 @@ function collectFieldsImpl(
         ) {
           continue;
         }
-        collectFieldsImpl(
-          schema,
-          fragments,
-          variableValues,
-          runtimeType,
-          fragment.selectionSet,
-          fields,
-          visitedFragmentNames,
-        );
+
+        foundFragments.push({ runtimeType, fragment });
         break;
       }
     }