@mastra/upstash 1.4.2 → 1.4.3

package/dist/index.js

@@ -230,20 +230,29 @@ var BackgroundTasksUpstash = class extends BackgroundTasksStorage {
 		const { key, processedRecord } = processRecord(TABLE_BACKGROUND_TASKS, toStorageRecord(task));
 		await this.client.set(key, processedRecord);
 	}
-	async updateTask(taskId, update) {
-		const existing = await this.getTask(taskId);
-		if (!existing) return;
-		const merged = { ...existing };
-		if ("status" in update) merged.status = update.status;
-		if ("result" in update) merged.result = update.result;
-		if ("error" in update) merged.error = update.error;
-		if ("suspendPayload" in update) merged.suspendPayload = update.suspendPayload;
-		if ("retryCount" in update) merged.retryCount = update.retryCount;
-		if ("startedAt" in update) merged.startedAt = update.startedAt;
-		if ("suspendedAt" in update) merged.suspendedAt = update.suspendedAt;
-		if ("completedAt" in update) merged.completedAt = update.completedAt;
-		const { key, processedRecord } = processRecord(TABLE_BACKGROUND_TASKS, toStorageRecord(merged));
-		await this.client.set(key, processedRecord);
+	async updateTask(taskId, update, options) {
+		const patch = {};
+		if ("status" in update) patch.status = update.status;
+		if ("result" in update) patch.result = update.result ?? null;
+		if ("error" in update) patch.error = update.error ?? null;
+		if ("suspendPayload" in update) patch.suspend_payload = update.suspendPayload ?? null;
+		if ("retryCount" in update) patch.retry_count = update.retryCount;
+		if ("startedAt" in update) patch.startedAt = update.startedAt?.toISOString() ?? null;
+		if ("suspendedAt" in update) patch.suspendedAt = update.suspendedAt?.toISOString() ?? null;
+		if ("completedAt" in update) patch.completedAt = update.completedAt?.toISOString() ?? null;
+		if (Object.keys(patch).length === 0) return false;
+		const key = getKey(TABLE_BACKGROUND_TASKS, { id: taskId });
+		const result = await this.client.eval(`
+        local existing = redis.call('GET', KEYS[1])
+        if not existing then return 0 end
+        local record = cjson.decode(existing)
+        if ARGV[1] ~= '' and record.status ~= ARGV[1] then return 0 end
+        local patch = cjson.decode(ARGV[2])
+        for field, value in pairs(patch) do record[field] = value end
+        redis.call('SET', KEYS[1], cjson.encode(record))
+        return 1
+      `, [key], [options?.expectedStatus ?? "", JSON.stringify(patch)]);
+		return Number(result) === 1;
 	}
 	async getTask(taskId) {
 		const key = getKey(TABLE_BACKGROUND_TASKS, { id: taskId });