@mastra/mssql 1.10.6 → 1.11.0

package/dist/index.js

@@ -1778,10 +1778,13 @@ function rowToTask(row) {
 		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,
 		ownerId: row.ownerId ?? void 0,
-		leaseExpiresAt: row.leaseExpiresAt ? row.leaseExpiresAt instanceof Date ? row.leaseExpiresAt : new Date(row.leaseExpiresAt) : void 0
+		leaseExpiresAt: row.leaseExpiresAt ? row.leaseExpiresAt instanceof Date ? row.leaseExpiresAt : new Date(row.leaseExpiresAt) : void 0,
+		expiresAt: row.expiresAt ? row.expiresAt instanceof Date ? row.expiresAt : new Date(row.expiresAt) : void 0,
+		metadata: parseJson(row.metadata)
 	};
 }
 var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksStorage {
+	supportsTaskExpiry = true;
 	pool;
 	db;
 	schema;
@@ -1819,7 +1822,9 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 				"suspend_payload",
 				"suspendedAt",
 				"ownerId",
-				"leaseExpiresAt"
+				"leaseExpiresAt",
+				"expiresAt",
+				"metadata"
 			]
 		});
 		await this.createDefaultIndexes();
@@ -1899,7 +1904,9 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 				suspendedAt: task.suspendedAt?.toISOString() ?? null,
 				completedAt: task.completedAt?.toISOString() ?? null,
 				ownerId: task.ownerId ?? null,
-				leaseExpiresAt: task.leaseExpiresAt?.toISOString() ?? null
+				leaseExpiresAt: task.leaseExpiresAt?.toISOString() ?? null,
+				expiresAt: task.expiresAt?.toISOString() ?? null,
+				metadata: serializeJson(task.metadata)
 			}
 		});
 	}
@@ -1947,6 +1954,14 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 			setClauses.push(`[leaseExpiresAt] = @p${idx}`);
 			params[`p${idx++}`] = update.leaseExpiresAt?.toISOString() ?? null;
 		}
+		if ("expiresAt" in update) {
+			setClauses.push(`[expiresAt] = @p${idx}`);
+			params[`p${idx++}`] = update.expiresAt?.toISOString() ?? null;
+		}
+		if ("metadata" in update) {
+			setClauses.push(`[metadata] = @p${idx}`);
+			params[`p${idx++}`] = serializeJson(update.metadata);
+		}
 		if (setClauses.length === 0) return false;
 		setClauses.push(`[id] = [id]`);
 		params[`p${idx}`] = taskId;
@@ -1970,6 +1985,13 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 				predicates.push(`[leaseExpiresAt] = @p${idx++}`);
 			}
 		}
+		if (options?.expectedSuspendedAt !== void 0) {
+			if (options.expectedSuspendedAt === null) predicates.push(`[suspendedAt] IS NULL`);
+			else {
+				params[`p${idx}`] = options.expectedSuspendedAt.toISOString();
+				predicates.push(`[suspendedAt] = @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);
@@ -2020,7 +2042,8 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 			params[`p${idx}`] = filter.toolCallId;
 			conditions.push(`[tool_call_id] = @p${idx++}`);
 		}
-		const dateCol = filter.dateFilterBy === "startedAt" ? "[startedAt]" : filter.dateFilterBy === "suspendedAt" ? "[suspendedAt]" : filter.dateFilterBy === "completedAt" ? "[completedAt]" : "[createdAt]";
+		const dateCol = filter.dateFilterBy === "startedAt" ? "[startedAt]" : filter.dateFilterBy === "suspendedAt" ? "[suspendedAt]" : filter.dateFilterBy === "completedAt" ? "[completedAt]" : filter.dateFilterBy === "expiresAt" ? "[expiresAt]" : "[createdAt]";
+		if (filter.hasExpiresAt !== void 0) conditions.push(filter.hasExpiresAt ? "[expiresAt] IS NOT NULL" : "[expiresAt] IS NULL");
 		if (filter.fromDate) {
 			params[`p${idx}`] = filter.fromDate.toISOString();
 			conditions.push(`${dateCol} >= @p${idx++}`);
@@ -2070,7 +2093,8 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends BackgroundTasksSto
 			});
 			conditions.push(`[status] IN (${placeholders.join(", ")})`);
 		}
-		const dateCol = filter.dateFilterBy === "startedAt" ? "[startedAt]" : filter.dateFilterBy === "suspendedAt" ? "[suspendedAt]" : filter.dateFilterBy === "completedAt" ? "[completedAt]" : "[createdAt]";
+		const dateCol = filter.dateFilterBy === "startedAt" ? "[startedAt]" : filter.dateFilterBy === "suspendedAt" ? "[suspendedAt]" : filter.dateFilterBy === "completedAt" ? "[completedAt]" : filter.dateFilterBy === "expiresAt" ? "[expiresAt]" : "[createdAt]";
+		if (filter.hasExpiresAt !== void 0) conditions.push(filter.hasExpiresAt ? "[expiresAt] IS NOT NULL" : "[expiresAt] IS NULL");
 		if (filter.fromDate) {
 			params[`p${idx}`] = filter.fromDate.toISOString();
 			conditions.push(`${dateCol} >= @p${idx++}`);