@dudousxd/nestjs-durable-store-mikro-orm 0.28.0 → 0.29.0

package/dist/index.cjs

@@ -1108,6 +1108,18 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     await em.upsert(StepCheckpointEntity, toCheckpointEntity(checkpoint));
     await em.flush();
   }
+  async saveCheckpointIfPending(checkpoint) {
+    const { runId: _runId, seq: _seq, ...data } = toCheckpointEntity(checkpoint);
+    const affected = await this.fork().nativeUpdate(StepCheckpointEntity, {
+      runId: checkpoint.runId,
+      seq: checkpoint.seq,
+      status: "pending"
+    }, {
+      ...data,
+      wakeAt: checkpoint.wakeAt != null ? new Date(checkpoint.wakeAt) : null
+    });
+    return affected > 0;
+  }
   async transaction(work) {
     return this.fork().transactional(async (em) => work({
       raw: em,
@@ -1146,7 +1158,7 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     });
     return rows.map(fromRunEntity);
   }
-  async listDueTimers(nowMs, namespace) {
+  async listDueTimers(nowMs, namespace, limit) {
     const em = this.fork();
     const rows = await em.find(WorkflowRunEntity, {
       status: "suspended",
@@ -1157,6 +1169,13 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
       ...namespace !== void 0 ? {
         namespace
       } : {}
+    }, {
+      orderBy: {
+        wakeAt: "asc"
+      },
+      ...limit !== void 0 ? {
+        limit
+      } : {}
     });
     return rows.map(fromRunEntity);
   }
@@ -1593,6 +1612,19 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
     });
     return entity ? fromCheckpointEntity(entity) : void 0;
   }
+  async countCheckpoints(runId, fromSeq, toSeq, statuses) {
+    if (statuses.length === 0) return 0;
+    return this.fork().count(StepCheckpointEntity, {
+      runId,
+      seq: {
+        $gte: fromSeq,
+        $lte: toSeq
+      },
+      status: {
+        $in: statuses
+      }
+    });
+  }
   async listCheckpointsByNamePrefix(runId, prefixes) {
     if (prefixes.length === 0) return [];
     const em = this.fork();