@bian-womp/spark-graph 0.2.67 → 0.2.69

package/lib/cjs/index.cjs

@@ -770,11 +770,11 @@ class GraphRuntime {
                     await node.runtime.onInputsChanged?.(capturedInputs, ctx);
                 }
                 catch (err) {
-                    // Suppress errors caused by expected cancellations (switch)
+                    // Suppress errors caused by expected cancellations (switch, snapshot)
                     if (controller.signal.aborted) {
                         const reason = controller.signal.reason;
-                        if (reason === "switch") {
-                            return; // ignore switched runs
+                        if (reason === "switch" || reason === "snapshot") {
+                            return; // ignore switched/snapshot-cancelled runs
                         }
                     }
                     hadError = true;
@@ -1345,6 +1345,47 @@ class GraphRuntime {
     scheduleInputsChanged(nodeId) {
         this.scheduleInputsChangedInternal(nodeId);
     }
+    cancelNodeRuns(nodeIds) {
+        if (nodeIds.length === 0)
+            return;
+        const toCancel = new Set(nodeIds);
+        const visited = new Set();
+        const queue = [...nodeIds];
+        for (let i = 0; i < queue.length; i++) {
+            const nodeId = queue[i];
+            if (visited.has(nodeId))
+                continue;
+            visited.add(nodeId);
+            for (const edge of this.edges) {
+                if (edge.source.nodeId === nodeId) {
+                    const targetId = edge.target.nodeId;
+                    if (!visited.has(targetId)) {
+                        toCancel.add(targetId);
+                        queue.push(targetId);
+                    }
+                }
+            }
+        }
+        for (const nodeId of toCancel) {
+            const node = this.nodes.get(nodeId);
+            if (!node)
+                continue;
+            for (const controller of Array.from(node.activeControllers)) {
+                try {
+                    controller.abort("snapshot");
+                }
+                catch {
+                    // ignore
+                }
+            }
+            node.activeControllers.clear();
+            node.stats.active = 0;
+            node.queue = [];
+            node.runSeq += 1;
+            const now = Date.now();
+            node.latestRunId = `${nodeId}:${node.runSeq}:${now}:snapshot`;
+        }
+    }
     // Hydrate inputs/outputs without triggering computation; optionally re-emit outputs downstream
     hydrate(payload, opts) {
         const prevPaused = this.paused;