@dudousxd/nestjs-durable-store-mikro-orm 0.17.0 → 0.18.0

package/dist/index.js

@@ -84,11 +84,28 @@ var BufferedSignalEntity = class {
   token;
   payload;
 };
+function namespaceFilterCond(args) {
+  if (args.namespace === void 0) return {};
+  return {
+    namespace: args.namespace
+  };
+}
+__name(namespaceFilterCond, "namespaceFilterCond");
 function durableEntities(options = {}) {
   const col = columnNamer(options.naming ?? "snake_case");
   const workflowRuns = new EntitySchema({
     class: WorkflowRunEntity,
     tableName: "durable_workflow_runs",
+    // Global filter: confines every read to the tenant's namespace when the store is scoped.
+    // The `cond` returns `{}` (no-op) when the `namespace` arg is `undefined`, so an unscoped
+    // (operator / control-plane) store sees all rows unchanged — existing behaviour is preserved.
+    filters: {
+      namespace: {
+        name: "namespace",
+        cond: namespaceFilterCond,
+        default: true
+      }
+    },
     // Indexes mirror the Prisma adapter so a store swap keeps the same plan. The poller hits these
     // every tick: `(status, wakeAt)` serves both the due-timer scan (status='suspended' AND wakeAt<=now)
     // and, via its `status` prefix, the pending/incomplete status lookups; `(workflow, status)` serves
@@ -614,25 +631,51 @@ async function alignDurableCollation(orm) {
 __name(alignDurableCollation, "alignDurableCollation");
 
 // src/mikro-orm-state-store.ts
-var MikroOrmStateStore = class {
+var MikroOrmStateStore = class _MikroOrmStateStore {
   static {
     __name(this, "MikroOrmStateStore");
   }
   orm;
-  constructor(orm) {
+  scopeNamespace;
+  constructor(orm, opts) {
     this.orm = orm;
+    this.scopeNamespace = opts?.scope?.namespace;
+  }
+  /**
+  * Derive a store confined to a single tenant `scope.namespace`, sharing this store's ORM (no new
+  * connection). Turns the operator (unscoped) store into a tenant-boundary view at wiring time —
+  * used by the NestJS module's `scopeReads` option, which receives a pre-built store and can't
+  * reconstruct it. Pass `{ namespace: undefined }` for the unscoped operator view.
+  */
+  withScope(scope) {
+    return new _MikroOrmStateStore(this.orm, {
+      scope
+    });
+  }
+  /**
+  * Create a forked EntityManager and activate the `namespace` global filter with the store's
+  * scope (if any). All read operations go through this so the tenant boundary is applied
+  * uniformly. Write operations (nativeDelete, nativeUpdate, upsert, insertMany) bypass global
+  * filters in MikroORM and are unaffected.
+  */
+  fork() {
+    const em = this.orm.em.fork();
+    em.setFilterParams("namespace", {
+      namespace: this.scopeNamespace
+    });
+    return em;
   }
   async ensureSchema() {
     await ensureMikroOrmDurableSchema(this.orm);
   }
   async createRun(run) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     em.create(WorkflowRunEntity, toRunEntity(run));
     await em.flush();
     await this.reindexAttributes(run.id, run.searchAttributes);
   }
   async updateRun(runId, patch) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const entity = await em.findOneOrFail(WorkflowRunEntity, {
       id: runId
     });
@@ -646,7 +689,7 @@ var MikroOrmStateStore = class {
   /** 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. */
   async reindexAttributes(runId, attributes, forked) {
-    const em = forked ?? this.orm.em.fork();
+    const em = forked ?? this.fork();
     await em.nativeDelete(RunAttributeEntity, {
       runId
     });
@@ -654,14 +697,14 @@ var MikroOrmStateStore = class {
     if (rows.length) await em.insertMany(RunAttributeEntity, rows);
   }
   async getRun(runId) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const entity = await em.findOne(WorkflowRunEntity, {
       id: runId
     });
     return entity ? fromRunEntity(entity) : null;
   }
   async deleteRun(runId) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     await em.nativeDelete(StepCheckpointEntity, {
       runId
     });
@@ -677,7 +720,7 @@ var MikroOrmStateStore = class {
   }
   async pruneTerminalRuns(policy, nowMs, limit) {
     if (policy.statuses.length === 0 || limit <= 0) return 0;
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const status = {
       $in: policy.statuses
     };
@@ -743,7 +786,7 @@ var MikroOrmStateStore = class {
     return idList.length;
   }
   async getCheckpoint(runId, seq) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const entity = await em.findOne(StepCheckpointEntity, {
       runId,
       seq
@@ -751,12 +794,12 @@ var MikroOrmStateStore = class {
     return entity ? fromCheckpointEntity(entity) : null;
   }
   async saveCheckpoint(checkpoint) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     await em.upsert(StepCheckpointEntity, toCheckpointEntity(checkpoint));
     await em.flush();
   }
   async transaction(work) {
-    return this.orm.em.fork().transactional(async (em) => work({
+    return this.fork().transactional(async (em) => work({
       raw: em,
       saveCheckpoint: /* @__PURE__ */ __name(async (cp) => {
         await em.upsert(StepCheckpointEntity, toCheckpointEntity(cp));
@@ -764,7 +807,7 @@ var MikroOrmStateStore = class {
     }));
   }
   async listIncompleteRuns(namespace) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const rows = await em.find(WorkflowRunEntity, {
       status: {
         $in: [
@@ -779,7 +822,7 @@ var MikroOrmStateStore = class {
     return rows.map(fromRunEntity);
   }
   async listPendingRuns(limit, namespace) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const rows = await em.find(WorkflowRunEntity, {
       status: "pending",
       ...namespace !== void 0 ? {
@@ -794,7 +837,7 @@ var MikroOrmStateStore = class {
     return rows.map(fromRunEntity);
   }
   async listDueTimers(nowMs, namespace) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const rows = await em.find(WorkflowRunEntity, {
       status: "suspended",
       wakeAt: {
@@ -808,7 +851,7 @@ var MikroOrmStateStore = class {
     return rows.map(fromRunEntity);
   }
   async tryLockRun(runId, owner, leaseUntilMs, nowMs) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const affected = await em.nativeUpdate(WorkflowRunEntity, {
       id: runId,
       $or: [
@@ -828,7 +871,7 @@ var MikroOrmStateStore = class {
     return affected === 1;
   }
   async releaseRunLock(runId) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     await em.nativeUpdate(WorkflowRunEntity, {
       id: runId
     }, {
@@ -837,7 +880,7 @@ var MikroOrmStateStore = class {
     });
   }
   async renewRunLock(runId, owner, leaseUntilMs) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const affected = await em.nativeUpdate(WorkflowRunEntity, {
       id: runId,
       lockedBy: owner
@@ -847,7 +890,7 @@ var MikroOrmStateStore = class {
     return affected === 1;
   }
   async listRuns(query) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const where = {};
     if (query.workflow) where.workflow = query.workflow;
     if (query.namespace !== void 0) where.namespace = query.namespace;
@@ -873,6 +916,7 @@ var MikroOrmStateStore = class {
       createdAt: "desc"
     };
     if (query.tag || query.attributes?.length) {
+      if (this.scopeNamespace !== void 0) where.namespace = this.scopeNamespace;
       const quote = this.idQuote(em);
       const cols = this.attributeColumns(em);
       const qb = em.createQueryBuilder(WorkflowRunEntity, "r").where(where).orderBy(orderBy);
@@ -955,7 +999,7 @@ var MikroOrmStateStore = class {
     return (id) => `${ch}${id}${ch}`;
   }
   async listCheckpoints(runId) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const rows = await em.find(StepCheckpointEntity, {
       runId
     }, {
@@ -966,7 +1010,7 @@ var MikroOrmStateStore = class {
     return rows.map(fromCheckpointEntity);
   }
   async getLatestCheckpointByName(runId, name) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const entity = await em.findOne(StepCheckpointEntity, {
       runId,
       name
@@ -979,7 +1023,7 @@ var MikroOrmStateStore = class {
   }
   async listCheckpointsByNamePrefix(runId, prefixes) {
     if (prefixes.length === 0) return [];
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const rows = await em.find(StepCheckpointEntity, {
       runId,
       $or: prefixes.map((p) => ({
@@ -995,7 +1039,7 @@ var MikroOrmStateStore = class {
     return rows.map(fromCheckpointEntity);
   }
   async putSignalWaiter(waiter) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     await em.upsert(SignalWaiterEntity, {
       token: waiter.token,
       runId: waiter.runId,
@@ -1005,7 +1049,7 @@ var MikroOrmStateStore = class {
     await em.flush();
   }
   async takeSignalWaiter(token) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const entity = await em.findOne(SignalWaiterEntity, {
       token
     });
@@ -1020,7 +1064,7 @@ var MikroOrmStateStore = class {
     return waiter;
   }
   async listSignalWaiters(prefix) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const rows = await em.find(SignalWaiterEntity, {
       token: {
         $like: `${prefix}%`
@@ -1034,7 +1078,7 @@ var MikroOrmStateStore = class {
     }));
   }
   async bufferSignal(token, payload) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const e = new BufferedSignalEntity();
     e.token = token;
     e.payload = payload ?? null;
@@ -1042,7 +1086,7 @@ var MikroOrmStateStore = class {
     await em.flush();
   }
   async takeBufferedSignal(token) {
-    const em = this.orm.em.fork();
+    const em = this.fork();
     const entity = await em.findOne(BufferedSignalEntity, {
       token
     }, {