@mastra/mysql 0.10.2 → 0.11.0-alpha.0

package/dist/index.js

@@ -1406,7 +1406,9 @@ function rowToTask(row) {
 		createdAt: parseDateTime(row.createdAt),
 		startedAt: parseDateTime(row.startedAt),
 		suspendedAt: parseDateTime(row.suspendedAt),
-		completedAt: parseDateTime(row.completedAt)
+		completedAt: parseDateTime(row.completedAt),
+		ownerId: row.ownerId != null ? String(row.ownerId) : void 0,
+		leaseExpiresAt: parseDateTime(row.leaseExpiresAt)
 	};
 }
 var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksStorage {
@@ -1427,6 +1429,16 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 			tableName: TABLE_BACKGROUND_TASKS,
 			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS]
 		});
+		await this.operations.alterTable({
+			tableName: TABLE_BACKGROUND_TASKS,
+			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS],
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
+		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
@@ -1478,7 +1490,7 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 		await this.operations.clearTable({ tableName: TABLE_BACKGROUND_TASKS });
 	}
 	async createTask(task) {
-		await this.pool.execute(`INSERT INTO ${formatTableName(TABLE_BACKGROUND_TASKS)} (${quoteIdentifier("id", "column name")}, ${quoteIdentifier("tool_call_id", "column name")}, ${quoteIdentifier("tool_name", "column name")}, ${quoteIdentifier("agent_id", "column name")}, ${quoteIdentifier("thread_id", "column name")}, ${quoteIdentifier("resource_id", "column name")}, ${quoteIdentifier("run_id", "column name")}, ${quoteIdentifier("status", "column name")}, ${quoteIdentifier("args", "column name")}, ${quoteIdentifier("result", "column name")}, ${quoteIdentifier("error", "column name")}, ${quoteIdentifier("suspend_payload", "column name")}, ${quoteIdentifier("retry_count", "column name")}, ${quoteIdentifier("max_retries", "column name")}, ${quoteIdentifier("timeout_ms", "column name")}, ${quoteIdentifier("createdAt", "column name")}, ${quoteIdentifier("startedAt", "column name")}, ${quoteIdentifier("suspendedAt", "column name")}, ${quoteIdentifier("completedAt", "column name")}) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
+		await this.pool.execute(`INSERT INTO ${formatTableName(TABLE_BACKGROUND_TASKS)} (${quoteIdentifier("id", "column name")}, ${quoteIdentifier("tool_call_id", "column name")}, ${quoteIdentifier("tool_name", "column name")}, ${quoteIdentifier("agent_id", "column name")}, ${quoteIdentifier("thread_id", "column name")}, ${quoteIdentifier("resource_id", "column name")}, ${quoteIdentifier("run_id", "column name")}, ${quoteIdentifier("status", "column name")}, ${quoteIdentifier("args", "column name")}, ${quoteIdentifier("result", "column name")}, ${quoteIdentifier("error", "column name")}, ${quoteIdentifier("suspend_payload", "column name")}, ${quoteIdentifier("retry_count", "column name")}, ${quoteIdentifier("max_retries", "column name")}, ${quoteIdentifier("timeout_ms", "column name")}, ${quoteIdentifier("createdAt", "column name")}, ${quoteIdentifier("startedAt", "column name")}, ${quoteIdentifier("suspendedAt", "column name")}, ${quoteIdentifier("completedAt", "column name")}, ${quoteIdentifier("ownerId", "column name")}, ${quoteIdentifier("leaseExpiresAt", "column name")}) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
 			task.id,
 			task.toolCallId,
 			task.toolName,
@@ -1497,7 +1509,9 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 			transformToSqlValue(task.createdAt),
 			transformToSqlValue(task.startedAt),
 			transformToSqlValue(task.suspendedAt),
-			transformToSqlValue(task.completedAt)
+			transformToSqlValue(task.completedAt),
+			task.ownerId ?? null,
+			transformToSqlValue(task.leaseExpiresAt)
 		]);
 	}
 	async updateTask(taskId, update, options) {
@@ -1535,11 +1549,33 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 			setClauses.push(`${quoteIdentifier("completedAt", "column name")} = ?`);
 			params.push(transformToSqlValue(update.completedAt));
 		}
+		if ("ownerId" in update) {
+			setClauses.push(`${quoteIdentifier("ownerId", "column name")} = ?`);
+			params.push(update.ownerId ?? null);
+		}
+		if ("leaseExpiresAt" in update) {
+			setClauses.push(`${quoteIdentifier("leaseExpiresAt", "column name")} = ?`);
+			params.push(transformToSqlValue(update.leaseExpiresAt));
+		}
 		if (setClauses.length === 0) return false;
 		params.push(taskId);
-		const statusPredicate = options?.expectedStatus ? ` AND ${quoteIdentifier("status", "column name")} = ?` : "";
-		if (options?.expectedStatus) params.push(options.expectedStatus);
-		const [result] = await this.pool.execute(`UPDATE ${formatTableName(TABLE_BACKGROUND_TASKS)} SET ${setClauses.join(", ")} WHERE ${quoteIdentifier("id", "column name")} = ?${statusPredicate}`, params);
+		const predicates = [];
+		if (options?.expectedStatus) {
+			predicates.push(`${quoteIdentifier("status", "column name")} = ?`);
+			params.push(options.expectedStatus);
+		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates.push(`${quoteIdentifier("ownerId", "column name")} IS NULL`);
+		else {
+			predicates.push(`${quoteIdentifier("ownerId", "column name")} = ?`);
+			params.push(options.expectedOwnerId);
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} IS NULL`);
+		else {
+			predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} = ?`);
+			params.push(transformToSqlValue(options.expectedLeaseExpiresAt));
+		}
+		const predicateSql = predicates.length > 0 ? ` AND ${predicates.join(" AND ")}` : "";
+		const [result] = await this.pool.execute(`UPDATE ${formatTableName(TABLE_BACKGROUND_TASKS)} SET ${setClauses.join(", ")} WHERE ${quoteIdentifier("id", "column name")} = ?${predicateSql}`, params);
 		return result.affectedRows > 0;
 	}
 	async getTask(taskId) {