@dudousxd/nestjs-durable-store-typeorm 0.19.0 → 0.19.1

package/dist/index.cjs

@@ -886,6 +886,7 @@ __name(existingColumns, "existingColumns");
 // src/typeorm-state-store.ts
 var import_nestjs_durable_core = require("@dudousxd/nestjs-durable-core");
 var import_typeorm2 = require("typeorm");
+var sqliteSingletonClaims = /* @__PURE__ */ new WeakMap();
 var TypeOrmStateStore = class {
   static {
     __name(this, "TypeOrmStateStore");
@@ -928,11 +929,63 @@ var TypeOrmStateStore = class {
     const rows = (0, import_nestjs_durable_core.normalizeAttributeRows)(runId, attributes);
     if (rows.length) await repo.insert(rows);
   }
+  async tryAdmitSingleton(runId, tag, workflow, limit, retryWakeAt) {
+    const previous = sqliteSingletonClaims.get(this.dataSource) ?? Promise.resolve();
+    const sqlite = [
+      "sqlite",
+      "better-sqlite3",
+      "sqljs"
+    ].includes(this.dataSource.options.type);
+    let release;
+    if (sqlite) {
+      const next = new Promise((resolve) => {
+        release = resolve;
+      });
+      sqliteSingletonClaims.set(this.dataSource, previous.then(() => next));
+      await previous;
+    }
+    try {
+      return await this.dataSource.transaction(async (em) => {
+        const repo = em.getRepository(WorkflowRunEntity);
+        const where = {
+          workflow,
+          status: (0, import_typeorm2.In)(import_nestjs_durable_core.SINGLETON_ACTIVE_STATUSES)
+        };
+        await repo.update(where, {
+          workflow
+        });
+        const runs = (await repo.find({
+          where
+        })).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 repo.update({
+            id: runId
+          }, {
+            status: "suspended",
+            wakeAt: new Date(patch.wakeAt),
+            updatedAt: patch.updatedAt
+          });
+          return false;
+        }
+        await repo.update({
+          id: runId
+        }, {
+          tags
+        });
+        return true;
+      });
+    } finally {
+      release?.();
+    }
+  }
   async createRun(run) {
     await this.runs().save(toRunEntity(run));
     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.getRun(id)) : inputPatch;
     const update = {};
     if ("workflow" in patch) update.workflow = patch.workflow;
     if ("workflowVersion" in patch) update.workflowVersion = patch.workflowVersion;