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

package/dist/index.cjs

@@ -980,6 +980,7 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
         }, {
           status: "suspended",
           wakeAt: new Date(patch.wakeAt),
+          tags: patch.tags,
           updatedAt: patch.updatedAt
         });
         return false;
@@ -1011,6 +1012,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 && import_nestjs_durable_core.TERMINAL_RUN_STATUSES.includes(inputPatch.status) ? await (0, import_nestjs_durable_core.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) {
@@ -1175,6 +1200,46 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     });
     return rows.map(fromRunEntity);
   }
+  async listOrphanedRuns(nowMs, limit, namespace, ownedBy) {
+    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)
+          }
+        },
+        // A live lease left by a previous process with the caller's own instance id.
+        ...ownedBy ? [
+          {
+            lockedBy: ownedBy.owner,
+            lockedUntil: {
+              $lt: new Date(ownedBy.lockedBefore)
+            }
+          }
+        ] : []
+      ],
+      ...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, {
@@ -1566,7 +1631,7 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
   *  single statement. */
   async updateSchedule(id, patch, expectedNextFireAt) {
     const data = {};
-    if (patch.namespace !== void 0) data.namespace = patch.namespace;
+    if (patch.namespace !== void 0) data.namespace = patch.namespace || "default";
     if (patch.workflow !== void 0) data.workflow = patch.workflow;
     if (patch.paused !== void 0) data.paused = patch.paused;
     if ("nextFireAt" in patch) data.nextFireAt = patch.nextFireAt ?? null;
@@ -1936,6 +2001,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;
@@ -1953,7 +2041,7 @@ function toRunEntity(run) {
   e.tags = run.tags ?? null;
   e.searchAttributes = run.searchAttributes ?? null;
   e.priority = run.priority ?? null;
-  e.namespace = run.namespace ?? "default";
+  e.namespace = run.namespace || "default";
   e.origin = run.origin ?? null;
   e.createdAt = run.createdAt;
   e.updatedAt = run.updatedAt;