@mastra/upstash 1.4.7 → 1.5.0-alpha.1

package/dist/index.js

@@ -188,7 +188,9 @@ function toStorageRecord(task) {
 		createdAt: task.createdAt.toISOString(),
 		startedAt: task.startedAt?.toISOString() ?? null,
 		suspendedAt: task.suspendedAt?.toISOString() ?? null,
-		completedAt: task.completedAt?.toISOString() ?? null
+		completedAt: task.completedAt?.toISOString() ?? null,
+		ownerId: task.ownerId ?? null,
+		leaseExpiresAt: task.leaseExpiresAt?.toISOString() ?? null
 	};
 }
 function fromStorageRecord(record) {
@@ -211,7 +213,9 @@ function fromStorageRecord(record) {
 		createdAt: new Date(record.createdAt),
 		startedAt: record.startedAt ? new Date(record.startedAt) : void 0,
 		suspendedAt: record.suspendedAt ? new Date(record.suspendedAt) : void 0,
-		completedAt: record.completedAt ? new Date(record.completedAt) : void 0
+		completedAt: record.completedAt ? new Date(record.completedAt) : void 0,
+		ownerId: record.ownerId ?? void 0,
+		leaseExpiresAt: record.leaseExpiresAt ? new Date(record.leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksUpstash = class extends BackgroundTasksStorage {
@@ -240,6 +244,8 @@ var BackgroundTasksUpstash = class extends BackgroundTasksStorage {
 		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 ("ownerId" in update) patch.ownerId = update.ownerId ?? null;
+		if ("leaseExpiresAt" in update) patch.leaseExpiresAt = update.leaseExpiresAt?.toISOString() ?? null;
 		if (Object.keys(patch).length === 0) return false;
 		const key = getKey(TABLE_BACKGROUND_TASKS, { id: taskId });
 		const result = await this.client.eval(`
@@ -247,11 +253,28 @@ var BackgroundTasksUpstash = class extends BackgroundTasksStorage {
         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 function normalize(value)
+          if value == nil or value == cjson.null then return nil end
+          return value
+        end
+        if ARGV[3] == '1' then
+          if normalize(record.ownerId) ~= normalize(cjson.decode(ARGV[4])) then return 0 end
+        end
+        if ARGV[5] == '1' then
+          if normalize(record.leaseExpiresAt) ~= normalize(cjson.decode(ARGV[6])) then return 0 end
+        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)]);
+      `, [key], [
+			options?.expectedStatus ?? "",
+			JSON.stringify(patch),
+			options?.expectedOwnerId !== void 0 ? "1" : "0",
+			JSON.stringify(options?.expectedOwnerId ?? null),
+			options?.expectedLeaseExpiresAt !== void 0 ? "1" : "0",
+			JSON.stringify(options?.expectedLeaseExpiresAt?.toISOString() ?? null)
+		]);
 		return Number(result) === 1;
 	}
 	async getTask(taskId) {