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

package/dist/index.d.cts

@@ -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;

package/dist/index.d.ts

@@ -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;

package/dist/index.js

@@ -633,7 +633,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 +892,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