@mastra/spanner 1.8.2 → 1.9.0

package/dist/index.js

@@ -1859,7 +1859,9 @@ function rowToTask(row) {
 		createdAt: transformed.createdAt instanceof Date ? transformed.createdAt : new Date(transformed.createdAt),
 		startedAt: transformed.startedAt ? transformed.startedAt instanceof Date ? transformed.startedAt : new Date(transformed.startedAt) : void 0,
 		suspendedAt: transformed.suspendedAt ? transformed.suspendedAt instanceof Date ? transformed.suspendedAt : new Date(transformed.suspendedAt) : void 0,
-		completedAt: transformed.completedAt ? transformed.completedAt instanceof Date ? transformed.completedAt : new Date(transformed.completedAt) : void 0
+		completedAt: transformed.completedAt ? transformed.completedAt instanceof Date ? transformed.completedAt : new Date(transformed.completedAt) : void 0,
+		ownerId: transformed.ownerId != null ? String(transformed.ownerId) : void 0,
+		leaseExpiresAt: transformed.leaseExpiresAt ? transformed.leaseExpiresAt instanceof Date ? transformed.leaseExpiresAt : new Date(transformed.leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTasksStorage {
@@ -1885,6 +1887,16 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTask
 			tableName: TABLE_BACKGROUND_TASKS,
 			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS]
 		});
+		await this.db.alterTable({
+			tableName: TABLE_BACKGROUND_TASKS,
+			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS],
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
+		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
@@ -1963,7 +1975,9 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTask
 				createdAt: task.createdAt,
 				startedAt: task.startedAt ?? null,
 				suspendedAt: task.suspendedAt ?? null,
-				completedAt: task.completedAt ?? null
+				completedAt: task.completedAt ?? null,
+				ownerId: task.ownerId ?? null,
+				leaseExpiresAt: task.leaseExpiresAt ?? null
 			}
 		});
 	}
@@ -1977,16 +1991,32 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTask
 		if ("startedAt" in update) data.startedAt = update.startedAt ?? null;
 		if ("suspendedAt" in update) data.suspendedAt = update.suspendedAt ?? null;
 		if ("completedAt" in update) data.completedAt = update.completedAt ?? null;
+		if ("ownerId" in update) data.ownerId = update.ownerId ?? null;
+		if ("leaseExpiresAt" in update) data.leaseExpiresAt = update.leaseExpiresAt ?? null;
 		if (Object.keys(data).length === 0) return false;
 		const setClauses = Object.keys(data).map((field) => `${quoteIdent(field, "column name")} = @${field}`);
 		const params = {
 			...data,
-			id: taskId,
-			expectedStatus: options?.expectedStatus
+			id: taskId
 		};
-		const statusPredicate = options?.expectedStatus ? " AND status = @expectedStatus" : "";
+		const predicates = [];
+		if (options?.expectedStatus) {
+			predicates.push("status = @expectedStatus");
+			params.expectedStatus = options.expectedStatus;
+		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates.push("ownerId IS NULL");
+		else {
+			predicates.push("ownerId = @expectedOwnerId");
+			params.expectedOwnerId = options.expectedOwnerId;
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates.push("leaseExpiresAt IS NULL");
+		else {
+			predicates.push("leaseExpiresAt = @expectedLeaseExpiresAt");
+			params.expectedLeaseExpiresAt = options.expectedLeaseExpiresAt;
+		}
+		const predicateSql = predicates.length > 0 ? ` AND ${predicates.join(" AND ")}` : "";
 		const [rowCount] = await this.database.runUpdate({
-			sql: `UPDATE ${this.tableName()} SET ${setClauses.join(", ")} WHERE id = @id${statusPredicate}`,
+			sql: `UPDATE ${this.tableName()} SET ${setClauses.join(", ")} WHERE id = @id${predicateSql}`,
 			params
 		});
 		return rowCount > 0;