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

package/dist/index.cjs

@@ -514,6 +514,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",
@@ -934,13 +943,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
@@ -1603,6 +1649,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, {