@dudousxd/nestjs-durable-core 0.39.0 → 0.40.0

package/dist/index.cjs

@@ -1512,6 +1512,14 @@ function isNewerVersion(a, b) {
   }) > 0;
 }
 __name(isNewerVersion, "isNewerVersion");
+var NamespaceMismatch = class NamespaceMismatch2 extends Error {
+  static {
+    __name(this, "NamespaceMismatch");
+  }
+  constructor() {
+    super("namespace-mismatch");
+  }
+};
 var WorkflowEngine = class {
   static {
     __name(this, "WorkflowEngine");
@@ -1522,6 +1530,7 @@ var WorkflowEngine = class {
   controlPlane;
   clock;
   instanceId;
+  namespace;
   leaseMs;
   maxRecoveryAttempts;
   remoteAdvanceSilenceMs;
@@ -1588,6 +1597,7 @@ var WorkflowEngine = class {
     this.admission = deps.admission ?? new InMemoryAdmissionBackend(this.clock);
     this.admission.onFreed?.((queue) => this.wakeQueueWaiters(queue));
     this.instanceId = deps.instanceId ?? globalThis.crypto.randomUUID();
+    this.namespace = deps.namespace ?? "default";
     this.leaseMs = deps.leaseMs ?? 3e4;
     this.maxRecoveryAttempts = deps.maxRecoveryAttempts;
     this.remoteAdvanceSilenceMs = deps.remoteAdvanceSilenceMs;
@@ -1908,6 +1918,7 @@ var WorkflowEngine = class {
       workflow: name,
       workflowVersion: registered.version,
       status: "pending",
+      namespace: this.namespace,
       input,
       tags,
       searchAttributes: opts?.searchAttributes,
@@ -1944,6 +1955,10 @@ var WorkflowEngine = class {
         error: run.error
       };
     }
+    if (run.namespace !== void 0 && run.namespace !== this.namespace) {
+      await this.store.releaseRunLock(runId);
+      throw new NamespaceMismatch();
+    }
     const registered = this.workflows.get(versionKey(run.workflow, run.workflowVersion)) ?? await this.findInheritedRegistration(run);
     if (!registered) {
       throw new Error(`workflow ${run.workflow}@${run.workflowVersion} is not registered \u2014 keep the prior version deployed so in-flight runs can drain (skew protection)`);
@@ -2062,11 +2077,13 @@ var WorkflowEngine = class {
       const inflight = [
         ...await this.store.listRuns({
           workflow: reg.name,
-          status: "running"
+          status: "running",
+          namespace: this.namespace
         }),
         ...await this.store.listRuns({
           workflow: reg.name,
-          status: "suspended"
+          status: "suspended",
+          namespace: this.namespace
         })
       ];
       for (const run of inflight) {
@@ -2096,7 +2113,7 @@ var WorkflowEngine = class {
   async recoverIncomplete(nowMs = this.clock()) {
     if (this.draining) return [];
     const results = [];
-    for (const run of await this.store.listIncompleteRuns()) {
+    for (const run of await this.store.listIncompleteRuns(this.namespace)) {
       const acquired = await this.store.tryLockRun(run.id, this.instanceId, nowMs + this.leaseMs, nowMs);
       if (!acquired) continue;
       const settled = await this.countRecovery(run);
@@ -2166,7 +2183,7 @@ var WorkflowEngine = class {
   * boot. A run still not due re-suspends cheaply without running new work.
   */
   async resumeDueTimers(nowMs = this.clock()) {
-    return this.resumeLeased(await this.store.listDueTimers(nowMs), nowMs);
+    return this.resumeLeased(await this.store.listDueTimers(nowMs, this.namespace), nowMs);
   }
   /**
   * Lease and execute one run by id — the worker side of dispatch. Acquires the recovery lease (so
@@ -2179,7 +2196,10 @@ var WorkflowEngine = class {
     const nowMs = this.clock();
     const acquired = await this.store.tryLockRun(runId, this.instanceId, nowMs + this.leaseMs, nowMs);
     if (!acquired) return null;
-    return this.resume(runId);
+    return this.resume(runId).catch((err) => {
+      if (err instanceof NamespaceMismatch) return null;
+      throw err;
+    });
   }
   /**
   * Re-enqueue a run for a worker to (re-)execute — the dispatch-model **retry**. Sets it back to
@@ -2225,7 +2245,7 @@ var WorkflowEngine = class {
   * {@link resumeDueTimers}.
   */
   async runPending(nowMs = this.clock()) {
-    return this.resumeLeased(await this.store.listPendingRuns(100), nowMs);
+    return this.resumeLeased(await this.store.listPendingRuns(100, this.namespace), nowMs);
   }
   /**
   * Resolve once `runId` reaches a settled state — terminal (completed/failed/cancelled/dead) or
@@ -2337,7 +2357,10 @@ var WorkflowEngine = class {
       kind: "signal",
       output: payload
     }));
-    return this.resume(waiter.runId);
+    return this.resume(waiter.runId).catch((err) => {
+      if (err instanceof NamespaceMismatch) return null;
+      throw err;
+    });
   }
   /**
   * Ensure a run exists for `runId`, then deliver a signal to it — atomically race-free thanks to
@@ -3491,7 +3514,10 @@ var WorkflowEngine = class {
       queueMs: startedAt.getTime() - cp.enqueuedAt.getTime(),
       durationMs: finishedAt.getTime() - startedAt.getTime()
     });
-    await this.resume(result.runId);
+    await this.resume(result.runId).catch((err) => {
+      if (err instanceof NamespaceMismatch) return null;
+      throw err;
+    });
   }
   /**
   * A slot freed on `queue` (a fleet-wide signal): resume this instance's runs blocked on it so they
@@ -3789,14 +3815,14 @@ var CodecStateStore = class {
       saveCheckpoint: /* @__PURE__ */ __name((cp) => tx.saveCheckpoint(this.encCp(cp)), "saveCheckpoint")
     }));
   }
-  async listIncompleteRuns() {
-    return (await this.inner.listIncompleteRuns()).map((r) => this.decRun(r));
+  async listIncompleteRuns(namespace) {
+    return (await this.inner.listIncompleteRuns(namespace)).map((r) => this.decRun(r));
   }
-  async listPendingRuns(limit) {
-    return (await this.inner.listPendingRuns(limit)).map((r) => this.decRun(r));
+  async listPendingRuns(limit, namespace) {
+    return (await this.inner.listPendingRuns(limit, namespace)).map((r) => this.decRun(r));
   }
-  async listDueTimers(nowMs) {
-    return (await this.inner.listDueTimers(nowMs)).map((r) => this.decRun(r));
+  async listDueTimers(nowMs, namespace) {
+    return (await this.inner.listDueTimers(nowMs, namespace)).map((r) => this.decRun(r));
   }
   tryLockRun(runId, owner, leaseUntilMs, nowMs) {
     return this.inner.tryLockRun(runId, owner, leaseUntilMs, nowMs);
@@ -4151,7 +4177,8 @@ var InMemoryStateStore = class {
   }
   async createRun(run) {
     this.runs.set(run.id, {
-      ...run
+      ...run,
+      namespace: run.namespace ?? "default"
     });
     this.reindexAttributes(run.id, run.searchAttributes);
   }
@@ -4192,24 +4219,24 @@ var InMemoryStateStore = class {
       ...checkpoint
     });
   }
-  async listIncompleteRuns() {
+  async listIncompleteRuns(namespace) {
     return [
       ...this.runs.values()
-    ].filter((r) => r.status === "running" || r.status === "cancelling").map((r) => ({
+    ].filter((r) => (r.status === "running" || r.status === "cancelling") && (namespace === void 0 || r.namespace === namespace)).map((r) => ({
       ...r
     }));
   }
-  async listPendingRuns(limit) {
+  async listPendingRuns(limit, namespace) {
     return [
       ...this.runs.values()
-    ].filter((r) => r.status === "pending").sort((a, b) => a.createdAt.getTime() - b.createdAt.getTime() || a.id.localeCompare(b.id)).slice(0, limit).map((r) => ({
+    ].filter((r) => r.status === "pending" && (namespace === void 0 || r.namespace === namespace)).sort((a, b) => a.createdAt.getTime() - b.createdAt.getTime() || a.id.localeCompare(b.id)).slice(0, limit).map((r) => ({
       ...r
     }));
   }
-  async listDueTimers(nowMs) {
+  async listDueTimers(nowMs, namespace) {
     return [
       ...this.runs.values()
-    ].filter((r) => r.status === "suspended" && r.wakeAt !== void 0 && r.wakeAt <= nowMs).map((r) => ({
+    ].filter((r) => r.status === "suspended" && r.wakeAt !== void 0 && r.wakeAt <= nowMs && (namespace === void 0 || r.namespace === namespace)).map((r) => ({
       ...r
     }));
   }
@@ -4324,6 +4351,7 @@ var InMemoryStateStore = class {
     if (query.status) runs = runs.filter((r) => r.status === query.status);
     if (query.statuses) runs = runs.filter((r) => query.statuses?.includes(r.status));
     if (query.tag) runs = runs.filter((r) => r.tags?.includes(query.tag));
+    if (query.namespace !== void 0) runs = runs.filter((r) => r.namespace === query.namespace);
     if (query.attributes?.length) {
       let candidates = null;
       for (const f of query.attributes) {