@mastra/dynamodb 1.3.3-alpha.1 → 1.4.0-alpha.0

package/dist/index.js

@@ -169,6 +169,14 @@ const backgroundTaskEntity = new Entity({
 		completedAtIso: {
 			type: "string",
 			required: false
+		},
+		ownerId: {
+			type: "string",
+			required: false
+		},
+		leaseExpiresAtIso: {
+			type: "string",
+			required: false
 		}
 	},
 	indexes: {
@@ -1224,7 +1232,9 @@ function toElectroRecord(task) {
 		createdAt: task.createdAt.toISOString(),
 		startedAtIso: task.startedAt?.toISOString(),
 		suspendedAtIso: task.suspendedAt?.toISOString(),
-		completedAtIso: task.completedAt?.toISOString()
+		completedAtIso: task.completedAt?.toISOString(),
+		ownerId: task.ownerId ?? void 0,
+		leaseExpiresAtIso: task.leaseExpiresAt?.toISOString()
 	};
 }
 function fromElectroRecord(data) {
@@ -1257,7 +1267,9 @@ function fromElectroRecord(data) {
 		createdAt: asDate(data.createdAt) ?? /* @__PURE__ */ new Date(),
 		startedAt: asDate(data.startedAtIso),
 		suspendedAt: asDate(data.suspendedAtIso),
-		completedAt: asDate(data.completedAtIso)
+		completedAt: asDate(data.completedAtIso),
+		ownerId: data.ownerId != null && data.ownerId !== "" ? String(data.ownerId) : void 0,
+		leaseExpiresAt: asDate(data.leaseExpiresAtIso)
 	};
 }
 var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
@@ -1300,6 +1312,10 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 			else setFields.suspendedAtIso = update.suspendedAt.toISOString();
 			if ("completedAt" in update) if (update.completedAt === void 0 || update.completedAt === null) removeFields.push("completedAtIso");
 			else setFields.completedAtIso = update.completedAt.toISOString();
+			if ("ownerId" in update) if (update.ownerId === void 0 || update.ownerId === null) removeFields.push("ownerId");
+			else setFields.ownerId = update.ownerId;
+			if ("leaseExpiresAt" in update) if (update.leaseExpiresAt === void 0 || update.leaseExpiresAt === null) removeFields.push("leaseExpiresAtIso");
+			else setFields.leaseExpiresAtIso = update.leaseExpiresAt.toISOString();
 			if (Object.keys(setFields).length === 0 && removeFields.length === 0) return false;
 			let op = this.service.entities.background_task.patch({
 				entity: ENTITY,
@@ -1307,10 +1323,18 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 			});
 			if (Object.keys(setFields).length > 0) op = op.set(setFields);
 			if (removeFields.length > 0) op = op.remove(removeFields);
-			if (options?.expectedStatus) op = op.where(({ status }, { eq }) => eq(status, options.expectedStatus));
+			if (options?.expectedStatus !== void 0) op = op.where(({ status }, { eq }) => eq(status, options.expectedStatus));
+			if (options?.expectedOwnerId !== void 0) {
+				const expectedOwnerId = options.expectedOwnerId;
+				op = op.where(({ ownerId }, { eq, notExists }) => expectedOwnerId === null ? notExists(ownerId) : eq(ownerId, expectedOwnerId));
+			}
+			if (options?.expectedLeaseExpiresAt !== void 0) {
+				const expectedLeaseExpiresAt = options.expectedLeaseExpiresAt;
+				op = op.where(({ leaseExpiresAtIso }, { eq, notExists }) => expectedLeaseExpiresAt === null ? notExists(leaseExpiresAtIso) : eq(leaseExpiresAtIso, expectedLeaseExpiresAt.toISOString()));
+			}
 			await op.go();
 		} catch (error) {
-			if (options?.expectedStatus && error?.cause?.name === "ConditionalCheckFailedException") return false;
+			if ((options?.expectedStatus !== void 0 || options?.expectedOwnerId !== void 0 || options?.expectedLeaseExpiresAt !== void 0) && error?.cause?.name === "ConditionalCheckFailedException") return false;
 			throw new MastraError({
 				id: createStorageErrorId("DYNAMODB", "BACKGROUND_TASKS_UPDATE", "FAILED"),
 				domain: ErrorDomain.STORAGE,