@dudousxd/nestjs-durable-core 0.36.0 → 0.38.0

package/dist/index.cjs

@@ -1434,9 +1434,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,
@@ -3165,6 +3184,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"),