@dudousxd/nestjs-durable-core 0.76.0 → 0.76.1

package/dist/index.cjs

@@ -3867,6 +3867,12 @@ var WorkflowEngine = class {
   * {@link redriveSettledSteps}).
   */
   awaitedRemoteSteps = /* @__PURE__ */ new Map();
+  /**
+  * When THIS instance settled a remote step from its result, by `runId:seq`. That result's resume
+  * runs here, and `execute` lets it into a lease this instance holds, so its wake is never the
+  * dropped one {@link redriveSettledSteps} recovers. Pruned by age (see {@link noteSettledHere}).
+  */
+  settledHere = /* @__PURE__ */ new Map();
   /** Flow-control admission backend for remote steps (see {@link registerQueue}). */
   admission;
   /** Runs on THIS instance blocked on admission, by queue — woken early on a freed-slot signal. */
@@ -5294,6 +5300,16 @@ var WorkflowEngine = class {
     for (const seq of seqs) {
       const cp = await this.store.getCheckpoint(runId, seq);
       if (cp && (cp.status === "completed" || cp.status === "failed")) {
+        if (this.settledHere.has(`${runId}:${seq}`)) return;
+        const nowMs = this.clock();
+        if (!await this.store.tryLockRun(runId, this.instanceId, nowMs + this.leaseMs, nowMs)) {
+          return;
+        }
+        const run = await this.store.getRun(runId);
+        if (run?.status !== "suspended") {
+          await this.store.releaseRunLock(runId);
+          return;
+        }
         await this.resumeImplicit(runId);
         return;
       }
@@ -7390,6 +7406,15 @@ var WorkflowEngine = class {
     this.awaitedRemoteSteps.get(runId)?.add(seq);
     throw new WorkflowSuspended();
   }
+  /** Remember a step settled here, forgetting entries older than a lease so the map stays small. */
+  noteSettledHere(key) {
+    const now = this.clock();
+    this.settledHere.set(key, now);
+    if (this.settledHere.size < 1024) return;
+    for (const [k, at] of this.settledHere) {
+      if (now - at > this.leaseMs) this.settledHere.delete(k);
+    }
+  }
   /** {@link StateStore.saveCheckpointIfPending}, or a plain save on a store without it. */
   async savePendingCheckpoint(checkpoint) {
     if (this.store.saveCheckpointIfPending) return this.store.saveCheckpointIfPending(checkpoint);
@@ -7436,6 +7461,7 @@ var WorkflowEngine = class {
       finishedAt
     });
     if (!claimed) return;
+    this.noteSettledHere(`${result.runId}:${result.seq}`);
     this.emit({
       type: result.status === "completed" ? "step.completed" : "step.failed",
       runId: result.runId,