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

package/dist/index.d.cts

@@ -204,7 +204,7 @@ declare class MikroOrmStateStore implements StateStore {
     }) => Promise<T>): Promise<T>;
     listIncompleteRuns(namespace?: string): Promise<WorkflowRun[]>;
     listPendingRuns(limit: number, namespace?: string): Promise<WorkflowRun[]>;
-    listDueTimers(nowMs: number, namespace?: string): Promise<WorkflowRun[]>;
+    listDueTimers(nowMs: number, namespace?: string, limit?: number): Promise<WorkflowRun[]>;
     tryLockRun(runId: string, owner: string, leaseUntilMs: number, nowMs: number): Promise<boolean>;
     releaseRunLock(runId: string): Promise<void>;
     renewRunLock(runId: string, owner: string, leaseUntilMs: number): Promise<boolean>;
@@ -288,6 +288,7 @@ declare class MikroOrmStateStore implements StateStore {
     listSchedules(query: ScheduleQuery): Promise<ScheduleRecord[]>;
     listCheckpoints(runId: string): Promise<StepCheckpoint[]>;
     getLatestCheckpointByName(runId: string, name: string): Promise<StepCheckpoint | undefined>;
+    countCheckpoints(runId: string, fromSeq: number, toSeq: number, statuses: StepCheckpoint['status'][]): Promise<number>;
     listCheckpointsByNamePrefix(runId: string, prefixes: string[]): Promise<StepCheckpoint[]>;
     putSignalWaiter(waiter: SignalWaiter): Promise<void>;
     takeSignalWaiter(token: string): Promise<SignalWaiter | null>;

package/dist/index.d.ts

@@ -204,7 +204,7 @@ declare class MikroOrmStateStore implements StateStore {
     }) => Promise<T>): Promise<T>;
     listIncompleteRuns(namespace?: string): Promise<WorkflowRun[]>;
     listPendingRuns(limit: number, namespace?: string): Promise<WorkflowRun[]>;
-    listDueTimers(nowMs: number, namespace?: string): Promise<WorkflowRun[]>;
+    listDueTimers(nowMs: number, namespace?: string, limit?: number): Promise<WorkflowRun[]>;
     tryLockRun(runId: string, owner: string, leaseUntilMs: number, nowMs: number): Promise<boolean>;
     releaseRunLock(runId: string): Promise<void>;
     renewRunLock(runId: string, owner: string, leaseUntilMs: number): Promise<boolean>;
@@ -288,6 +288,7 @@ declare class MikroOrmStateStore implements StateStore {
     listSchedules(query: ScheduleQuery): Promise<ScheduleRecord[]>;
     listCheckpoints(runId: string): Promise<StepCheckpoint[]>;
     getLatestCheckpointByName(runId: string, name: string): Promise<StepCheckpoint | undefined>;
+    countCheckpoints(runId: string, fromSeq: number, toSeq: number, statuses: StepCheckpoint['status'][]): Promise<number>;
     listCheckpointsByNamePrefix(runId: string, prefixes: string[]): Promise<StepCheckpoint[]>;
     putSignalWaiter(waiter: SignalWaiter): Promise<void>;
     takeSignalWaiter(token: string): Promise<SignalWaiter | null>;

package/dist/index.js

@@ -1116,7 +1116,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",
@@ -1127,6 +1127,13 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
       ...namespace !== void 0 ? {
         namespace
       } : {}
+    }, {
+      orderBy: {
+        wakeAt: "asc"
+      },
+      ...limit !== void 0 ? {
+        limit
+      } : {}
     });
     return rows.map(fromRunEntity);
   }
@@ -1563,6 +1570,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();