@ricsam/formula-engine 0.2.9 → 0.2.10

package/dist/cjs/core/managers/dependency-manager.cjs

@@ -74,6 +74,49 @@ class DependencyManager {
     this.cacheManager = cacheManager;
     this.workbookManager = workbookManager;
   }
+  getSnapshotEligibilityState(node) {
+    if (node instanceof import_resource_dependency_node.ResourceDependencyNode) {
+      return {
+        eligible: true,
+        reason: "resource nodes are always snapshot eligible"
+      };
+    }
+    if (node instanceof import_range_evaluation_node.RangeEvaluationNode) {
+      return node.result.type === "awaiting-evaluation" ? {
+        eligible: false,
+        reason: "range result is still awaiting evaluation"
+      } : {
+        eligible: true,
+        reason: `range result type is ${node.result.type}`
+      };
+    }
+    return node.evaluationResult.type === "awaiting-evaluation" ? {
+      eligible: false,
+      reason: "evaluation result is still awaiting evaluation"
+    } : {
+      eligible: true,
+      reason: `evaluation result type is ${node.evaluationResult.type}`
+    };
+  }
+  logSnapshotRestoreWarning(message, payload) {
+    console.warn("[FormulaEngine snapshot]", message, payload);
+  }
+  getLogicalNodeSnapshotId(node) {
+    if (node instanceof import_ast_evaluation_node.AstEvaluationNode) {
+      return import_engine_snapshot.getAstNodeSnapshotId(node);
+    }
+    return node.key;
+  }
+  resetRestoredNodeToColdState(node) {
+    this.unregisterNode(node);
+    if (node instanceof import_resource_dependency_node.ResourceDependencyNode) {
+      return;
+    }
+    if (node instanceof import_cell_value_node.CellValueNode) {
+      node.clearSpillMetaNode();
+    }
+    node.invalidate();
+  }
   getSheetWatcherKey(address) {
     return `${address.workbookName}:${address.sheetName}`;
   }
