@dudousxd/nestjs-durable-store-typeorm 0.20.1 → 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,
@@ -1404,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({