@mastra/mysql 0.14.1-alpha.0 → 0.14.1-alpha.1

package/dist/index.js

@@ -11335,6 +11335,7 @@ var WorkflowDefinitionsMySQL = class extends WorkflowDefinitionsStorage {
 	}
 };
 // 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);
@@ -11418,6 +11419,9 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 		this.#skipDefaultIndexes = skipDefaultIndexes;
 		this.#indexes = indexes?.filter((idx) => WorkflowsMySQL.MANAGED_TABLES.includes(idx.table));
 	}
+	supportsGuardedSnapshotWrites() {
+		return true;
+	}
 	supportsConcurrentUpdates() {
 		return true;
 	}
@@ -11702,25 +11706,45 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 			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(TABLE_WORKFLOW_SNAPSHOT);
 		try {
-			await withRunFence(this.pool, RUN_OWNERS_TABLE, resolveRunFence(this, fence, runId), "persistWorkflowSnapshot", async (q) => {
-				const [updateResult] = await q.execute(`UPDATE ${tableName}
+			return await withRunFence(this.pool, RUN_OWNERS_TABLE, 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 (!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")})`, [
@@ -11731,6 +11755,7 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 					transformToSqlValue(createdAtValue),
 					transformToSqlValue(updatedAtValue)
 				]);
+				return true;
 			});
 		} catch (error) {
 			if (isRunFenceConflictError(error)) throw error;
@@ -11745,6 +11770,22 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 			}, 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(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 MastraError({
+				id: createStorageErrorId("MYSQL", "LOAD_WORKFLOW_RUN_STATUS", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: {
+					workflowName,
+					runId
+				}
+			}, error);
+		}
+	}
 	async loadWorkflowSnapshot({ workflowName, runId }) {
 		try {
 			const row = await this.operations.load({