@graphitation/supermassive 3.17.2-canary.1 → 3.18.0

package/lib/executeWithoutSchema.mjs

@@ -9,6 +9,7 @@ import {
   collectSubfields as _collectSubfields
 } from "./collectFields.mjs";
 import { devAssert } from "./jsutils/devAssert.mjs";
+import { getObjectAtPath } from "./jsutils/getObjectAtPath.mjs";
 import { inspect } from "./jsutils/inspect.mjs";
 import { invariant } from "./jsutils/invariant.mjs";
 import { isIterableObject } from "./jsutils/isIterableObject.mjs";
@@ -69,7 +70,10 @@ function buildExecutionContext(args) {
     typeResolver,
     subscribeFieldResolver,
     fieldExecutionHooks,
-    enablePerEventContext
+    enablePerEventContext,
+    enableEarlyExecution,
+    enableDeferredMerge,
+    incrementalPayloadBatchingTimeoutMs
   } = args;
   assertValidExecutionArguments(document, variableValues);
   let operation;
@@ -127,7 +131,10 @@ function buildExecutionContext(args) {
     errors: [],
     fieldExecutionHooks,
     subsequentPayloads: /* @__PURE__ */ new Set(),
-    enablePerEventContext: enablePerEventContext != null ? enablePerEventContext : true
+    enablePerEventContext: enablePerEventContext != null ? enablePerEventContext : true,
+    enableEarlyExecution: enableEarlyExecution != null ? enableEarlyExecution : false,
+    enableDeferredMerge: enableDeferredMerge != null ? enableDeferredMerge : false,
+    incrementalPayloadBatchingTimeoutMs
   };
 }
 function buildPerEventExecutionContext(exeContext, payload) {
@@ -176,7 +183,8 @@ function executeOperation(exeContext) {
           rootValue,
           path,
           groupedFieldSet,
-          void 0
+          void 0,
+          exeContext.enableEarlyExecution ? patches : void 0
         );
         result = buildResponse(exeContext, result);
         break;
@@ -186,7 +194,8 @@ function executeOperation(exeContext) {
           rootTypeName,
           rootValue,
           path,
-          groupedFieldSet
+          groupedFieldSet,
+          exeContext.enableEarlyExecution ? patches : void 0
         );
         result = buildResponse(exeContext, result);
         break;
@@ -207,16 +216,18 @@ function executeOperation(exeContext) {
           `Operation "${operation.operation}" is not a part of GraphQL spec`
         );
     }
-    for (const patch of patches) {
-      const { label, groupedFieldSet: patchGroupedFieldSet } = patch;
-      executeDeferredFragment(
-        exeContext,
-        rootTypeName,
-        rootValue,
-        patchGroupedFieldSet,
-        label,
-        path
-      );
+    if (!exeContext.enableEarlyExecution) {
+      for (const patch of patches) {
+        const { label, groupedFieldSet: patchGroupedFieldSet } = patch;
+        executeDeferredFragment(
+          exeContext,
+          rootTypeName,
+          rootValue,
+          patchGroupedFieldSet,
+          label,
+          path
+        );
+      }
     }
     return result;
   } catch (error) {
@@ -224,7 +235,7 @@ function executeOperation(exeContext) {
     return buildResponse(exeContext, null);
   }
 }
