@dudousxd/nestjs-durable-store-typeorm 0.22.0 → 0.23.1

package/dist/index.cjs

@@ -1019,31 +1019,26 @@ var TypeOrmStateStore = class {
   }
   async updateRun(runId, inputPatch) {
     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.getRun(id)) : inputPatch;
-    const update = {};
-    if ("workflow" in patch) update.workflow = patch.workflow;
-    if ("workflowVersion" in patch) update.workflowVersion = patch.workflowVersion;
-    if ("status" in patch) update.status = patch.status;
-    if ("input" in patch) update.input = patch.input ?? null;
-    if ("output" in patch) update.output = patch.output ?? null;
-    if ("error" in patch) update.error = patch.error ?? null;
-    if ("wakeAt" in patch) update.wakeAt = patch.wakeAt == null ? null : new Date(patch.wakeAt);
-    if ("lockedBy" in patch) update.lockedBy = patch.lockedBy ?? null;
-    if ("lockedUntil" in patch) update.lockedUntil = patch.lockedUntil == null ? null : new Date(patch.lockedUntil);
-    if ("awaitingDecisionTaskId" in patch) update.awaitingDecisionTaskId = patch.awaitingDecisionTaskId ?? null;
-    if ("recoveryAttempts" in patch) update.recoveryAttempts = patch.recoveryAttempts;
-    if ("tags" in patch) update.tags = patch.tags ?? null;
-    if ("searchAttributes" in patch) update.searchAttributes = patch.searchAttributes ?? null;
-    if ("priority" in patch) update.priority = patch.priority ?? null;
-    if ("namespace" in patch) update.namespace = patch.namespace ?? "default";
-    if ("origin" in patch) update.origin = patch.origin ?? null;
-    if ("createdAt" in patch) update.createdAt = patch.createdAt;
-    if ("updatedAt" in patch) update.updatedAt = patch.updatedAt;
+    const update = runUpdateSet(patch);
     const result = await this.runs().createQueryBuilder().update().set(update).where({
       id: runId
     }).execute();
     if (!result.affected) throw new Error(`run ${runId} not found`);
     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.getRun(id)) : inputPatch;
+    const set = runUpdateSet(patch);
+    const where = {
+      id: runId,
+      status: (0, import_typeorm2.In)(expectedStatuses)
+    };
+    const matched = Object.keys(set).length ? (await this.runs().createQueryBuilder().update().set(set).where(where).execute()).affected ?? 0 : await this.runs().countBy(where);
+    if (!matched) return false;
+    if ("searchAttributes" in patch) await this.reindexAttributes(runId, patch.searchAttributes);
+    return true;
+  }
   async getRun(runId) {
     const e = await this.runs().findOneBy({
       id: runId
@@ -1163,6 +1158,39 @@ var TypeOrmStateStore = class {
     });
     return rows.map(fromRunEntity);
   }
+  async listOrphanedRuns(nowMs, limit, namespace) {
+    if (limit <= 0) return [];
+    const now = new Date(nowMs);
+    const rows = await this.runs().find({
+      where: [
+        {
+          status: (0, import_typeorm2.In)([
+            "running",
+            "cancelling"
+          ]),
+          lockedUntil: (0, import_typeorm2.IsNull)(),
+          ...namespace !== void 0 ? {
+            namespace
+          } : {}
+        },
+        {
+          status: (0, import_typeorm2.In)([
+            "running",
+            "cancelling"
+          ]),
+          lockedUntil: (0, import_typeorm2.LessThanOrEqual)(now),
+          ...namespace !== void 0 ? {
+            namespace
+          } : {}
+        }
+      ],
+      order: {
+        createdAt: "ASC"
+      },
+      take: limit
+    });
+    return rows.map(fromRunEntity);
+  }
   async listPendingRuns(limit, namespace) {
     const rows = await this.runs().find({
       where: {
@@ -1827,6 +1855,29 @@ function fromScheduleEntity(e) {
   };
 }
 __name(fromScheduleEntity, "fromScheduleEntity");
+function runUpdateSet(patch) {
+  const update = {};
+  if ("workflow" in patch) update.workflow = patch.workflow;
+  if ("workflowVersion" in patch) update.workflowVersion = patch.workflowVersion;
+  if ("status" in patch) update.status = patch.status;
+  if ("input" in patch) update.input = patch.input ?? null;
+  if ("output" in patch) update.output = patch.output ?? null;
+  if ("error" in patch) update.error = patch.error ?? null;
+  if ("wakeAt" in patch) update.wakeAt = patch.wakeAt == null ? null : new Date(patch.wakeAt);
+  if ("lockedBy" in patch) update.lockedBy = patch.lockedBy ?? null;
+  if ("lockedUntil" in patch) update.lockedUntil = patch.lockedUntil == null ? null : new Date(patch.lockedUntil);
+  if ("awaitingDecisionTaskId" in patch) update.awaitingDecisionTaskId = patch.awaitingDecisionTaskId ?? null;
+  if ("recoveryAttempts" in patch) update.recoveryAttempts = patch.recoveryAttempts;
+  if ("tags" in patch) update.tags = patch.tags ?? null;
+  if ("searchAttributes" in patch) update.searchAttributes = patch.searchAttributes ?? null;
+  if ("priority" in patch) update.priority = patch.priority ?? null;
+  if ("namespace" in patch) update.namespace = patch.namespace ?? "default";
+  if ("origin" in patch) update.origin = patch.origin ?? null;
+  if ("createdAt" in patch) update.createdAt = patch.createdAt;
+  if ("updatedAt" in patch) update.updatedAt = patch.updatedAt;
+  return update;
+}
+__name(runUpdateSet, "runUpdateSet");
 async function updateIfPending(repo, checkpoint) {
   const { runId, seq, ...data } = toCheckpointEntity(checkpoint);
   const set = {