@mastra/cloudflare-d1 1.3.2 → 1.4.0

package/dist/index.js

@@ -666,7 +666,9 @@ function rowToTask(row) {
 		createdAt: new Date(row.createdAt),
 		startedAt: row.startedAt ? new Date(row.startedAt) : void 0,
 		suspendedAt: row.suspendedAt ? new Date(row.suspendedAt) : void 0,
-		completedAt: row.completedAt ? new Date(row.completedAt) : void 0
+		completedAt: row.completedAt ? new Date(row.completedAt) : void 0,
+		ownerId: row.ownerId || void 0,
+		leaseExpiresAt: row.leaseExpiresAt ? new Date(row.leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
@@ -684,7 +686,12 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
 		await this.#db.alterTable({
 			tableName: TABLE_BACKGROUND_TASKS,
 			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS],
-			ifNotExists: ["suspend_payload", "suspendedAt"]
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
 		});
 	}
 	async dangerouslyClearAll() {
@@ -711,7 +718,9 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
 			"createdAt",
 			"startedAt",
 			"suspendedAt",
-			"completedAt"
+			"completedAt",
+			"ownerId",
+			"leaseExpiresAt"
 		], [
 			task.id,
 			task.toolCallId,
@@ -731,7 +740,9 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
 			task.createdAt.toISOString(),
 			task.startedAt?.toISOString() ?? null,
 			task.suspendedAt?.toISOString() ?? null,
-			task.completedAt?.toISOString() ?? null
+			task.completedAt?.toISOString() ?? null,
+			task.ownerId ?? null,
+			task.leaseExpiresAt?.toISOString() ?? null
 		]).build();
 		await this.#db.executeQuery({
 			sql,
@@ -773,13 +784,34 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
 			sets.push("completedAt = ?");
 			params.push(update.completedAt?.toISOString() ?? null);
 		}
+		if ("ownerId" in update) {
+			sets.push("ownerId = ?");
+			params.push(update.ownerId ?? null);
+		}
+		if ("leaseExpiresAt" in update) {
+			sets.push("leaseExpiresAt = ?");
+			params.push(update.leaseExpiresAt?.toISOString() ?? null);
+		}
 		if (sets.length === 0) return false;
 		params.push(taskId);
-		const statusPredicate = options?.expectedStatus ? " AND status = ?" : "";
-		if (options?.expectedStatus) params.push(options.expectedStatus);
+		let predicates = "";
+		if (options?.expectedStatus) {
+			predicates += " AND status = ?";
+			params.push(options.expectedStatus);
+		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates += " AND ownerId IS NULL";
+		else {
+			predicates += " AND ownerId = ?";
+			params.push(options.expectedOwnerId);
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates += " AND leaseExpiresAt IS NULL";
+		else {
+			predicates += " AND leaseExpiresAt = ?";
+			params.push(options.expectedLeaseExpiresAt.toISOString());
+		}
 		const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
 		const result = await this.#db.executeQuery({
-			sql: `UPDATE ${fullTableName} SET ${sets.join(", ")} WHERE id = ?${statusPredicate} RETURNING id`,
+			sql: `UPDATE ${fullTableName} SET ${sets.join(", ")} WHERE id = ?${predicates} RETURNING id`,
 			params
 		});
 		return Array.isArray(result) && result.length > 0;