-function buildResponse(exeContext, data) {
+function buildResponse(exeContext, data, deferredMergeAwaited = false) {
   var _a, _b;
   if (isPromise(data)) {
     return data.then(
@@ -237,6 +248,16 @@ function buildResponse(exeContext, data) {
   }
   const hooks = exeContext.fieldExecutionHooks;
   try {
+    if (exeContext.enableDeferredMerge && data !== null) {
+      const pendingDeferredFragments = includeCompletedDeferredFragmentsInResult(exeContext, data);
+      const batchTimeout = exeContext.incrementalPayloadBatchingTimeoutMs;
+      if (typeof batchTimeout === "number" && !deferredMergeAwaited && pendingDeferredFragments.length > 0) {
+        return raceIncrementalPayloadBatch(
+          pendingDeferredFragments,
+          batchTimeout
+        ).then(() => buildResponse(exeContext, data, true));
+      }
+    }
     const initialResult = exeContext.errors.length === 0 ? { data } : { errors: exeContext.errors, data };
     if (exeContext.subsequentPayloads.size > 0) {
       return {
@@ -266,7 +287,17 @@ function buildResponse(exeContext, data) {
     return buildResponse(exeContext, null);
   }
 }
-function executeFieldsSerially(exeContext, parentTypeName, sourceValue, path, groupedFieldSet) {
+function executeFieldsSerially(exeContext, parentTypeName, sourceValue, path, groupedFieldSet, patches) {
+  if (exeContext.enableEarlyExecution) {
+    startExecutingPatches(
+      exeContext,
+      parentTypeName,
+      sourceValue,
+      path,
+      patches,
+      void 0
+    );
+  }
   return promiseReduce(
     groupedFieldSet,
     (results, [responseName, fieldGroup]) => {
@@ -297,9 +328,19 @@ function executeFieldsSerially(exeContext, parentTypeName, sourceValue, path, gr
     /* @__PURE__ */ Object.create(null)
   );
 }
-function executeFields(exeContext, parentTypeName, sourceValue, path, groupedFieldSet, incrementalDataRecord) {
+function executeFields(exeContext, parentTypeName, sourceValue, path, groupedFieldSet, incrementalDataRecord, patches) {
   const results = /* @__PURE__ */ Object.create(null);
   let containsPromise = false;
+  if (exeContext.enableEarlyExecution) {
+    startExecutingPatches(
+      exeContext,
+      parentTypeName,
+      sourceValue,
+      path,
+      patches,
+      incrementalDataRecord
+    );
+  }
   try {
     for (const [responseName, fieldGroup] of groupedFieldSet) {
       const fieldPath = addPath(path, responseName, parentTypeName);
@@ -331,6 +372,46 @@ function executeFields(exeContext, parentTypeName, sourceValue, path, groupedFie
   }
   return promiseForObject(results);
 }
+function startExecutingPatches(exeContext, parentTypeName, sourceValue, path, patches, incrementalDataRecord) {
+  if (!patches) {
+    return;
+  }
+  for (const { label, groupedFieldSet: patchGroupedFieldSet } of patches) {
+    executeDeferredFragment(
+      exeContext,
+      parentTypeName,
+      sourceValue,
+      patchGroupedFieldSet,
+      label,
+      path,
+      incrementalDataRecord
+    );
+  }
+}
+function includeCompletedDeferredFragmentsInResult(exeContext, result) {
+  const pendingDeferredFragments = [];
+  for (const incrementalDataRecord of exeContext.subsequentPayloads) {
+    if (!isDeferredFragmentRecord(incrementalDataRecord)) {
+      continue;
+    }
+    if (!incrementalDataRecord.isCompleted) {
+      pendingDeferredFragments.push(incrementalDataRecord.promise);
+      continue;
+    }
+    const target = getObjectAtPath(result, incrementalDataRecord.path);
+    if (!target) {
+      continue;
+    }
+    if (incrementalDataRecord.data != null) {
+      Object.assign(target, incrementalDataRecord.data);
+    }
+    if (incrementalDataRecord.errors.length > 0) {
+      exeContext.errors.push(...incrementalDataRecord.errors);
+    }
+    exeContext.subsequentPayloads.delete(incrementalDataRecord);
+  }
+  return pendingDeferredFragments;
+}
 function executeField(exeContext, parentTypeName, source, fieldGroup, path, incrementalDataRecord) {
   const schemaFragment = exeContext.schemaFragment;
   const fieldName = fieldGroup[0].name.value;
@@ -1308,6 +1389,17 @@ function completeObjectValue(exeContext, returnTypeName, fieldGroup, path, resul
 }
 function collectAndExecuteSubfields(exeContext, returnTypeName, fieldGroup, path, result, incrementalDataRecord) {
   const { groupedFieldSet: subGroupedFieldSet, patches: subPatches } = collectSubfields(exeContext, { name: returnTypeName }, fieldGroup);
+  if (exeContext.enableEarlyExecution) {
+    return executeFields(
+      exeContext,
+      returnTypeName,
+      result,
+      path,
+      subGroupedFieldSet,
+      incrementalDataRecord,
+      subPatches
+    );
+  }
   const subFields = executeFields(
     exeContext,
     returnTypeName,
@@ -1926,6 +2018,34 @@ function getCompletedIncrementalResults(exeContext) {
   }
   return incrementalResults;
 }
+async function getCompletedIncrementalResultsWithBatching(exeContext, batchTimeout) {
+  const incremental = getCompletedIncrementalResults(exeContext);
+  if (incremental.length && exeContext.subsequentPayloads.size > 0) {
+    await raceIncrementalPayloadBatch(
+      Array.from(exeContext.subsequentPayloads).map((p) => p.promise),
+      batchTimeout
+    );
+    incremental.push(...getCompletedIncrementalResults(exeContext));
+  }
+  return incremental;
+}
+function raceIncrementalPayloadBatch(promises, batchTimeout) {
+  if (!Number.isFinite(batchTimeout) || batchTimeout <= 0) {
+    return Promise.resolve();
+  }
+  let timeoutId;
+  const timeoutPromise = new Promise((resolve) => {
+    timeoutId = setTimeout(resolve, batchTimeout);
+  });
+  return Promise.race([
+    Promise.all(promises).then(() => void 0),
+    timeoutPromise
+  ]).finally(() => {
+    if (timeoutId !== void 0) {
+      clearTimeout(timeoutId);
+    }
+  });
+}
 function yieldSubsequentPayloads(exeContext) {
   let isDone = false;
   async function next() {
@@ -1938,7 +2058,11 @@ function yieldSubsequentPayloads(exeContext) {
     if (isDone) {
       return { value: void 0, done: true };
     }
-    const incremental = getCompletedIncrementalResults(exeContext);
+    const batchTimeout = exeContext.incrementalPayloadBatchingTimeoutMs;
+    const incremental = typeof batchTimeout === "number" ? await getCompletedIncrementalResultsWithBatching(
+      exeContext,
+      batchTimeout
+    ) : getCompletedIncrementalResults(exeContext);
     const hasNext = exeContext.subsequentPayloads.size > 0;
     if (!incremental.length && hasNext) {
       return next();
@@ -1981,6 +2105,9 @@ function yieldSubsequentPayloads(exeContext) {
 function isStreamItemsRecord(incrementalDataRecord) {
   return incrementalDataRecord.type === "stream";
 }
+function isDeferredFragmentRecord(incrementalDataRecord) {
+  return incrementalDataRecord.type === "defer";
+}
 var DeferredFragmentRecord = class {
   constructor(opts) {
     this.type = "defer";
@@ -1996,10 +2123,11 @@ var DeferredFragmentRecord = class {
       this._resolve = (promiseOrValue) => {
         resolve(promiseOrValue);
       };
-    }).then((data) => {
-      this.data = data;
-      this.isCompleted = true;
-    });
+    }).then((data) => this.complete(data));
+  }
+  complete(data) {
+    this.data = data;
+    this.isCompleted = true;
   }
   addData(data) {
     var _a, _b, _c;
@@ -2008,6 +2136,9 @@ var DeferredFragmentRecord = class {
       (_b = this._resolve) == null ? void 0 : _b.call(this, parentData.then(() => data));
       return;
     }
+    if (!isPromise(data)) {
+      this.complete(data);
+    }
     (_c = this._resolve) == null ? void 0 : _c.call(this, data);
   }
 };