@mastra/pg 1.27.2-alpha.0 → 1.28.0-alpha.2

package/dist/index.js

@@ -5024,6 +5024,7 @@ function rowToTask(row) {
 	const startedAt = row.startedAtZ || row.startedAt;
 	const suspendedAt = row.suspendedAtZ || row.suspendedAt;
 	const completedAt = row.completedAtZ || row.completedAt;
+	const leaseExpiresAt = row.leaseExpiresAtZ || row.leaseExpiresAt;
 	return {
 		id: row.id,
 		status: row.status,
@@ -5043,7 +5044,9 @@ function rowToTask(row) {
 		createdAt: createdAt instanceof Date ? createdAt : new Date(createdAt),
 		startedAt: startedAt ? startedAt instanceof Date ? startedAt : new Date(startedAt) : void 0,
 		suspendedAt: suspendedAt ? suspendedAt instanceof Date ? suspendedAt : new Date(suspendedAt) : void 0,
-		completedAt: completedAt ? completedAt instanceof Date ? completedAt : new Date(completedAt) : void 0
+		completedAt: completedAt ? completedAt instanceof Date ? completedAt : new Date(completedAt) : void 0,
+		ownerId: row.ownerId ?? void 0,
+		leaseExpiresAt: leaseExpiresAt ? leaseExpiresAt instanceof Date ? leaseExpiresAt : new Date(leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
@@ -5083,7 +5086,12 @@ var BackgroundTasksPG = class BackgroundTasksPG 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"
+			]
 		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -5213,7 +5221,10 @@ var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
 				suspendedAt: task.suspendedAt?.toISOString() ?? null,
 				suspendedAtZ: task.suspendedAt?.toISOString() ?? null,
 				completedAt: task.completedAt?.toISOString() ?? null,
-				completedAtZ: task.completedAt?.toISOString() ?? null
+				completedAtZ: task.completedAt?.toISOString() ?? null,
+				ownerId: task.ownerId ?? null,
+				leaseExpiresAt: task.leaseExpiresAt?.toISOString() ?? null,
+				leaseExpiresAtZ: task.leaseExpiresAt?.toISOString() ?? null
 			}
 		});
 	}
@@ -5259,14 +5270,34 @@ var BackgroundTasksPG = class BackgroundTasksPG extends BackgroundTasksStorage {
 			const val = update.completedAt?.toISOString() ?? null;
 			params.push(val, val);
 		}
+		if ("ownerId" in update) {
+			setClauses.push(`"ownerId" = $${paramIdx++}`);
+			params.push(update.ownerId ?? null);
+		}
+		if ("leaseExpiresAt" in update) {
+			setClauses.push(`"leaseExpiresAt" = $${paramIdx++}`);
+			setClauses.push(`"leaseExpiresAtZ" = $${paramIdx++}`);
+			const val = update.leaseExpiresAt?.toISOString() ?? null;
+			params.push(val, val);
+		}
 		if (setClauses.length === 0) return false;
 		const table = getTableName$4(getSchemaName$4(this.#schema));
 		params.push(taskId);
 		let where = `"id" = $${paramIdx++}`;
 		if (options?.expectedStatus) {
-			where += ` AND "status" = $${paramIdx}`;
+			where += ` AND "status" = $${paramIdx++}`;
 			params.push(options.expectedStatus);
 		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) where += ` AND "ownerId" IS NULL`;
+		else {
+			where += ` AND "ownerId" = $${paramIdx++}`;
+			params.push(options.expectedOwnerId);
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) where += ` AND "leaseExpiresAtZ" IS NULL`;
+		else {
+			where += ` AND "leaseExpiresAtZ" = $${paramIdx++}`;
+			params.push(options.expectedLeaseExpiresAt.toISOString());
+		}
 		return ((await this.#db.client.query(`UPDATE ${table} SET ${setClauses.join(", ")} WHERE ${where}`, params)).rowCount ?? 0) > 0;
 	}
 	async getTask(taskId) {
@@ -23007,7 +23038,7 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}, error);
 		}
 	}
-	async listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, threadId, status } = {}) {
+	async listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, threadId, status, summary } = {}) {
 		try {
 			const conditions = [];
 			const values = [];
@@ -23055,8 +23086,13 @@ var WorkflowsPG = class WorkflowsPG extends WorkflowsStorage {
 			}
 			const normalizedPerPage = usePagination ? normalizePerPage(perPage, Number.MAX_SAFE_INTEGER) : 0;
 			const offset = usePagination ? page * normalizedPerPage : void 0;
+			let selectList = "*";
+			if (summary) {
+				const snapshotJson = await this.#db.getColumnType(TABLE_WORKFLOW_SNAPSHOT, "snapshot") === "jsonb" ? "snapshot" : `regexp_replace(snapshot::text, '\\\\u(0000|[Dd][89A-Fa-f][0-9A-Fa-f]{2})', '', 'g')::jsonb`;
+				selectList = `workflow_name, run_id, "resourceId", "createdAt", "createdAtZ", "updatedAt", "updatedAtZ", jsonb_build_object('status', ${snapshotJson} -> 'status', 'timestamp', ${snapshotJson} -> 'timestamp') AS snapshot`;
+			}
 			const query = `
-          SELECT * FROM ${getTableName({
+          SELECT ${selectList} FROM ${getTableName({
 				indexName: TABLE_WORKFLOW_SNAPSHOT,
 				schemaName: getSchemaName(this.#schema)
 			})}