@mastra/mysql 0.10.2 → 0.11.0-alpha.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-mysql
 description: Documentation for @mastra/mysql. Use when working with @mastra/mysql APIs, configuration, or implementation.
 metadata:
   package: "@mastra/mysql"
-  version: "0.10.2"
+  version: "0.11.0-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.10.2",
+  "version": "0.11.0-alpha.0",
   "package": "@mastra/mysql",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -1407,7 +1407,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 _mastra_core_storage.BackgroundTasksStorage {
@@ -1428,6 +1430,16 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends _mastra_core_stora
 			tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS,
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_BACKGROUND_TASKS]
 		});
+		await this.operations.alterTable({
+			tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS,
+			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_BACKGROUND_TASKS],
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
+		});
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
@@ -1479,7 +1491,7 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends _mastra_core_stora
 		await this.operations.clearTable({ tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS });
 	}
 	async createTask(task) {
-		await this.pool.execute(`INSERT INTO ${formatTableName(_mastra_core_storage.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(_mastra_core_storage.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,
@@ -1498,7 +1510,9 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends _mastra_core_stora
 			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) {
@@ -1536,11 +1550,33 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends _mastra_core_stora
 			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(_mastra_core_storage.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(_mastra_core_storage.TABLE_BACKGROUND_TASKS)} SET ${setClauses.join(", ")} WHERE ${quoteIdentifier("id", "column name")} = ?${predicateSql}`, params);
 		return result.affectedRows > 0;
 	}
 	async getTask(taskId) {