@dudousxd/nestjs-durable-store-mikro-orm 0.30.0 → 0.31.0

package/dist/index.d.cts

@@ -188,6 +188,7 @@ declare class MikroOrmStateStore implements StateStore {
     tryAdmitSingleton(runId: string, tag: string, workflow: string, limit: number, retryWakeAt?: number): Promise<boolean>;
     createRun(run: WorkflowRun): Promise<void>;
     updateRun(runId: string, inputPatch: Partial<WorkflowRun>): Promise<void>;
+    updateRunIf(runId: string, expectedStatuses: RunStatus[], inputPatch: Partial<WorkflowRun>): Promise<boolean>;
     /** Rewrite a run's normalized attribute rows: delete the old set, insert the current one. Mirrors
      *  the in-memory store's reindex so the side-table always reflects the run's live searchAttributes. */
     private reindexAttributes;
@@ -204,6 +205,7 @@ declare class MikroOrmStateStore implements StateStore {
         saveCheckpoint: (cp: StepCheckpoint) => Promise<void>;
     }) => Promise<T>): Promise<T>;
     listIncompleteRuns(namespace?: string): Promise<WorkflowRun[]>;
+    listOrphanedRuns(nowMs: number, limit: number, namespace?: string): Promise<WorkflowRun[]>;
     listPendingRuns(limit: number, namespace?: string): Promise<WorkflowRun[]>;
     listDueTimers(nowMs: number, namespace?: string, limit?: number): Promise<WorkflowRun[]>;
     tryLockRun(runId: string, owner: string, leaseUntilMs: number, nowMs: number): Promise<boolean>;

package/dist/index.d.ts

@@ -188,6 +188,7 @@ declare class MikroOrmStateStore implements StateStore {
     tryAdmitSingleton(runId: string, tag: string, workflow: string, limit: number, retryWakeAt?: number): Promise<boolean>;
     createRun(run: WorkflowRun): Promise<void>;
     updateRun(runId: string, inputPatch: Partial<WorkflowRun>): Promise<void>;
+    updateRunIf(runId: string, expectedStatuses: RunStatus[], inputPatch: Partial<WorkflowRun>): Promise<boolean>;
     /** Rewrite a run's normalized attribute rows: delete the old set, insert the current one. Mirrors
      *  the in-memory store's reindex so the side-table always reflects the run's live searchAttributes. */
     private reindexAttributes;
@@ -204,6 +205,7 @@ declare class MikroOrmStateStore implements StateStore {
         saveCheckpoint: (cp: StepCheckpoint) => Promise<void>;
     }) => Promise<T>): Promise<T>;
     listIncompleteRuns(namespace?: string): Promise<WorkflowRun[]>;
+    listOrphanedRuns(nowMs: number, limit: number, namespace?: string): Promise<WorkflowRun[]>;
     listPendingRuns(limit: number, namespace?: string): Promise<WorkflowRun[]>;
     listDueTimers(nowMs: number, namespace?: string, limit?: number): Promise<WorkflowRun[]>;
     tryLockRun(runId: string, owner: string, leaseUntilMs: number, nowMs: number): Promise<boolean>;

package/dist/index.js

@@ -969,6 +969,30 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     await em.flush();
     if ("searchAttributes" in patch) await this.reindexAttributes(runId, patch.searchAttributes);
   }
+  async updateRunIf(runId, expectedStatuses, inputPatch) {
+    if (expectedStatuses.length === 0) return false;
+    const patch = inputPatch.status && TERMINAL_RUN_STATUSES.includes(inputPatch.status) ? await clearTerminalSingletonAdmission(runId, inputPatch, (id) => this.withScope({}).getRun(id)) : inputPatch;
+    const data = runUpdateData(patch);
+    const em = this.fork();
+    let applied;
+    if (Object.keys(data).length) {
+      applied = await em.nativeUpdate(WorkflowRunEntity, {
+        id: runId,
+        status: {
+          $in: expectedStatuses
+        }
+      }, data) > 0;
+    } else {
+      applied = await em.count(WorkflowRunEntity, {
+        id: runId,
+        status: {
+          $in: expectedStatuses
+        }
+      }) > 0;
+    }
+    if (applied && "searchAttributes" in patch) await this.reindexAttributes(runId, patch.searchAttributes);
+    return applied;
+  }
   /** Rewrite a run's normalized attribute rows: delete the old set, insert the current one. Mirrors
   *  the in-memory store's reindex so the side-table always reflects the run's live searchAttributes. */
   async reindexAttributes(runId, attributes, forked) {
@@ -1133,6 +1157,37 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     });
     return rows.map(fromRunEntity);
   }
+  async listOrphanedRuns(nowMs, limit, namespace) {
+    if (limit <= 0) return [];
+    const em = this.fork();
+    const rows = await em.find(WorkflowRunEntity, {
+      status: {
+        $in: [
+          "running",
+          "cancelling"
+        ]
+      },
+      $or: [
+        {
+          lockedUntil: null
+        },
+        {
+          lockedUntil: {
+            $lte: new Date(nowMs)
+          }
+        }
+      ],
+      ...namespace !== void 0 ? {
+        namespace
+      } : {}
+    }, {
+      orderBy: {
+        createdAt: "ASC"
+      },
+      limit
+    });
+    return rows.map(fromRunEntity);
+  }
   async listPendingRuns(limit, namespace) {
     const em = this.fork();
     const rows = await em.find(WorkflowRunEntity, {
@@ -1894,6 +1949,29 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     return affected > 0;
   }
 };
+function runUpdateData(patch) {
+  const data = {};
+  if ("workflow" in patch) data.workflow = patch.workflow;
+  if ("workflowVersion" in patch) data.workflowVersion = patch.workflowVersion;
+  if ("status" in patch) data.status = patch.status;
+  if ("input" in patch) data.input = patch.input ?? null;
+  if ("output" in patch) data.output = patch.output ?? null;
+  if ("error" in patch) data.error = patch.error ?? null;
+  if ("wakeAt" in patch) data.wakeAt = patch.wakeAt == null ? null : new Date(patch.wakeAt);
+  if ("lockedBy" in patch) data.lockedBy = patch.lockedBy ?? null;
+  if ("lockedUntil" in patch) data.lockedUntil = patch.lockedUntil == null ? null : new Date(patch.lockedUntil);
+  if ("awaitingDecisionTaskId" in patch) data.awaitingDecisionTaskId = patch.awaitingDecisionTaskId ?? null;
+  if ("recoveryAttempts" in patch) data.recoveryAttempts = patch.recoveryAttempts ?? null;
+  if ("tags" in patch) data.tags = patch.tags ?? null;
+  if ("searchAttributes" in patch) data.searchAttributes = patch.searchAttributes ?? null;
+  if ("priority" in patch) data.priority = patch.priority ?? null;
+  if ("namespace" in patch) data.namespace = patch.namespace ?? "default";
+  if ("origin" in patch) data.origin = patch.origin ?? null;
+  if (patch.createdAt != null) data.createdAt = patch.createdAt;
+  if (patch.updatedAt != null) data.updatedAt = patch.updatedAt;
+  return data;
+}
+__name(runUpdateData, "runUpdateData");
 function toRunEntity(run) {
   const e = new WorkflowRunEntity();
   e.id = run.id;