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

package/dist/index.d.cts

@@ -197,6 +197,7 @@ declare class MikroOrmStateStore implements StateStore {
     pruneTerminalRuns(policy: RetentionPolicy, nowMs: number, limit: number): Promise<number>;
     getCheckpoint(runId: string, seq: number): Promise<StepCheckpoint | null>;
     saveCheckpoint(checkpoint: StepCheckpoint): Promise<void>;
+    saveCheckpointIfPending(checkpoint: StepCheckpoint): Promise<boolean>;
     transaction<T>(work: (tx: {
         raw: unknown;
         saveCheckpoint: (cp: StepCheckpoint) => Promise<void>;

package/dist/index.d.ts

@@ -197,6 +197,7 @@ declare class MikroOrmStateStore implements StateStore {
     pruneTerminalRuns(policy: RetentionPolicy, nowMs: number, limit: number): Promise<number>;
     getCheckpoint(runId: string, seq: number): Promise<StepCheckpoint | null>;
     saveCheckpoint(checkpoint: StepCheckpoint): Promise<void>;
+    saveCheckpointIfPending(checkpoint: StepCheckpoint): Promise<boolean>;
     transaction<T>(work: (tx: {
         raw: unknown;
         saveCheckpoint: (cp: StepCheckpoint) => Promise<void>;

package/dist/index.js

@@ -1066,6 +1066,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,