@dudousxd/nestjs-durable-core 0.37.0 → 0.38.0

package/dist/index.d.cts

@@ -1122,6 +1122,15 @@ interface WorkflowCtx {
     random(): Promise<number>;
     /** Deterministic UUID v4: recorded once, then replayed. Use instead of `crypto.randomUUID()`. */
     uuid(): Promise<string>;
+    /**
+     * Merge `attrs` into THIS run's {@link WorkflowRun.searchAttributes} — the indexed metadata the
+     * dashboard and {@link RunQuery} filter on. Shallow merge: keys you don't pass are kept. Durable +
+     * exactly-once — recorded at this position on the first run and SKIPPED on replay, so it does one
+     * write, not one per turn. Use this instead of injecting the {@link StateStore} to mutate the run
+     * you're executing (`@Inject(state-store)` + `store.updateRun(ctx.runId, …)` becomes
+     * `ctx.upsertSearchAttributes(…)`).
+     */
+    upsertSearchAttributes(attrs: SearchAttributes): Promise<void>;
 }
 /** Result of executing or resuming a workflow run. */
 interface RunResult {

package/dist/index.d.ts

@@ -1122,6 +1122,15 @@ interface WorkflowCtx {
     random(): Promise<number>;
     /** Deterministic UUID v4: recorded once, then replayed. Use instead of `crypto.randomUUID()`. */
     uuid(): Promise<string>;
+    /**
+     * Merge `attrs` into THIS run's {@link WorkflowRun.searchAttributes} — the indexed metadata the
+     * dashboard and {@link RunQuery} filter on. Shallow merge: keys you don't pass are kept. Durable +
+     * exactly-once — recorded at this position on the first run and SKIPPED on replay, so it does one
+     * write, not one per turn. Use this instead of injecting the {@link StateStore} to mutate the run
+     * you're executing (`@Inject(state-store)` + `store.updateRun(ctx.runId, …)` becomes
+     * `ctx.upsertSearchAttributes(…)`).
+     */
+    upsertSearchAttributes(attrs: SearchAttributes): Promise<void>;
 }
 /** Result of executing or resuming a workflow run. */
 interface RunResult {

package/dist/index.js

@@ -1362,9 +1362,28 @@ function createWorkflowCtx(host, runId, compensations, workflow = "") {
   const now = /* @__PURE__ */ __name(() => step("now", async () => host.clock()), "now");
   const random = /* @__PURE__ */ __name(() => step("random", async () => Math.random()), "random");
   const uuid = /* @__PURE__ */ __name(() => step("uuid", async () => globalThis.crypto.randomUUID()), "uuid");
+  const upsertSearchAttributes = /* @__PURE__ */ __name(async (attrs) => {
+    const current = pos.next();
+    const existing = await readCheckpoint(current);
+    if (existing) {
+      if (existing.name !== "searchAttributes") {
+        throw new NonDeterminismError(runId, current, "searchAttributes", existing.name);
+      }
+      return;
+    }
+    await host.upsertSearchAttributes(runId, attrs);
+    await writeCheckpoint(instantCheckpoint({
+      runId,
+      seq: current,
+      name: "searchAttributes",
+      kind: "local",
+      output: attrs
+    }));
+  }, "upsertSearchAttributes");
   return {
     runId,
     step,
+    upsertSearchAttributes,
     transaction,
     callEntity,
     signalEntity,
@@ -3093,6 +3112,17 @@ var WorkflowEngine = class {
           priority
         }).catch(() => void 0));
       }, "startChild"),
+      // Shallow-merge into the run's searchAttributes (the ctx primitive makes this exactly-once).
+      upsertSearchAttributes: /* @__PURE__ */ __name(async (runId, attrs) => {
+        const run = await this.store.getRun(runId);
+        await this.store.updateRun(runId, {
+          searchAttributes: {
+            ...run?.searchAttributes ?? {},
+            ...attrs
+          },
+          updatedAt: /* @__PURE__ */ new Date()
+        });
+      }, "upsertSearchAttributes"),
       signalEntity: /* @__PURE__ */ __name((name, key, op, arg, reply) => {
         queueMicrotask(() => void this.entities.dispatch(name, key, op, arg, reply).catch(() => void 0));
       }, "signalEntity"),