@mastra/mssql 1.9.0 → 1.10.0-alpha.0

package/dist/index.js

@@ -1780,7 +1780,9 @@ function rowToTask(row) {
 		createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
 		startedAt: row.startedAt ? row.startedAt instanceof Date ? row.startedAt : new Date(row.startedAt) : void 0,
 		suspendedAt: row.suspendedAt ? row.suspendedAt instanceof Date ? row.suspendedAt : new Date(row.suspendedAt) : void 0,
-		completedAt: row.completedAt ? row.completedAt instanceof Date ? row.completedAt : new Date(row.completedAt) : void 0
+		completedAt: row.completedAt ? row.completedAt instanceof Date ? row.completedAt : new Date(row.completedAt) : void 0,
+		ownerId: row.ownerId ?? void 0,
+		leaseExpiresAt: row.leaseExpiresAt ? row.leaseExpiresAt instanceof Date ? row.leaseExpiresAt : new Date(row.leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksStorage {
@@ -1817,7 +1819,12 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 		await this.db.alterTable({
 			tableName: TABLE_BACKGROUND_TASKS,
 			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS],
-			ifNotExists: ["suspend_payload", "suspendedAt"]
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
 		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -1894,7 +1901,9 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 				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
 			}
 		});
 	}
@@ -1934,18 +1943,37 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 			setClauses.push(`[completedAt] = @p${idx}`);
 			params[`p${idx++}`] = update.completedAt?.toISOString() ?? null;
 		}
+		if ("ownerId" in update) {
+			setClauses.push(`[ownerId] = @p${idx}`);
+			params[`p${idx++}`] = update.ownerId ?? null;
+		}
+		if ("leaseExpiresAt" in update) {
+			setClauses.push(`[leaseExpiresAt] = @p${idx}`);
+			params[`p${idx++}`] = update.leaseExpiresAt?.toISOString() ?? null;
+		}
 		if (setClauses.length === 0) return false;
 		setClauses.push(`[id] = [id]`);
 		params[`p${idx}`] = taskId;
 		const taskIdParam = `p${idx++}`;
-		let statusPredicate = "";
+		const predicates = [];
 		if (options?.expectedStatus) {
 			params[`p${idx}`] = options.expectedStatus;
-			statusPredicate = ` AND [status] = @p${idx}`;
+			predicates.push(`[status] = @p${idx++}`);
+		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates.push(`[ownerId] IS NULL`);
+		else {
+			params[`p${idx}`] = options.expectedOwnerId;
+			predicates.push(`[ownerId] = @p${idx++}`);
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates.push(`[leaseExpiresAt] IS NULL`);
+		else {
+			params[`p${idx}`] = options.expectedLeaseExpiresAt.toISOString();
+			predicates.push(`[leaseExpiresAt] = @p${idx++}`);
 		}
+		const predicateSql = predicates.length > 0 ? ` AND ${predicates.join(" AND ")}` : "";
 		const request = this.pool.request();
 		for (const [name, value] of Object.entries(params)) request.input(name, value);
-		return ((await request.query(`UPDATE ${this.tableName()} SET ${setClauses.join(", ")} WHERE [id] = @${taskIdParam}${statusPredicate}`)).rowsAffected[0] ?? 0) > 0;
+		return ((await request.query(`UPDATE ${this.tableName()} SET ${setClauses.join(", ")} WHERE [id] = @${taskIdParam}${predicateSql}`)).rowsAffected[0] ?? 0) > 0;
 	}
 	async getTask(taskId) {
 		const request = this.pool.request();