@mastra/mysql 0.14.1-alpha.0 → 0.14.1

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.14.1-alpha.0"
+  version: "0.14.1"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

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

package/dist/index.cjs

@@ -11336,6 +11336,7 @@ var WorkflowDefinitionsMySQL = class extends _mastra_core_storage.WorkflowDefini
 	}
 };
 // src/storage/domains/workflows/index.ts
+const DUPLICATE_KEY_ERROR_CODES = /* @__PURE__ */ new Set([1062, "ER_DUP_ENTRY"]);
 function parseSnapshot(snapshot) {
 	if (typeof snapshot === "string") try {
 		return JSON.parse(snapshot);
@@ -11419,6 +11420,9 @@ var WorkflowsMySQL = class WorkflowsMySQL extends _mastra_core_storage.Workflows
 		this.#skipDefaultIndexes = skipDefaultIndexes;
 		this.#indexes = indexes?.filter((idx) => WorkflowsMySQL.MANAGED_TABLES.includes(idx.table));
 	}
+	supportsGuardedSnapshotWrites() {
+		return true;
+	}
 	supportsConcurrentUpdates() {
 		return true;
 	}
@@ -11703,25 +11707,45 @@ var WorkflowsMySQL = class WorkflowsMySQL extends _mastra_core_storage.Workflows
 			connection.release();
 		}
 	}
-	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, fence }) {
+	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, fence, expectedStatus }) {
 		const now = /* @__PURE__ */ new Date();
+		const expectedStatuses = expectedStatus === void 0 ? void 0 : Array.isArray(expectedStatus) ? expectedStatus : [expectedStatus];
+		const guard = expectedStatuses === void 0 ? "" : expectedStatuses.length === 0 ? " AND FALSE" : ` AND JSON_UNQUOTE(JSON_EXTRACT(${quoteIdentifier("snapshot", "column name")}, '$.status')) IN (${expectedStatuses.map(() => "?").join(", ")})`;
 		const createdAtValue = createdAt ?? now;
 		const updatedAtValue = updatedAt ?? now;
 		const tableName = formatTableName(_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT);
 		try {
-			await withRunFence(this.pool, RUN_OWNERS_TABLE, (0, _mastra_core_storage.resolveRunFence)(this, fence, runId), "persistWorkflowSnapshot", async (q) => {
-				const [updateResult] = await q.execute(`UPDATE ${tableName}
+			return await withRunFence(this.pool, RUN_OWNERS_TABLE, (0, _mastra_core_storage.resolveRunFence)(this, fence, runId), "persistWorkflowSnapshot", async (q) => {
+				const update = () => q.execute(`UPDATE ${tableName}
            SET ${quoteIdentifier("resourceId", "column name")} = ?,
                ${quoteIdentifier("snapshot", "column name")} = ?,
                ${quoteIdentifier("updatedAt", "column name")} = ?
-           WHERE ${quoteIdentifier("workflow_name", "column name")} = ? AND ${quoteIdentifier("run_id", "column name")} = ?`, [
+           WHERE ${quoteIdentifier("workflow_name", "column name")} = ? AND ${quoteIdentifier("run_id", "column name")} = ?${guard}`, [
 					resourceId ?? null,
 					JSON.stringify(snapshot),
 					transformToSqlValue(updatedAtValue),
 					workflowName,
-					runId
+					runId,
+					...expectedStatuses ?? []
 				]);
-				if (updateResult.affectedRows > 0) return;
+				const [updateResult] = await update();
+				if (updateResult.affectedRows > 0) return true;
+				if (expectedStatuses !== void 0) try {
+					await q.execute(`INSERT INTO ${tableName} (${quoteIdentifier("workflow_name", "column name")}, ${quoteIdentifier("run_id", "column name")}, ${quoteIdentifier("resourceId", "column name")}, ${quoteIdentifier("snapshot", "column name")}, ${quoteIdentifier("createdAt", "column name")}, ${quoteIdentifier("updatedAt", "column name")})
+               VALUES (?, ?, ?, ?, ?, ?)`, [
+						workflowName,
+						runId,
+						resourceId ?? null,
+						JSON.stringify(snapshot),
+						transformToSqlValue(createdAtValue),
+						transformToSqlValue(updatedAtValue)
+					]);
+					return true;
+				} catch (error) {
+					if (!(0, _mastra_core_storage.hasErrorCode)(error, DUPLICATE_KEY_ERROR_CODES)) throw error;
+					const [retryResult] = await update();
+					return retryResult.affectedRows > 0;
+				}
 				await q.execute(`INSERT INTO ${tableName} (${quoteIdentifier("workflow_name", "column name")}, ${quoteIdentifier("run_id", "column name")}, ${quoteIdentifier("resourceId", "column name")}, ${quoteIdentifier("snapshot", "column name")}, ${quoteIdentifier("createdAt", "column name")}, ${quoteIdentifier("updatedAt", "column name")})
            VALUES (?, ?, ?, ?, ?, ?)
            ON DUPLICATE KEY UPDATE ${quoteIdentifier("resourceId", "column name")} = VALUES(${quoteIdentifier("resourceId", "column name")}), ${quoteIdentifier("snapshot", "column name")} = VALUES(${quoteIdentifier("snapshot", "column name")}), ${quoteIdentifier("updatedAt", "column name")} = VALUES(${quoteIdentifier("updatedAt", "column name")})`, [
@@ -11732,6 +11756,7 @@ var WorkflowsMySQL = class WorkflowsMySQL extends _mastra_core_storage.Workflows
 					transformToSqlValue(createdAtValue),
 					transformToSqlValue(updatedAtValue)
 				]);
+				return true;
 			});
 		} catch (error) {
 			if ((0, _mastra_core_storage.isRunFenceConflictError)(error)) throw error;
@@ -11746,6 +11771,22 @@ var WorkflowsMySQL = class WorkflowsMySQL extends _mastra_core_storage.Workflows
 			}, error);
 		}
 	}
+	async loadWorkflowRunStatus({ workflowName, runId }) {
+		try {
+			const [rows] = await this.pool.execute(`SELECT JSON_UNQUOTE(JSON_EXTRACT(${quoteIdentifier("snapshot", "column name")}, '$.status')) AS status FROM ${formatTableName(_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT)} WHERE ${quoteIdentifier("workflow_name", "column name")} = ? AND ${quoteIdentifier("run_id", "column name")} = ? LIMIT 1`, [workflowName, runId]);
+			return rows[0]?.status ?? null;
+		} catch (error) {
+			throw new _mastra_core_error.MastraError({
+				id: (0, _mastra_core_storage.createStorageErrorId)("MYSQL", "LOAD_WORKFLOW_RUN_STATUS", "FAILED"),
+				domain: _mastra_core_error.ErrorDomain.STORAGE,
+				category: _mastra_core_error.ErrorCategory.THIRD_PARTY,
+				details: {
+					workflowName,
+					runId
+				}
+			}, error);
+		}
+	}
 	async loadWorkflowSnapshot({ workflowName, runId }) {
 		try {
 			const row = await this.operations.load({