@apollo/client 4.3.0-alpha.1 → 4.3.0-alpha.11

package/cache/inmemory/readFromStore.js

@@ -1,8 +1,10 @@
+import { equal } from "@wry/equality";
+import { Trie } from "@wry/trie";
 import { Kind } from "graphql";
 import { wrap } from "optimism";
 import { addTypenameToDocument, cacheSizes, canonicalStringify, isReference, } from "@apollo/client/utilities";
 import { __DEV__ } from "@apollo/client/utilities/environment";
-import { DeepMerger, getDefaultValues, getFragmentFromSelection, getMainDefinition, getQueryDefinition, isArray, isField, isNonNullObject, makeReference, maybeDeepFreeze, mergeDeepArray, resultKeyNameFromField, shouldInclude, } from "@apollo/client/utilities/internal";
+import { compact, DeepMerger, getDefaultValues, getDirectiveArgValue, getFragmentFromSelection, getMainDefinition, getQueryDefinition, handleIncrementalSymbol, isArray, isDeferredFragment, isField, isNonNullObject, isStreamField, makeReference, maybeDeepFreeze, mergeDeepArray, resultKeyNameFromField, shouldInclude, } from "@apollo/client/utilities/internal";
 import { invariant, newInvariantError, } from "@apollo/client/utilities/invariant";
 import { MissingFieldError } from "../core/types/common.js";
 import { maybeDependOnExistenceOfEntity, supportsResultCaching, } from "./entityStore.js";
@@ -15,8 +17,11 @@ export class StoreReader {
     executeSelectionSet;
     // cached version of executeSubSelectedArray
     executeSubSelectedArray;
+    prunePartialStreamArray;
+    prunePartialBoundaries;
     config;
     knownResults = new WeakMap();
+    keyMaker = new Trie();
     constructor(config) {
         this.config = config;
         // memoized functions in this class will be "garbage-collected"
@@ -59,47 +64,124 @@ export class StoreReader {
                 }
             },
         });
+        this.prunePartialBoundaries = wrap((options) => this.prunePartialBoundariesImpl(options), {
+            max: cacheSizes["inMemoryCache.prunePartialBoundaries"] ||
+                20000 /* defaultCacheSizes["inMemoryCache.prunePartialBoundaries"] */,
+            makeCacheKey: ({ boundaries, context, selectionSet }) => {
+                if (supportsResultCaching(context.store)) {
+                    return this.keyMaker.lookup(selectionSet, boundaries, context.streamInfo, context.deferInfo);
+                }
+            },
+        });
+        this.prunePartialStreamArray = wrap((options) => {
+            const { field, context, path } = options;
+            if (isStreamField(field, context.variables)) {
+                context.streamInfo?.lookupArray(path).state.depend();
+            }
+            return this.prunePartialStreamArrayImpl(options);
+        }, {
+            max: cacheSizes["inMemoryCache.prunePartialStreamArray"] ||
+                20000 /* defaultCacheSizes["inMemoryCache.prunePartialStreamArray"] */,
+            makeCacheKey: ({ field, context, boundaries }) => {
+                if (supportsResultCaching(context.store)) {
+                    return this.keyMaker.lookup(field, boundaries, context.streamInfo, context.deferInfo);
+                }
+            },
+        });
     }
