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

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.27.2-alpha.1"
+  version: "1.28.0-alpha.2"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.27.2-alpha.1",
+  "version": "1.28.0-alpha.2",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -5048,6 +5048,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,
@@ -5067,7 +5068,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 _mastra_core_storage.BackgroundTasksStorage {
@@ -5107,7 +5110,12 @@ var BackgroundTasksPG = class BackgroundTasksPG extends _mastra_core_storage.Bac
 		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"]
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
 		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -5237,7 +5245,10 @@ var BackgroundTasksPG = class BackgroundTasksPG extends _mastra_core_storage.Bac
 				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
 			}
 		});
 	}
@@ -5283,14 +5294,34 @@ var BackgroundTasksPG = class BackgroundTasksPG extends _mastra_core_storage.Bac
 			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) {