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

package/dist/index.js

@@ -84,11 +84,35 @@ 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,
+        // Optional args: the store's own forks set the scope via `setFilterParams` (undefined =
+        // operator, sees all), but an external consumer that shares this ORM (an operator app reading
+        // WorkflowRunEntity through its own EntityManager) never sets it. Without this, MikroORM
+        // throws "No arguments provided for filter 'namespace'" on those queries. `args: false` makes
+        // the param optional — cond then receives `undefined` and no-ops (sees all), which is exactly
+        // the operator view, while a scoped store still passes its namespace through.
+        args: false
+      }
+    },
     // 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 +638,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 +696,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 +704,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 +727,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 +793,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 +801,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 +814,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 +829,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 +844,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 +858,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 +878,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 +887,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 +897,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 +923,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 +1006,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 +1017,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 +1030,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 +1046,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 +1056,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 +1071,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 +1085,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 +1093,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
     }, {