@dudousxd/nestjs-durable-core 0.71.0 → 0.71.1

package/dist/index.js

@@ -3516,9 +3516,14 @@ var WorkflowEngine = class {
   async recoverIncomplete(nowMs = this.clock()) {
     if (this.draining) return [];
     const results = [];
-    for (const run of await this.store.listIncompleteRuns(this.namespace)) {
-      const acquired = await this.store.tryLockRun(run.id, this.instanceId, nowMs + this.leaseMs, nowMs);
+    for (const listed of await this.store.listIncompleteRuns(this.namespace)) {
+      const acquired = await this.store.tryLockRun(listed.id, this.instanceId, nowMs + this.leaseMs, nowMs);
       if (!acquired) continue;
+      const run = await this.store.getRun(listed.id);
+      if (!run || run.status !== "running" && run.status !== "cancelling") {
+        await this.store.releaseRunLock(listed.id);
+        continue;
+      }
       if (run.status === "running") {
         const dispatched = await this.inFlightRemoteStep(run.id);
         if (dispatched) {
@@ -5378,16 +5383,16 @@ var WorkflowEngine = class {
     try {
       const output = await runInWorkflowCtx(ctx, () => fn(ctx, run.input));
       if (this.cancelRequested.has(run.id)) {
-        return this.compensateAndCancel(run, compensations, replay);
+        return await this.compensateAndCancel(run, compensations, replay);
       }
-      return this.settleRun(run, {
+      return await this.settleRun(run, {
         kind: "completed",
         output
       });
     } catch (err) {
       if (err instanceof ContinueAsNew) {
         if (this.cancelRequested.has(run.id)) {
-          return this.compensateAndCancel(run, compensations, replay);
+          return await this.compensateAndCancel(run, compensations, replay);
         }
         await this.store.updateRun(run.id, {
           status: "completed",
@@ -5414,35 +5419,35 @@ var WorkflowEngine = class {
       }
       if (err instanceof WorkflowBlocked) {
         if (this.cancelRequested.has(run.id)) {
-          return this.compensateAndCancel(run, compensations, replay);
+          return await this.compensateAndCancel(run, compensations, replay);
         }
-        return this.settleRun(run, {
+        return await this.settleRun(run, {
           kind: "blocked",
           blocked: err.plan
         });
       }
       if (err instanceof WorkflowSuspended) {
         if (this.cancelRequested.has(run.id)) {
-          return this.compensateAndCancel(run, compensations, replay);
+          return await this.compensateAndCancel(run, compensations, replay);
         }
-        return this.settleRun(run, {
+        return await this.settleRun(run, {
           kind: "suspended",
           wakeAt: err.wakeAt
         });
       }
       if (this.cancelRequested.has(run.id)) {
-        return this.compensateAndCancel(run, compensations, replay);
+        return await this.compensateAndCancel(run, compensations, replay);
       }
       const error2 = {
         message: err instanceof Error ? err.message : String(err),
         stack: err instanceof Error ? err.stack : void 0
       };
       const pending = await this.unwindCompensations(run, compensations, replay);
-      if (pending) return this.settleRun(run, {
+      if (pending) return await this.settleRun(run, {
         kind: "suspended",
         wakeAt: pending.wakeAt
       });
-      return this.settleRun(run, {
+      return await this.settleRun(run, {
         kind: "failed",
         error: error2
       });
@@ -5856,7 +5861,10 @@ var WorkflowEngine = class {
   }
   async callRemote(runId, seq, step, input, queue, transport, replay, admission, routing) {
     const group = stepGroup(step, routing?.namespace);
-    const existing = replay?.get(seq) ?? await this.store.getCheckpoint(runId, seq);
+    let existing = replay?.get(seq) ?? await this.store.getCheckpoint(runId, seq);
+    if (existing?.status === "pending" && replay?.has(seq)) {
+      existing = await this.store.getCheckpoint(runId, seq) ?? existing;
+    }
     if (existing && existing.name !== step.name) {
       throw new NonDeterminismError(runId, seq, step.name, existing.name);
     }