@dudousxd/nestjs-durable-core 0.28.0 → 0.28.1

package/dist/index.cjs

@@ -2103,7 +2103,12 @@ var WorkflowEngine = class {
     for (const w of await this.store.listSignalWaiters("child:")) {
       if (w.runId === parentRunId) childIds.add(w.token.slice("child:".length));
     }
-    for (const cp of await this.store.listCheckpoints(parentRunId)) {
+    const prefixes = [
+      "signal:child:",
+      "spawn:"
+    ];
+    const childCheckpoints = this.store.listCheckpointsByNamePrefix ? await this.store.listCheckpointsByNamePrefix(parentRunId, prefixes) : (await this.store.listCheckpoints(parentRunId)).filter((cp) => prefixes.some((p) => cp.name.startsWith(p)));
+    for (const cp of childCheckpoints) {
       if (cp.name.startsWith("signal:child:")) childIds.add(cp.name.slice("signal:child:".length));
       if (cp.name.startsWith("spawn:") && typeof cp.output === "string") childIds.add(cp.output);
     }
@@ -2894,9 +2899,12 @@ var WorkflowEngine = class {
   */
   async getEvent(runId, key) {
     const name = `event:${key}`;
-    const checkpoints = await this.store.listCheckpoints(runId);
+    if (this.store.getLatestCheckpointByName) {
+      const latest2 = await this.store.getLatestCheckpointByName(runId, name);
+      return latest2?.output;
+    }
     let latest;
-    for (const cp of checkpoints) if (cp.name === name) latest = cp.output;
+    for (const cp of await this.store.listCheckpoints(runId)) if (cp.name === name) latest = cp.output;
     return latest;
   }
   /**
@@ -3349,6 +3357,19 @@ var CodecStateStore = class {
   async listCheckpoints(runId) {
     return (await this.inner.listCheckpoints(runId)).map((c) => this.decCp(c));
   }
+  async getLatestCheckpointByName(runId, name) {
+    if (this.inner.getLatestCheckpointByName) {
+      const cp = await this.inner.getLatestCheckpointByName(runId, name);
+      return cp ? this.decCp(cp) : void 0;
+    }
+    let latest;
+    for (const cp of await this.inner.listCheckpoints(runId)) if (cp.name === name) latest = cp;
+    return latest ? this.decCp(latest) : void 0;
+  }
+  async listCheckpointsByNamePrefix(runId, prefixes) {
+    const matches = this.inner.listCheckpointsByNamePrefix ? await this.inner.listCheckpointsByNamePrefix(runId, prefixes) : (await this.inner.listCheckpoints(runId)).filter((cp) => prefixes.some((p) => cp.name.startsWith(p)));
+    return matches.map((c) => this.decCp(c));
+  }
 };
 
 // src/metrics.ts
@@ -3373,11 +3394,16 @@ function collectMetrics(engine) {
   };
   const wf = /* @__PURE__ */ __name((name) => {
     const k = name ?? "unknown";
-    return m.byWorkflow[k] ??= {
-      started: 0,
-      completed: 0,
-      failed: 0
-    };
+    let entry = m.byWorkflow[k];
+    if (!entry) {
+      entry = {
+        started: 0,
+        completed: 0,
+        failed: 0
+      };
+      m.byWorkflow[k] = entry;
+    }
+    return entry;
   }, "wf");
   const stop = engine.subscribe((e) => {
     switch (e.type) {
@@ -3624,6 +3650,13 @@ var InMemoryStateStore = class {
   attributeIndex = /* @__PURE__ */ new Map();
   /** Test/observability hook: how many candidate runs the last attribute query's index produced. */
   lastAttributeCandidates = 0;
+  /**
+  * Test/observability hook: how many checkpoint rows the LAST checkpoint read materialized (the rows
+  * actually returned to the caller). `listCheckpoints` returns all rows for the run; the targeted
+  * {@link getLatestCheckpointByName} / {@link listCheckpointsByNamePrefix} return only the matches —
+  * so this lets the benchmark contrast "fetch N, scan in JS" against "fetch the k that match".
+  */
+  lastCheckpointRowsRead = 0;
   key(runId, seq) {
     return `${runId}:${seq}`;
   }
@@ -3830,11 +3863,33 @@ var InMemoryStateStore = class {
     }));
   }
   async listCheckpoints(runId) {
-    return [
+    const rows = [
       ...this.checkpoints.values()
     ].filter((cp) => cp.runId === runId).sort((a, b) => a.seq - b.seq).map((cp) => ({
       ...cp
     }));
+    this.lastCheckpointRowsRead = rows.length;
+    return rows;
+  }
+  async getLatestCheckpointByName(runId, name) {
+    let latest;
+    for (const cp of this.checkpoints.values()) {
+      if (cp.runId !== runId || cp.name !== name) continue;
+      if (!latest || cp.seq > latest.seq) latest = cp;
+    }
+    this.lastCheckpointRowsRead = latest ? 1 : 0;
+    return latest ? {
+      ...latest
+    } : void 0;
+  }
+  async listCheckpointsByNamePrefix(runId, prefixes) {
+    const rows = [
+      ...this.checkpoints.values()
+    ].filter((cp) => cp.runId === runId && prefixes.some((p) => cp.name.startsWith(p))).sort((a, b) => a.seq - b.seq).map((cp) => ({
+      ...cp
+    }));
+    this.lastCheckpointRowsRead = rows.length;
+    return rows;
   }
 };