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

package/dist/index.cjs

@@ -934,13 +934,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: import_nestjs_durable_core.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 = (0, import_nestjs_durable_core.singletonAdmissionTags)(runs, runId, tag, workflow, limit);
+      if (!tags) {
+        const patch = (0, import_nestjs_durable_core.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 && import_nestjs_durable_core.TERMINAL_RUN_STATUSES.includes(inputPatch.status) ? await (0, import_nestjs_durable_core.clearTerminalSingletonAdmission)(runId, inputPatch, (id) => this.withScope({}).getRun(id)) : inputPatch;
     const em = this.fork();
     const entity = await em.findOneOrFail(WorkflowRunEntity, {
       id: runId