-    /**
-     * Given a store and a query, return as much of the result as possible and
-     * identify if any data was missing from the store.
-     */
-    diffQueryAgainstStore({ store, query, rootId = "ROOT_QUERY", variables, returnPartialData = true, }) {
+    diffQueryAgainstStore({ store, query, rootId = "ROOT_QUERY", variables, returnPartialData = true, ...options }) {
+        const returnIncremental = Object.hasOwn(options, handleIncrementalSymbol);
         const policies = this.config.cache.policies;
-        variables = {
-            ...getDefaultValues(getQueryDefinition(query)),
-            ...variables,
-        };
+        variables = compact(getDefaultValues(getQueryDefinition(query)), variables);
         const rootRef = makeReference(rootId);
-        const execResult = this.executeSelectionSet({
+        const context = {
+            store,
+            query,
+            policies,
+            variables,
+            varString: canonicalStringify(variables),
+            ...extractFragmentContext(query, this.config.fragments),
+            ...options[handleIncrementalSymbol],
+        };
+        let execResult = this.executeSelectionSet({
             selectionSet: getMainDefinition(query).selectionSet,
             objectOrReference: rootRef,
             enclosingRef: rootRef,
-            context: {
-                store,
-                query,
-                policies,
-                variables,
-                varString: canonicalStringify(variables),
-                ...extractFragmentContext(query, this.config.fragments),
-            },
+            context,
         });
-        let missing;
-        if (execResult.missing) {
-            missing = new MissingFieldError(firstMissing(execResult.missing), execResult.missing, query, variables);
+        // Since executeSelectionSet doesn't know about returnPartialData, we need
+        // to perform a 2nd pass over the result to prune any fields inside
+        // partial defer boundaries. The "deferPartial" data state tells us that the
+        // only part of the result that contributed to its partiality is data inside
+        // a defer boundary.
+        if (returnIncremental && shouldPrune(execResult, context)) {
+            const pruned = this.prunePartialBoundaries({
+                selectionSet: getMainDefinition(query).selectionSet,
+                data: execResult.result,
+                boundaries: execResult.partialBoundaries,
+                context,
+                path: [],
+            });
+            const changed = execResult.result !== pruned.result;
+            // It's possible that pruning didn't actually change the result which can
+            // happen if a defer boundary is misclassified as "deferPartial" instead
+            // of "streaming" (sibling defer boundaries with overlapping selection
+            // sets, one of which is complete). In this case, pruning corrects the
+            // dataState to streaming instead of leaving it as partial. If we tolerate
+            // partial results and pruning changed the result by dropping fields, then
+            // we want to keep the original execResult which contains the partial
+            // data.
+            if (!changed || !returnPartialData) {
+                // Omit `missing` property since pruning puts it in a state that doesn't
+                // report missing fields.
+                execResult = {
+                    result: pruned.result,
+                    partialBoundaries: execResult.partialBoundaries,
+                    dataState: pruned.dataState,
+                };
+            }
         }
-        const complete = !missing;
-        const { result } = execResult;
-        return {
-            result: complete ? result
-                : returnPartialData ?
-                    Object.keys(result).length === 0 ?
-                        null
-                        : result
-                    : null,
+        let { result, dataState, missing } = execResult;
+        // Evaluate this condition before we start mucking with dataState for the
+        // publicly returned value
+        const includeMissing = !!missing &&
+            // We don't need to report missing fields inside defer boundaries since
+            // the "streaming" dataState tells us that the only missing fields in
+            // the object is inside a defer boundary.
+            (dataState !== "streaming" || !returnIncremental);
+        // If we get all root @defer boundaries with an empty result, report it as
+        // empty instead of streaming.
+        if (dataState === "streaming" && Object.keys(result).length === 0) {
+            dataState = "empty";
+        }
+        let missingError;
+        if (dataState === "deferPartial" ||
+            dataState === "streamPartial" ||
+            (dataState === "streaming" && !returnIncremental)) {
+            dataState = "partial";
+        }
+        if (dataState === "partial" && !returnPartialData) {
+            dataState = "empty";
+        }
+        const complete = dataState === "complete";
+        const keepResult = complete ||
+            dataState === "streaming" ||
+            (returnPartialData && Object.keys(result).length);
+        const diffResult = {
+            result: keepResult ? result : null,
             complete,
-            missing,
+            get missing() {
+                if (includeMissing) {
+                    missingError ||= new MissingFieldError(firstMissing(missing), missing, query, variables);
+                }
+                return missingError;
+            },
         };
+        if (returnIncremental) {
+            diffResult.dataState =
+                dataState;
+        }
+        return diffResult;
     }
     isFresh(result, parent, selectionSet, context) {
         if (supportsResultCaching(context.store) &&
@@ -118,14 +200,18 @@ export class StoreReader {
             !context.store.has(objectOrReference.__ref)) {
             return {
                 result: {},
+                dataState: "empty",
                 missing: `Dangling reference to missing ${objectOrReference.__ref} object`,
+                partialBoundaries: new PartialBoundaries(),
             };
         }
         const { variables, policies, store } = context;
         const typename = store.getFieldValue(objectOrReference, "__typename");
         const objectsToMerge = [];
+        let dataState;
         let missing;
         const missingMerger = new DeepMerger();
+        const partialBoundaries = new PartialBoundaries();
         if (typeof typename === "string" && !policies.rootIdsByTypename[typename]) {
             // Ensure we always include a default value for the __typename
             // field, if we have one. Note that this field can be overridden by other
@@ -138,7 +224,7 @@ export class StoreReader {
                     [resultName]: result.missing,
                 });
             }
-            return result.result;
+            return result;
         }
         const workSet = new Set(selectionSet.selections);
         workSet.forEach((selection) => {
@@ -156,36 +242,75 @@ export class StoreReader {
                 const resultName = resultKeyNameFromField(selection);
                 if (fieldValue === void 0) {
                     if (!addTypenameToDocument.added(selection)) {
+                        const id = isReference(objectOrReference) ? objectOrReference.__ref
+                            : objectOrReference ? policies.identify(objectOrReference)[0]
+                                : undefined;
                         missing = missingMerger.merge(missing, {
-                            [resultName]: `Can't find field '${selection.name.value}' on ${isReference(objectOrReference) ?
-                                objectOrReference.__ref + " object"
-                                : "object " + JSON.stringify(objectOrReference, null, 2)}`,
+                            [resultName]: `Can't find field '${selection.name.value}' on ${id ?
+                                `${id} object`
+                                : `object ${JSON.stringify(objectOrReference || {}, null, 2)}`}`,
                         });
+                        dataState = mergeDataState(dataState, "empty");
                     }
                 }
                 else if (isArray(fieldValue)) {
                     if (fieldValue.length > 0) {
-                        fieldValue = handleMissing(this.executeSubSelectedArray({
+                        const execResult = handleMissing(this.executeSubSelectedArray({
                             field: selection,
                             array: fieldValue,
                             enclosingRef,
                             context,
                         }), resultName);
+                        fieldValue = execResult.result;
+                        dataState = mergeDataState(dataState, execResult.dataState);
+                        partialBoundaries.set(resultName, execResult.partialBoundaries);
+                    }
+                    else {
+                        dataState = mergeDataState(dataState, "complete");
                     }
                 }
                 else if (!selection.selectionSet) {
-                    // do nothing
+                    // Auto-inserted __typename should not affect dataState (see empty
+                    // @defer boundaries, which must stay "empty" → parent "streaming").
+                    if (!addTypenameToDocument.added(selection)) {
+                        dataState = mergeDataState(dataState, "complete");
+                    }
                 }
                 else if (fieldValue != null) {
+                    if (__DEV__) {
+                        const fieldName = selection.name.value;
+                        if (typename) {
+                            const policy = policies["getFieldPolicy"](typename, fieldName);
+                            if (policy?.scalar) {
+                                __DEV__ && invariant.warn(116, `${typename}.${fieldName}`, policy.scalar);
+                            }
+                        }
+                    }
                     // In this case, because we know the field has a selection set,
                     // it must be trying to query a GraphQLObjectType, which is why
                     // fieldValue must be != null.
-                    fieldValue = handleMissing(this.executeSelectionSet({
+                    const execResult = handleMissing(this.executeSelectionSet({
                         selectionSet: selection.selectionSet,
                         objectOrReference: fieldValue,
                         enclosingRef: isReference(fieldValue) ? fieldValue : enclosingRef,
                         context,
                     }), resultName);
+                    fieldValue = execResult.result;
+                    partialBoundaries.set(resultName, execResult.partialBoundaries);
+                    // If the object's fields resolved to an "empty" dataState (e.g. no
+                    // field resolved with a non-undefined value), but the fieldValue
+                    // object itself is present, this object should be considered
+                    // partial instead of empty. This also ensures defer boundaries that
+                    // select this object remain as partial defer boundaries rather than
+                    // mistakenly get reported as streaming. This is especially necessary
+                    // when combined with GraphQL Codegen which generates its type and
+                    // relies on the outer object to be absent when its fields haven't
+                    // streamed in. Reporting the defer boundary as "streaming" instead of
+                    // "partial" would otherwise have the potential to cause runtime
+                    // crashes since the runtime values and types would not line up
+                    // properly (types expect object to be undefined, but its instead
+                    // present without its fields)
+                    dataState = mergeDataState(dataState, execResult.dataState === "empty" ? "partial" : execResult.dataState);
                 }
                 if (fieldValue !== void 0) {
                     objectsToMerge.push({ [resultName]: fieldValue });
@@ -194,15 +319,60 @@ export class StoreReader {
             else {
                 const fragment = getFragmentFromSelection(selection, context.lookupFragment);
                 if (!fragment && selection.kind === Kind.FRAGMENT_SPREAD) {
-                    throw newInvariantError(112, selection.name.value);
+                    throw newInvariantError(117, selection.name.value);
                 }
                 if (fragment && policies.fragmentMatches(fragment, typename)) {
-                    fragment.selectionSet.selections.forEach(workSet.add, workSet);
+                    const isDeferBoundary = isDeferredFragment(selection, context.variables);
+                    // Prior to 4.3, this branch just flattened the fragment's
+                    // selectionSet into the existing workSet so that it continued
+                    // iterating as if the fragment didn't exist. The cache is
+                    // incremental aware as of 4.3 and as such, we need to resolve the
+                    // per-fragment selection set so that we can properly strip partial
+                    // defer fragment data when returnPartialData is false. We need to
+                    // call execSelectionSetImpl directly (non-cached version) so that we
+                    // scope the dataState correctly for its fields. Using the cached
+                    // executeSelectionSet can result in cache poisoning when combined
+                    // with the fragment registry where it might cache either a) an error
+                    // thrown when a registered fragment references a named fragment that
+                    // the query is expected to supply and doesn't or b) resolve to the
+                    // wrong data result when combined with queries that provide different
+                    // implementations of the same fragment (see inmemory/fragmentRegistry and
+                    // cache.diff/incremental tests which provide guards against this
+                    // behavior).
+                    const execResult = this.execSelectionSetImpl({
+                        selectionSet: fragment.selectionSet,
+                        objectOrReference,
+                        enclosingRef,
+                        context,
+                    });
+                    const { result, dataState: nextDataState } = execResult;
+                    partialBoundaries.merge(execResult.partialBoundaries);
+                    if (result !== void 0) {
+                        objectsToMerge.push(result);
+                    }
+                    if (execResult.missing) {
+                        missing = missingMerger.merge(missing, execResult.missing);
+                    }
+                    if (isDeferBoundary &&
+                        (nextDataState === "partial" || nextDataState === "empty")) {
+                        partialBoundaries.add(selection);
+                    }
+                    dataState = mergeDataState(dataState, isDeferBoundary ?
+                        nextDataState === "empty" ? "streaming"
+                            : nextDataState === "partial" ? "deferPartial"
+                                : nextDataState
+                        : nextDataState);
                 }
             }
         });
+        dataState ||= "complete";
         const result = mergeDeepArray(objectsToMerge);
-        const finalResult = { result, missing };
+        const finalResult = {
+            result,
+            missing,
+            dataState,
+            partialBoundaries,
+        };
         const frozen = maybeDeepFreeze(finalResult);
         // Store this result with its selection set so that we can quickly
         // recognize it again in the StoreReader#isFresh method.
@@ -213,8 +383,11 @@ export class StoreReader {
     }
     // Uncached version of executeSubSelectedArray.
     execSubSelectedArrayImpl({ field, array, enclosingRef, context, }) {
+        let dataState = "complete";
         let missing;
         let missingMerger = new DeepMerger();
+        const partialBoundaries = new PartialBoundaries();
+        const isStreamed = isStreamField(field, context.variables);
         function handleMissing(childResult, i) {
             if (childResult.missing) {
                 missing = missingMerger.merge(missing, { [i]: childResult.missing });
@@ -229,23 +402,36 @@ export class StoreReader {
             if (item === null) {
                 return null;
             }
+            let execResult;
             // This is a nested array, recurse
             if (isArray(item)) {
-                return handleMissing(this.executeSubSelectedArray({
+                execResult = this.executeSubSelectedArray({
                     field,
                     array: item,
                     enclosingRef,
                     context,
-                }), i);
+                });
             }
-            // This is an object, run the selection set on it
-            if (field.selectionSet) {
-                return handleMissing(this.executeSelectionSet({
+            else if (field.selectionSet) {
+                execResult = this.executeSelectionSet({
                     selectionSet: field.selectionSet,
                     objectOrReference: item,
                     enclosingRef: isReference(item) ? item : enclosingRef,
                     context,
-                }), i);
+                });
+            }
+            if (execResult) {
+                const { dataState: nextDataState } = execResult;
+                partialBoundaries.set(i, nextDataState === "partial" ?
+                    // avoid mutating the execResult partialBoundaries object
+                    execResult.partialBoundaries.clone().add(field)
+                    : execResult.partialBoundaries);
+                dataState = mergeDataState(dataState, isStreamed ?
+                    nextDataState === "partial" ?
+                        "streamPartial"
+                        : nextDataState
+                    : nextDataState);
+                return handleMissing(execResult, i);
             }
             if (__DEV__) {
                 assertSelectionSetForIdValue(context.store, field, item);
@@ -254,9 +440,162 @@ export class StoreReader {
         });
         return {
             result: array,
+            dataState,
             missing,
+            partialBoundaries,
         };
     }
+    prunePartialBoundariesImpl({ boundaries, context, data, path, selectionSet, }) {
+        const { variables, lookupFragment, policies } = context;
+        if (data == null || !boundaries) {
+            return { result: data, dataState: "complete" };
+        }
+        const merger = new DeepMerger();
+        let changed = false;
+        let dataState = "complete";
+        const result = {};
+        // __typename might not be part of the selection set, so preserve it when
+        // available, otherwise it gets removed since it's never visited when
+        // iterating the selection set.
+        if (Object.hasOwn(data, "__typename")) {
+            result.__typename = data.__typename;
+        }
+        const workSet = new Set(selectionSet.selections);
+        workSet.forEach((selection) => {
+            if (!shouldInclude(selection, variables))
+                return;
+            if (isField(selection)) {
+                const resultName = resultKeyNameFromField(selection);
+                if (!Object.hasOwn(data, resultName)) {
+                    return;
+                }
+                const fieldValue = data[resultName];
+                if (Array.isArray(fieldValue)) {
+                    const pruned = this.prunePartialStreamArray({
+                        field: selection,
+                        array: fieldValue,
+                        boundaries: boundaries.getChild(resultName),
+                        context,
+                        path: path.concat(resultName),
+                    });
+                    changed ||= pruned.result !== fieldValue;
+                    result[resultName] = pruned.result;
+                    dataState = mergeDataState(dataState, pruned.dataState);
+                }
+                else if (!selection.selectionSet) {
+                    result[resultName] = fieldValue;
+                }
+                else {
+                    const pruned = this.prunePartialBoundaries({
+                        data: fieldValue,
+                        selectionSet: selection.selectionSet,
+                        boundaries: boundaries.getChild(resultName),
+                        context,
+                        path: path.concat(resultName),
+                    });
+                    changed ||= pruned.result !== fieldValue;
+                    dataState = mergeDataState(dataState, pruned.dataState);
+                    // A response key can be selected by more than one selection (e.g. a
+                    // field and an overlapping fragment), so merge their kept fields.
+                    result[resultName] =
+                        Object.hasOwn(result, resultName) ?
+                            merger.merge(result[resultName], pruned.result)
+                            : pruned.result;
+                }
+                return;
+            }
+            // Note: we do NOT set `changed` to true anywhere in this branch of the
+            // conditional, despite the fact that we might have encountered a
+            // partial @defer boundary. Dropping a fragment does not guarantee keys
+            // are actually dropped which can happen when overlapping sibling
+            // selections contribute to the construction of the object. The final
+            // Object.keys(result).length check actually detects whether keys were
+            // dropped or not.
+            const fragment = getFragmentFromSelection(selection, lookupFragment);
+            let prune = false;
+            if (context.deferInfo && isDeferredFragment(selection, variables)) {
+                const directive = selection.directives?.find((d) => d.name.value === "defer");
+                const label = directive && getDirectiveArgValue(directive, "label", Kind.STRING);
+                prune = !!context.deferInfo.peekArray(path.concat(label || []));
+            }
+            if (fragment && policies.fragmentMatches(fragment, data.__typename)) {
+                if (boundaries.has(selection) || prune) {
+                    dataState = mergeDataState(dataState, "streaming");
+                }
+                else {
+                    fragment.selectionSet.selections.forEach(workSet.add, workSet);
+                }
+            }
+        });
+        if (Object.keys(result).length !== Object.keys(data).length) {
+            changed = true;
+        }
+        else if (changed && boundaries.hasSelections()) {
+            // Overlapping siblings may rebuild the same fields under a new object
+            // identity (e.g. changed === true) after a partial @defer is skipped.
+            // We perform a deep equality check to verify whether anything was
+            // actually dropped by the partial @defer fragment.
+            changed = !equal(result, data);
+        }
+        return { result: changed ? result : data, dataState };
+    }
+    prunePartialStreamArrayImpl({ field, array, boundaries, context, path, }) {
+        if (!boundaries)
+            return { result: array, dataState: "complete" };
+        let changed = false;
+        let dataState = "complete";
+        let pruned = [];
+        const state = context.streamInfo?.peekArray(path)?.state;
+        const length = Math.min(array.length, state?.truncate ? state.streamPosition : Number.MAX_SAFE_INTEGER);
+        for (let i = 0; i < length; i++) {
+            const item = array[i];
+            let prunedResult = {
+                result: item,
+                dataState: "complete",
+            };
+            const boundary = boundaries.getChild(i);
+            if (boundary?.has(field)) {
+                // The presence of streamInfo determines how we truncate partial
+                // stream arrays. Stream info is only given to cache.diff during
+                // in-flight requests so we want keep items in the array equal to the
+                // total that have streamed in (this is represented by streamPosition
+                // above). For all other cache reads, partial stream boundaries are
+                // pruned back to an empty array.
+                if (state) {
+                    state.truncate = true;
+                    pruned = pruned.slice(0, state.streamPosition);
+                }
+                else {
+                    pruned = [];
+                    dataState = "complete";
+                }
+                break;
+            }
+            if (Array.isArray(item)) {
+                prunedResult = this.prunePartialStreamArray({
+                    field,
+                    array: item,
+                    boundaries: boundaries.getChild(i),
+                    context,
+                    path: path.concat(i),
+                });
+            }
+            else if (field.selectionSet) {
+                prunedResult = this.prunePartialBoundaries({
+                    data: item,
+                    selectionSet: field.selectionSet,
+                    boundaries: boundaries.getChild(i),
+                    context,
+                    path: path.concat(i),
+                });
+            }
+            pruned.push(prunedResult.result);
+            changed ||= prunedResult.result !== item;
+            dataState = mergeDataState(dataState, prunedResult.dataState);
+        }
+        changed ||= pruned.length !== array.length;
+        return { result: changed ? pruned : array, dataState };
+    }
 }
 function firstMissing(tree) {
     try {
@@ -277,7 +616,7 @@ function assertSelectionSetForIdValue(store, field, fieldValue) {
             if (isNonNullObject(value)) {
                 invariant(
                     !isReference(value),
-                    113,
+                    118,
                     getTypenameFromStoreObject(store, value),
                     field.name.value
                 );
@@ -286,4 +625,102 @@ function assertSelectionSetForIdValue(store, field, fieldValue) {
         });
     }
 }
+// We deliberately leave `returnPartialData` out of `executeSelectionSet`'s
+// cache key. `isFresh` runs during writes and cannot provide a reliable value
+// for this option, so including it would prevent a reliable cache hit.
+//
+// When `returnPartialData` is false, `diffQueryAgainstStore` prunes data from
+// partial @defer boundaries after reading the cached result. `PartialBoundaries`
+// records the selection paths needed for that pass, including empty nodes along
+// a path. Overlapping non-deferred selections must still be rebuilt so fields
+// contributed only by a partial deferred sibling are removed. The prune pass
+// can then skip unrelated result branches.
+class PartialBoundaries {
+    selections = new Set();
+    children = new Map();
+    add(selection) {
+        this.selections.add(selection);
+        return this;
+    }
+    has(selection) {
+        return this.selections.has(selection);
+    }
+    hasSelections() {
+        return this.selections.size > 0;
+    }
+    getChild(key) {
+        return this.children.get(key);
+    }
+    clone() {
+        return new PartialBoundaries().merge(this);
+    }
+    set(key, boundary) {
+        const child = this.getChild(key);
+        this.children.set(key, 
+        // Create a new PartialBoundaries instance to avoid mutating any cached
+        // execResult partialBoundaries objects
+        child ? child.clone().merge(boundary) : boundary);
+    }
+    merge(boundaries) {
+        boundaries.selections.forEach((selection) => this.add(selection));
+        boundaries.children.forEach((child, key) => this.set(key, child));
+        return this;
+    }
+}
+function shouldPrune({ dataState }, context) {
+    if (dataState === "deferPartial" || dataState === "streamPartial") {
+        return true;
+    }
+    if (dataState === "complete" || dataState === "streaming") {
+        return !!(
+        // If the last cache write repaired a partial @stream array to a
+        // complete array, the stream array might contain stale entries after
+        // the last written value. We only want to deliver the results up to
+        // the index the network wrote so we need to prune it too.
+        (context.streamInfo ||
+            // The network hasn't delivered these @defer boundaries yet, so prune
+            // the (possibly complete) cached data sitting at them.
+            context.deferInfo));
+    }
+    return false;
+}
+// Describes the data state transitions that change the running state when it's
+// combined with the next data state. Omitted object values represent
+// "impossible" merges where the data state should remain the same.
+const DATA_STATE_MERGES = {
+    empty: {
+        complete: "partial",
+        deferPartial: "partial",
+        streaming: "partial",
+        streamPartial: "partial",
+    },
+    deferPartial: {
+        empty: "partial",
+    },
+    streamPartial: {
+        deferPartial: "deferPartial",
+        empty: "partial",
+    },
+    streaming: {
+        deferPartial: "deferPartial",
+        empty: "partial",
+        streamPartial: "streamPartial",
+    },
+    complete: {
+        deferPartial: "deferPartial",
+        streaming: "streaming",
+        streamPartial: "streamPartial",
+        empty: "partial",
+    },
+    partial: {},
+};
+function mergeDataState(current, next) {
+    if (next === "partial") {
+        return "partial";
+    }
+    if (!current || current === next) {
+        return next;
+    }
+    return DATA_STATE_MERGES[current][next] || current;
+}
 //# sourceMappingURL=readFromStore.js.map