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

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
@@ -185,8 +185,9 @@ declare class MikroOrmStateStore implements StateStore {
      */
     private fork;
     ensureSchema(): Promise<void>;
+    tryAdmitSingleton(runId: string, tag: string, workflow: string, limit: number, retryWakeAt?: number): Promise<boolean>;
     createRun(run: WorkflowRun): Promise<void>;
-    updateRun(runId: string, patch: Partial<WorkflowRun>): Promise<void>;
+    updateRun(runId: string, inputPatch: Partial<WorkflowRun>): Promise<void>;
     /** Rewrite a run's normalized attribute rows: delete the old set, insert the current one. Mirrors
      *  the in-memory store's reindex so the side-table always reflects the run's live searchAttributes. */
     private reindexAttributes;
@@ -290,6 +291,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
@@ -185,8 +185,9 @@ declare class MikroOrmStateStore implements StateStore {
      */
     private fork;
     ensureSchema(): Promise<void>;
+    tryAdmitSingleton(runId: string, tag: string, workflow: string, limit: number, retryWakeAt?: number): Promise<boolean>;
     createRun(run: WorkflowRun): Promise<void>;
-    updateRun(runId: string, patch: Partial<WorkflowRun>): Promise<void>;
+    updateRun(runId: string, inputPatch: Partial<WorkflowRun>): Promise<void>;
     /** Rewrite a run's normalized attribute rows: delete the old set, insert the current one. Mirrors
      *  the in-memory store's reindex so the side-table always reflects the run's live searchAttributes. */
     private reindexAttributes;
@@ -290,6 +291,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",
@@ -633,7 +642,7 @@ __name(durableEntities, "durableEntities");
 var ENTITIES = durableEntities();
 
 // src/mikro-orm-state-store.ts
-import { ENGINE_MINTED_TAG_PREFIXES, RUN_VALUE_FACET_LIMIT, TERMINAL_RUN_STATUSES, attributePredicateOperands, axisIsRunColumn, facetOrigin, mergeRunFacetRows, normalizeAttributeRows, parseDuration } from "@dudousxd/nestjs-durable-core";
+import { ENGINE_MINTED_TAG_PREFIXES, RUN_VALUE_FACET_LIMIT, SINGLETON_ACTIVE_STATUSES, TERMINAL_RUN_STATUSES, attributePredicateOperands, axisIsRunColumn, clearTerminalSingletonAdmission, facetOrigin, mergeRunFacetRows, normalizeAttributeRows, parseDuration, singletonAdmissionDenialPatch, singletonAdmissionTags } from "@dudousxd/nestjs-durable-core";
 import { raw } from "@mikro-orm/core";
 
 // src/schema.ts
@@ -892,13 +901,50 @@ var MikroOrmStateStore = class _MikroOrmStateStore {
   async ensureSchema() {
     await ensureMikroOrmDurableSchema(this.orm);
   }
+  async tryAdmitSingleton(runId, tag, workflow, limit, retryWakeAt) {
+    return this.fork().transactional(async (em) => {
+      const where = {
+        workflow,
+        status: {
+          $in: SINGLETON_ACTIVE_STATUSES
+        },
+        ...this.scopeNamespace !== void 0 ? {
+          namespace: this.scopeNamespace
+        } : {}
+      };
+      await em.nativeUpdate(WorkflowRunEntity, where, {
+        workflow
+      });
+      const rows = await em.find(WorkflowRunEntity, where);
+      const runs = rows.map(fromRunEntity);
+      const tags = singletonAdmissionTags(runs, runId, tag, workflow, limit);
+      if (!tags) {
+        const patch = singletonAdmissionDenialPatch(runs, runId, tag, workflow, retryWakeAt);
+        if (patch) await em.nativeUpdate(WorkflowRunEntity, {
+          id: runId
+        }, {
+          status: "suspended",
+          wakeAt: new Date(patch.wakeAt),
+          updatedAt: patch.updatedAt
+        });
+        return false;
+      }
+      await em.nativeUpdate(WorkflowRunEntity, {
+        id: runId
+      }, {
+        tags
+      });
+      return true;
+    });
+  }
   async createRun(run) {
     const em = this.fork();
     em.create(WorkflowRunEntity, toRunEntity(run));
     await em.flush();
     await this.reindexAttributes(run.id, run.searchAttributes);
   }
-  async updateRun(runId, patch) {
+  async updateRun(runId, inputPatch) {
+    const patch = inputPatch.status && TERMINAL_RUN_STATUSES.includes(inputPatch.status) ? await clearTerminalSingletonAdmission(runId, inputPatch, (id) => this.withScope({}).getRun(id)) : inputPatch;
     const em = this.fork();
     const entity = await em.findOneOrFail(WorkflowRunEntity, {
       id: runId
@@ -1561,6 +1607,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, {