@dudousxd/nestjs-durable-store-typeorm 0.20.0 → 0.21.0

package/dist/index.cjs

@@ -1104,6 +1104,22 @@ var TypeOrmStateStore = class {
       "seq"
     ]);
   }
+  async saveCheckpointIfPending(checkpoint) {
+    const { runId, seq, ...data } = toCheckpointEntity(checkpoint);
+    const set = {
+      ...data,
+      wakeAt: checkpoint.wakeAt != null ? new Date(checkpoint.wakeAt) : () => "NULL"
+    };
+    for (const key of Object.keys(set)) {
+      if (set[key] === void 0) delete set[key];
+    }
+    const result = await this.checkpoints().createQueryBuilder().update().set(set).where({
+      runId,
+      seq,
+      status: "pending"
+    }).execute();
+    return (result.affected ?? 0) > 0;
+  }
   async transaction(work) {
     return this.dataSource.transaction(async (em) => work({
       raw: em,
@@ -1146,12 +1162,20 @@ var TypeOrmStateStore = class {
     });
     return rows.map(fromRunEntity);
   }
-  async listDueTimers(nowMs, namespace) {
-    const rows = await this.runs().findBy({
-      status: "suspended",
-      wakeAt: (0, import_typeorm2.LessThanOrEqual)(new Date(nowMs)),
-      ...namespace !== void 0 ? {
-        namespace
+  async listDueTimers(nowMs, namespace, limit) {
+    const rows = await this.runs().find({
+      where: {
+        status: "suspended",
+        wakeAt: (0, import_typeorm2.LessThanOrEqual)(new Date(nowMs)),
+        ...namespace !== void 0 ? {
+          namespace
+        } : {}
+      },
+      order: {
+        wakeAt: "ASC"
+      },
+      ...limit !== void 0 ? {
+        take: limit
       } : {}
     });
     return rows.map(fromRunEntity);
@@ -1396,6 +1420,15 @@ var TypeOrmStateStore = class {
     });
     return e ? fromCheckpointEntity(e) : void 0;
   }
+  async countCheckpoints(runId, fromSeq, toSeq, statuses) {
+    if (statuses.length === 0) return 0;
+    const row = await this.checkpoints().createQueryBuilder("cp").select("COUNT(*)", "count").where({
+      runId,
+      seq: (0, import_typeorm2.Between)(fromSeq, toSeq),
+      status: (0, import_typeorm2.In)(statuses)
+    }).getRawOne();
+    return Number(row?.count ?? 0);
+  }
   async listCheckpointsByNamePrefix(runId, prefixes) {
     if (prefixes.length === 0) return [];
     const rows = await this.checkpoints().find({