@mastra/spanner 1.8.2 → 1.9.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-spanner
 description: Documentation for @mastra/spanner. Use when working with @mastra/spanner APIs, configuration, or implementation.
 metadata:
   package: "@mastra/spanner"
-  version: "1.8.2"
+  version: "1.9.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.8.2",
+  "version": "1.9.0",
   "package": "@mastra/spanner",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-storage-retention.md

@@ -202,7 +202,7 @@ async function retentionTick() {
 }
 ```
 
-You can also cancel a long-running prune with an `AbortSignal`: the loop stops between batches and returns partial results with `done: false`, so the next run resumes cleanly.
+You can also cancel a long-running prune with an `AbortSignal`: the loop stops between batches and returns partial results with `done: false`. The next run resumes cleanly.
 
 ## ClickHouse native TTL
 

package/dist/index.cjs

@@ -1860,7 +1860,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 _mastra_core_storage.BackgroundTasksStorage {
@@ -1886,6 +1888,16 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends _mastra_core_s
 			tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS,
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_BACKGROUND_TASKS]
 		});
+		await this.db.alterTable({
+			tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS,
+			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_BACKGROUND_TASKS],
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
+		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
@@ -1964,7 +1976,9 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends _mastra_core_s
 				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
 			}
 		});
 	}
@@ -1978,16 +1992,32 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends _mastra_core_s
 		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;