@mastra/upstash 1.4.1 → 1.4.2

package/dist/index.js

@@ -1728,10 +1728,11 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
 				run_id: runId
 			});
 			const now = (/* @__PURE__ */ new Date()).toISOString();
-			const resultJson = await this.client.eval(`
+			const luaScript = `
         local key = KEYS[1]
         local optsJson = ARGV[1]
         local now = ARGV[2]
+        local expectedStatusJson = ARGV[3]
 
         -- Get existing data
         local existing = redis.call('GET', key)
@@ -1751,6 +1752,21 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
           return nil
         end
 
+        -- Compare-and-set guard: bail out unless the persisted status is one the caller expects.
+        -- This runs inside the same script as the write, so the check and the write are atomic.
+        if expectedStatusJson ~= '' then
+          local expected = cjson.decode(expectedStatusJson)
+          local matched = false
+          for _, status in ipairs(expected) do
+            if snapshot.status == status then
+              matched = true
+            end
+          end
+          if not matched then
+            return nil
+          end
+        end
+
         -- Merge the new options with the existing snapshot
         local opts = cjson.decode(optsJson)
         for k, v in pairs(opts) do
@@ -1766,7 +1782,14 @@ var WorkflowsUpstash = class extends WorkflowsStorage {
 
         -- Return the full updated data
         return cjson.encode(data)
-      `, [key], [JSON.stringify(opts), now]);
+      `;
+			const { expectedStatus, ...state } = opts;
+			const expectedStatusJson = expectedStatus === void 0 ? "" : JSON.stringify(Array.isArray(expectedStatus) ? expectedStatus : [expectedStatus]);
+			const resultJson = await this.client.eval(luaScript, [key], [
+				JSON.stringify(state),
+				now,
+				expectedStatusJson
+			]);
 			if (!resultJson) return;
 			let data;
 			if (typeof resultJson === "string") data = JSON.parse(resultJson);