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

package/dist/index.d.cts

@@ -1,6 +1,6 @@
 import * as _mikro_orm_core from '@mikro-orm/core';
 import { EntityRepository, EntityRepositoryType, EntitySchema, MikroORM } from '@mikro-orm/core';
-import { StepKind, RunStatus, StateStore, WorkflowRun, RetentionPolicy, StepCheckpoint, RunQuery, RunFacetQuery, RunFacetRow, RunValueAxis, RunValueFacetOptions, RunValueFacetRow, ScheduleRecord, ScheduleQuery, SignalWaiter } from '@dudousxd/nestjs-durable-core';
+import { StepKind, RunStatus, StateStore, WorkflowRun, RetentionPolicy, StepCheckpoint, RunQuery, RunFacetQuery, RunFacetRow, RunValueAxis, RunValueFacetOptions, RunValueFacetRow, ScheduleRecord, ScheduleQuery, SignalWaiter, ChildWait } from '@dudousxd/nestjs-durable-core';
 
 /**
  * Custom repositories for the durable entities, so a host app can `@InjectRepository`/inject a
@@ -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>;
@@ -291,6 +292,10 @@ declare class MikroOrmStateStore implements StateStore {
     putSignalWaiter(waiter: SignalWaiter): Promise<void>;
     takeSignalWaiter(token: string): Promise<SignalWaiter | null>;
     listSignalWaiters(prefix: string): Promise<SignalWaiter[]>;
+    registerChildWaits(entries: ChildWait[]): Promise<void>;
+    compactCheckpoints(runId: string, fromSeq: number, toSeq: number, into: StepCheckpoint): Promise<void>;
+    listSignalWaitersByRun(runIds: string[]): Promise<SignalWaiter[]>;
+    hasSignalWaiter(runId: string, tokenPrefix: string): Promise<boolean>;
     removeSignalWaiter(waiter: SignalWaiter): Promise<void>;
     bufferSignal(token: string, payload: unknown): Promise<void>;
     takeBufferedSignal(token: string): Promise<{

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import * as _mikro_orm_core from '@mikro-orm/core';
 import { EntityRepository, EntityRepositoryType, EntitySchema, MikroORM } from '@mikro-orm/core';
-import { StepKind, RunStatus, StateStore, WorkflowRun, RetentionPolicy, StepCheckpoint, RunQuery, RunFacetQuery, RunFacetRow, RunValueAxis, RunValueFacetOptions, RunValueFacetRow, ScheduleRecord, ScheduleQuery, SignalWaiter } from '@dudousxd/nestjs-durable-core';
+import { StepKind, RunStatus, StateStore, WorkflowRun, RetentionPolicy, StepCheckpoint, RunQuery, RunFacetQuery, RunFacetRow, RunValueAxis, RunValueFacetOptions, RunValueFacetRow, ScheduleRecord, ScheduleQuery, SignalWaiter, ChildWait } from '@dudousxd/nestjs-durable-core';
 
 /**
  * Custom repositories for the durable entities, so a host app can `@InjectRepository`/inject a
@@ -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>;
@@ -291,6 +292,10 @@ declare class MikroOrmStateStore implements StateStore {
     putSignalWaiter(waiter: SignalWaiter): Promise<void>;
     takeSignalWaiter(token: string): Promise<SignalWaiter | null>;
     listSignalWaiters(prefix: string): Promise<SignalWaiter[]>;
+    registerChildWaits(entries: ChildWait[]): Promise<void>;
+    compactCheckpoints(runId: string, fromSeq: number, toSeq: number, into: StepCheckpoint): Promise<void>;
+    listSignalWaitersByRun(runIds: string[]): Promise<SignalWaiter[]>;
+    hasSignalWaiter(runId: string, tokenPrefix: string): Promise<boolean>;
     removeSignalWaiter(waiter: SignalWaiter): Promise<void>;
     bufferSignal(token: string, payload: unknown): Promise<void>;
     takeBufferedSignal(token: string): Promise<{

package/dist/index.js

@@ -472,6 +472,15 @@ function durableEntities(options = {}) {
     class: SignalWaiterEntity,
     tableName: "durable_signal_waiters",
     repository: /* @__PURE__ */ __name(() => SignalWaiterRepository, "repository"),
+    // A run's waiters: the `ctx.all` join barrier probes them once per settled child.
+    indexes: [
+      {
+        name: "durable_signal_waiters_run_id_idx",
+        properties: [
+          "runId"
+        ]
+      }
+    ],
     properties: {
       token: {
         type: "string",
@@ -1057,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,
@@ -1598,6 +1619,66 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
       parallelGroup: e.parallelGroup ?? void 0
     }));
   }
+  async registerChildWaits(entries) {
+    if (entries.length === 0) return;
+    await this.fork().transactional(async (em) => {
+      await em.upsertMany(SignalWaiterEntity, entries.map(({ waiter }) => ({
+        token: waiter.token,
+        runId: waiter.runId,
+        seq: waiter.seq,
+        parallelGroup: waiter.parallelGroup ?? null
+      })));
+      await em.upsertMany(StepCheckpointEntity, entries.map(({ checkpoint }) => toCheckpointEntity(checkpoint)));
+    });
+  }
+  async compactCheckpoints(runId, fromSeq, toSeq, into) {
+    await this.fork().transactional(async (em) => {
+      await em.nativeDelete(StepCheckpointEntity, {
+        runId,
+        seq: {
+          $gte: fromSeq,
+          $lte: toSeq
+        }
+      });
+      await em.upsert(StepCheckpointEntity, toCheckpointEntity(into));
+    });
+  }
+  async listSignalWaitersByRun(runIds) {
+    if (runIds.length === 0) return [];
+    const rows = await this.fork().find(SignalWaiterEntity, {
+      runId: {
+        $in: runIds
+      }
+    });
+    return rows.map((e) => ({
+      token: e.token,
+      runId: e.runId,
+      seq: e.seq,
+      parallelGroup: e.parallelGroup ?? void 0
+    }));
+  }
+  async hasSignalWaiter(runId, tokenPrefix) {
+    const em = this.fork();
+    const where = {
+      runId,
+      token: {
+        $like: `${tokenPrefix}%`
+      }
+    };
+    const first = await em.findOne(SignalWaiterEntity, where, {
+      fields: [
+        "token"
+      ]
+    });
+    if (!first) return false;
+    if (first.token.startsWith(tokenPrefix)) return true;
+    const rows = await em.find(SignalWaiterEntity, where, {
+      fields: [
+        "token"
+      ]
+    });
+    return rows.some((r) => r.token.startsWith(tokenPrefix));
+  }
   async removeSignalWaiter(waiter) {
     const em = this.fork();
     await em.nativeDelete(SignalWaiterEntity, {