@@ -381,13 +424,7 @@ class DependencyManager {
   }
   toSnapshot(evaluationManager) {
     const isNodeSnapshotEligible = (node) => {
-      if (node instanceof import_resource_dependency_node.ResourceDependencyNode) {
-        return true;
-      }
-      if (node instanceof import_range_evaluation_node.RangeEvaluationNode) {
-        return node.result.type !== "awaiting-evaluation";
-      }
-      return node.evaluationResult.type !== "awaiting-evaluation";
+      return this.getSnapshotEligibilityState(node).eligible;
     };
     const astNodes = new Set;
     for (const astEntries of this.asts.values()) {
@@ -425,15 +462,17 @@ class DependencyManager {
     }
     const resolvedNodes = Array.from(resolvedNodesSet);
     const allNodeSnapshotIds = new Map;
+    const availableSnapshotIds = new Set;
     for (const node of resolvedNodes) {
-      if (node instanceof import_ast_evaluation_node.AstEvaluationNode) {
-        allNodeSnapshotIds.set(node, import_engine_snapshot.getAstNodeSnapshotId(node));
-      } else {
-        allNodeSnapshotIds.set(node, node.key);
-      }
-    }
-    const getAllNodeSnapshotId = (node) => {
-      const snapshotId = allNodeSnapshotIds.get(node);
+      const snapshotId = this.getLogicalNodeSnapshotId(node);
+      allNodeSnapshotIds.set(node, snapshotId);
+      availableSnapshotIds.add(snapshotId);
+    }
+    const getAllNodeSnapshotId = (node, _context) => {
+      const snapshotId = allNodeSnapshotIds.get(node) ?? (() => {
+        const logicalSnapshotId = this.getLogicalNodeSnapshotId(node);
+        return availableSnapshotIds.has(logicalSnapshotId) ? logicalSnapshotId : undefined;
+      })();
       if (!snapshotId) {
         throw new Error(`Missing snapshot id for dependency node ${node.key}`);
       }
@@ -479,9 +518,25 @@ class DependencyManager {
   restoreFromSnapshot(snapshots, evaluationManager) {
     this.clearEvaluationCache();
     const nodesBySnapshotId = new Map;
+    let hadRestoreFailures = false;
     for (const snapshot of snapshots.dependency.nodes) {
-      const node = this.createNodeFromSnapshot(snapshot);
-      nodesBySnapshotId.set(snapshot.snapshotId, node);
+      try {
+        const node = this.createNodeFromSnapshot(snapshot);
+        nodesBySnapshotId.set(snapshot.snapshotId, node);
+      } catch (error) {
+        hadRestoreFailures = true;
+        this.logSnapshotRestoreWarning("Failed to create warm snapshot node; discarding warm dependency state", {
+          phase: "create-node",
+          snapshotId: snapshot.snapshotId,
+          kind: snapshot.kind,
+          key: snapshot.key,
+          error: error instanceof Error ? { name: error.name, message: error.message } : { message: String(error) }
+        });
+      }
+    }
+    if (hadRestoreFailures) {
+      this.clearEvaluationCache();
+      return;
     }
     const resolveNodeSnapshotId = (snapshotId) => nodesBySnapshotId.get(snapshotId);
     const resolveRequiredNodeSnapshotId = (snapshotId) => {
@@ -492,54 +547,71 @@ class DependencyManager {
       return node;
     };
     for (const snapshot of snapshots.dependency.nodes) {
-      const dependencies = new Set(snapshot.dependencies.map(resolveNodeSnapshotId).filter((node2) => node2 !== undefined));
-      if (snapshot.kind === "cell-value") {
-        const node2 = nodesBySnapshotId.get(snapshot.snapshotId);
-        node2.restoreResolvedSnapshot({
-          dependencies,
-          evaluationResult: evaluationManager.deserializeSingleEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
-        });
-        if (snapshot.spillMetaSnapshotId) {
-          const spillMetaNode = resolveNodeSnapshotId(snapshot.spillMetaSnapshotId);
-          if (spillMetaNode instanceof import_spill_meta_node.SpillMetaNode) {
-            node2.setSpillMetaNode(spillMetaNode);
-          }
-        }
-        continue;
-      }
-      if (snapshot.kind === "spill-meta") {
-        const node2 = nodesBySnapshotId.get(snapshot.snapshotId);
-        node2.restoreResolvedSnapshot({
-          dependencies,
-          evaluationResult: evaluationManager.deserializeSpillMetaEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
-        });
+      const targetNode = nodesBySnapshotId.get(snapshot.snapshotId);
+      if (!targetNode) {
+        hadRestoreFailures = true;
         continue;
       }
-      if (snapshot.kind === "empty") {
-        const node2 = nodesBySnapshotId.get(snapshot.snapshotId);
-        node2.restoreResolvedSnapshot({
+      const dependencies = new Set(snapshot.dependencies.map(resolveNodeSnapshotId).filter((node) => node !== undefined));
+      try {
+        if (snapshot.kind === "cell-value") {
+          const node2 = targetNode;
+          node2.restoreResolvedSnapshot({
+            dependencies,
+            evaluationResult: evaluationManager.deserializeSingleEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
+          });
+          if (snapshot.spillMetaSnapshotId) {
+            const spillMetaNode = resolveNodeSnapshotId(snapshot.spillMetaSnapshotId);
+            if (spillMetaNode instanceof import_spill_meta_node.SpillMetaNode) {
+              node2.setSpillMetaNode(spillMetaNode);
+            }
+          }
+          continue;
+        }
+        if (snapshot.kind === "spill-meta") {
+          const node2 = targetNode;
+          node2.restoreResolvedSnapshot({
+            dependencies,
+            evaluationResult: evaluationManager.deserializeSpillMetaEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
+          });
+          continue;
+        }
+        if (snapshot.kind === "empty") {
+          const node2 = targetNode;
+          node2.restoreResolvedSnapshot({
+            dependencies,
+            evaluationResult: evaluationManager.deserializeSingleEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
+          });
+          continue;
+        }
+        if (snapshot.kind === "range") {
+          const node2 = targetNode;
+          node2.restoreResolvedSnapshot({
+            dependencies,
+            result: evaluationManager.deserializeEvaluateAllCellsResultSnapshot(snapshot.result, resolveRequiredNodeSnapshotId)
+          });
+          continue;
+        }
+        if (snapshot.kind === "resource") {
+          continue;
+        }
+        const node = targetNode;
+        node.restoreResolvedSnapshot({
           dependencies,
-          evaluationResult: evaluationManager.deserializeSingleEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
+          evaluationResult: evaluationManager.deserializeFunctionEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
         });
-        continue;
-      }
-      if (snapshot.kind === "range") {
-        const node2 = nodesBySnapshotId.get(snapshot.snapshotId);
-        node2.restoreResolvedSnapshot({
-          dependencies,
-          result: evaluationManager.deserializeEvaluateAllCellsResultSnapshot(snapshot.result, resolveRequiredNodeSnapshotId)
+        this.saveAstNode(node, snapshot.contextDependency);
+      } catch (error) {
+        hadRestoreFailures = true;
+        this.resetRestoredNodeToColdState(targetNode);
+        this.logSnapshotRestoreWarning("Failed to deserialize warm snapshot node; discarding warm dependency state", {
+          phase: "restore-node",
+          snapshotId: snapshot.snapshotId,
+          kind: snapshot.kind,
+          key: snapshot.key,
+          error: error instanceof Error ? { name: error.name, message: error.message } : { message: String(error) }
         });
-        continue;
       }
-      if (snapshot.kind === "resource") {
-        continue;
-      }
-      const node = nodesBySnapshotId.get(snapshot.snapshotId);
-      node.restoreResolvedSnapshot({
-        dependencies,
-        evaluationResult: evaluationManager.deserializeFunctionEvaluationResultSnapshot(snapshot.evaluationResult, resolveRequiredNodeSnapshotId)
-      });
-      this.saveAstNode(node, snapshot.contextDependency);
     }
     this._spilledValues = new Map(snapshots.dependency.spilledValues);
     this.rebuildRuntimeIndexes();
@@ -547,20 +619,32 @@ class DependencyManager {
   }
   serializeNodeSnapshot(node, evaluationManager, getNodeSnapshotId, allNodeSnapshotIds) {
     const snapshotId = getNodeSnapshotId(node);
+    const getKnownSnapshotId = (dependency) => {
+      const directSnapshotId = allNodeSnapshotIds.get(dependency);
+      if (directSnapshotId) {
+        return directSnapshotId;
+      }
+      const logicalSnapshotId = this.getLogicalNodeSnapshotId(dependency);
+      return allNodeSnapshotIds.values().some((knownSnapshotId) => knownSnapshotId === logicalSnapshotId) ? logicalSnapshotId : undefined;
+    };
     const dependencies = [];
     for (const dependency of node.getDependencies()) {
-      const dependencySnapshotId = allNodeSnapshotIds.get(dependency);
+      const dependencySnapshotId = getKnownSnapshotId(dependency);
       if (!dependencySnapshotId) {
         return;
       }
       dependencies.push(dependencySnapshotId);
     }
+    const getReferencedNodeSnapshotId = (dependency) => getNodeSnapshotId(dependency, {
+      sourceNode: node,
+      relation: "evaluation result errAddress"
+    });
     if (node instanceof import_cell_value_node.CellValueNode) {
-      const spillMetaSnapshotId = node.spillMeta ? allNodeSnapshotIds.get(node.spillMeta) : undefined;
+      const spillMetaSnapshotId = node.spillMeta ? getKnownSnapshotId(node.spillMeta) : undefined;
       if (node.spillMeta && !spillMetaSnapshotId) {
         return;
       }
-      const evaluationResult2 = evaluationManager.serializeSingleEvaluationResultSnapshot(node.evaluationResult, getNodeSnapshotId);
+      const evaluationResult2 = evaluationManager.serializeSingleEvaluationResultSnapshot(node.evaluationResult, getReferencedNodeSnapshotId);
       if (!evaluationResult2) {
         return;
       }
@@ -576,7 +660,7 @@ class DependencyManager {
     if (node instanceof import_spill_meta_node.SpillMetaNode) {
       const evaluationResult2 = evaluationManager.serializeSpillMetaEvaluationResultSnapshot(node.evaluationResult, {
         sourceNode: node,
-        getNodeSnapshotId
+        getNodeSnapshotId: getReferencedNodeSnapshotId
       });
       if (!evaluationResult2) {
         return;
@@ -590,7 +674,7 @@ class DependencyManager {
       };
     }
     if (node instanceof import_empty_cell_evaluation_node.EmptyCellEvaluationNode) {
-      const evaluationResult2 = evaluationManager.serializeSingleEvaluationResultSnapshot(node.evaluationResult, getNodeSnapshotId);
+      const evaluationResult2 = evaluationManager.serializeSingleEvaluationResultSnapshot(node.evaluationResult, getReferencedNodeSnapshotId);
       if (!evaluationResult2) {
         return;
       }
@@ -603,7 +687,7 @@ class DependencyManager {
       };
     }
     if (node instanceof import_range_evaluation_node.RangeEvaluationNode) {
-      const result = evaluationManager.serializeEvaluateAllCellsResultSnapshot(node.result, getNodeSnapshotId);
+      const result = evaluationManager.serializeEvaluateAllCellsResultSnapshot(node.result, getReferencedNodeSnapshotId);
       if (!result) {
         return;
       }
@@ -625,7 +709,7 @@ class DependencyManager {
     }
     const evaluationResult = evaluationManager.serializeFunctionEvaluationResultSnapshot(node.evaluationResult, {
       sourceNode: node,
-      getNodeSnapshotId
+      getNodeSnapshotId: getReferencedNodeSnapshotId
     });
     if (!evaluationResult) {
       return;
@@ -1451,4 +1535,4 @@ class DependencyManager {
   }
 }
 
-//# debugId=1436FDFA1813AF4B64756E2164756E21
+//# debugId=1B8CBFB24400CA2564756E2164756E21