@bian-womp/spark-graph 0.3.87 → 0.3.89

package/lib/cjs/index.cjs

@@ -3825,6 +3825,7 @@ class GraphRuntime {
         this.runMode = "manual";
         this.pauseRefCount = 0;
         this.persistentPauseToken = null;
+        this.deferredAutoRunNodeIds = new Set();
         // Initialize components
         this.eventEmitter = new EventEmitter();
         this.graph = new Graph(this.eventEmitter);
@@ -3953,6 +3954,10 @@ class GraphRuntime {
         const canExecute = this.graph.allInboundHaveValue(nodeId);
         if (!shouldAutoRun || !canExecute)
             return;
+        if (this.isPaused()) {
+            this.deferredAutoRunNodeIds.add(nodeId);
+            return;
+        }
         this.withManualNoPropagateContext(nodeId, (runContextIds) => {
             this.execute(nodeId, {
                 runContextIds,
@@ -3960,6 +3965,15 @@ class GraphRuntime {
             });
         });
     }
+    flushDeferredAutoRunNodes() {
+        if (this.isPaused() || this.deferredAutoRunNodeIds.size === 0)
+            return;
+        const queuedNodeIds = Array.from(this.deferredAutoRunNodeIds);
+        this.deferredAutoRunNodeIds.clear();
+        for (const nodeId of queuedNodeIds) {
+            this.executeNodeAutoRun(nodeId, { reason: "deferred-auto-run" });
+        }
+    }
     setInputs(nodeId, inputs) {
         const node = this.graph.getNode(nodeId);
         if (!node)
@@ -4216,6 +4230,10 @@ class GraphRuntime {
                 return;
             released = true;
             this.pauseRefCount--;
+            if (this.pauseRefCount <= 0) {
+                this.pauseRefCount = 0;
+                this.flushDeferredAutoRunNodes();
+            }
         };
     }
     isPaused() {