@mastra/mysql 0.14.0 → 0.14.1-alpha.1

package/dist/index.js

@@ -5,7 +5,7 @@ import { parseSqlIdentifier } from "@mastra/core/utils";
 import { createHash } from "crypto";
 import { MessageList } from "@mastra/core/agent";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
-//#region src/storage/db/schema-snapshot.ts
+// src/storage/db/schema-snapshot.ts
 /** Builds the `table.index` presence key the snapshot uses. */
 function indexKey(table, index) {
 	return `${table.toLowerCase()}.${index.toLowerCase()}`;
@@ -44,8 +44,7 @@ async function loadSchemaSnapshot(pool, schemaName) {
 		indexes
 	};
 }
-//#endregion
-//#region src/storage/domains/utils.ts
+// src/storage/domains/utils.ts
 function quoteIdentifier(value, context) {
 	return `\`${parseSqlIdentifier(value, context)}\``;
 }
@@ -180,8 +179,7 @@ function parseDateTime(value) {
 	const withZone = /[zZ]|[+-]\d\d:?\d\d$/.test(normalized) ? normalized : `${normalized}Z`;
 	return new Date(withZone);
 }
-//#endregion
-//#region src/storage/domains/operations/index.ts
+// src/storage/domains/operations/index.ts
 var StoreOperationsMySQL = class extends StoreOperations {
 	pool;
 	database;
@@ -775,8 +773,7 @@ function mapToMySqlType(type, isKey = false) {
 		default: return "TEXT";
 	}
 }
-//#endregion
-//#region src/storage/domains/agents/index.ts
+// src/storage/domains/agents/index.ts
 var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 	pool;
 	operations;
@@ -1373,8 +1370,7 @@ var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/background-tasks/index.ts
+// src/storage/domains/background-tasks/index.ts
 function serializeJson(v) {
 	if (typeof v === "object" && v != null) return JSON.stringify(v);
 	return v ?? null;
@@ -1692,8 +1688,7 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 		return Number(rows[0]?.count ?? 0);
 	}
 };
-//#endregion
-//#region src/storage/domains/blobs/index.ts
+// src/storage/domains/blobs/index.ts
 const INSERT_IGNORE_SQL = `INSERT IGNORE INTO ${formatTableName(TABLE_SKILL_BLOBS)} (${[
 	"hash",
 	"content",
@@ -1774,8 +1769,7 @@ var BlobsMySQL = class extends BlobStore {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/channels/index.ts
+// src/storage/domains/channels/index.ts
 var ChannelsMySQL = class ChannelsMySQL extends ChannelsStorage {
 	pool;
 	operations;
@@ -1909,8 +1903,7 @@ var ChannelsMySQL = class ChannelsMySQL extends ChannelsStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/datasets/index.ts
+// src/storage/domains/datasets/index.ts
 function parseJSON$3(value) {
 	if (value === null || value === void 0) return void 0;
 	if (typeof value === "string") try {
@@ -3061,8 +3054,7 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/experiments/index.ts
+// src/storage/domains/experiments/index.ts
 function parseJSON$2(value) {
 	if (value === null || value === void 0) return void 0;
 	if (typeof value === "string") {
@@ -3934,8 +3926,7 @@ var ExperimentsMySQL = class ExperimentsMySQL extends ExperimentsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/favorites/index.ts
+// src/storage/domains/favorites/index.ts
 /**
 * Maps a favorite entity type to its parent entity table.
 */
@@ -4171,8 +4162,7 @@ var FavoritesMySQL = class FavoritesMySQL extends FavoritesStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/knowledge/index.ts
+// src/storage/domains/knowledge/index.ts
 let assertDescriptionWithinBound;
 async function assertKnowledgeDescriptionWithinBoundCompat(description) {
 	let assertWithinBound = assertDescriptionWithinBound;
@@ -5067,8 +5057,7 @@ var KnowledgeMySQL = class extends KnowledgeStorage {
 		});
 	}
 };
-//#endregion
-//#region src/storage/domains/mcp-clients/index.ts
+// src/storage/domains/mcp-clients/index.ts
 var MCPClientsMySQL = class MCPClientsMySQL extends MCPClientsStorage {
 	pool;
 	operations;
@@ -5541,8 +5530,7 @@ var MCPClientsMySQL = class MCPClientsMySQL extends MCPClientsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/mcp-servers/index.ts
+// src/storage/domains/mcp-servers/index.ts
 var MCPServersMySQL = class MCPServersMySQL extends MCPServersStorage {
 	pool;
 	operations;
@@ -6031,8 +6019,7 @@ var MCPServersMySQL = class MCPServersMySQL extends MCPServersStorage {
 		}
 	}
 };
-//#endregion
-//#region ../../packages/_internal-core/dist/memory/index.js
+// ../../packages/_internal-core/dist/memory/index.js
 /**
 * Shared rules for observational memory lifecycle writes. Every storage adapter applies
 * these inside its own atomic section (lock, transaction, or conditional update), so the
@@ -6117,8 +6104,7 @@ function isAppendOnlySince(storedObservations, snapshotObservations) {
 	const snapshot = snapshotObservations ?? "";
 	return stored === snapshot || stored.startsWith(snapshot);
 }
-//#endregion
-//#region src/storage/db/run-fencing.ts
+// src/storage/db/run-fencing.ts
 /**
 * Epoch milliseconds on the database clock. `UTC_TIMESTAMP` is the time the
 * statement started and doesn't depend on the session time zone. Every lease
@@ -6186,8 +6172,7 @@ async function claimTransaction(pool, fn) {
 		if (attempt >= MAX_CLAIM_ATTEMPTS || errno !== ER_DUP_ENTRY && errno !== ER_LOCK_DEADLOCK) throw error;
 	}
 }
-//#endregion
-//#region src/storage/domains/memory/index.ts
+// src/storage/domains/memory/index.ts
 const OM_TABLE = "mastra_observational_memory";
 const OM_TABLE_QUOTED = quoteIdentifier(OM_TABLE, "table name");
 const RUN_FENCES_TABLE = formatTableName(TABLE_MEMORY_RUN_FENCES);
@@ -8378,8 +8363,7 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 		};
 	}
 };
-//#endregion
-//#region src/storage/domains/observability/index.ts
+// src/storage/domains/observability/index.ts
 const JSON_SPAN_FIELDS = [
 	"input",
 	"output",
@@ -8913,8 +8897,7 @@ var ObservabilityMySQL = class ObservabilityMySQL extends ObservabilityStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/prompt-blocks/index.ts
+// src/storage/domains/prompt-blocks/index.ts
 var PromptBlocksMySQL = class PromptBlocksMySQL extends PromptBlocksStorage {
 	pool;
 	operations;
@@ -9386,8 +9369,7 @@ var PromptBlocksMySQL = class PromptBlocksMySQL extends PromptBlocksStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/schedules/index.ts
+// src/storage/domains/schedules/index.ts
 function parseJson(val) {
 	if (val == null) return void 0;
 	if (typeof val === "string") try {
@@ -9656,8 +9638,7 @@ var SchedulesMySQL = class SchedulesMySQL extends SchedulesStorage {
 		return rows.map((r) => rowToTrigger(r));
 	}
 };
-//#endregion
-//#region src/storage/domains/scorer-definitions/index.ts
+// src/storage/domains/scorer-definitions/index.ts
 var ScorerDefinitionsMySQL = class ScorerDefinitionsMySQL extends ScorerDefinitionsStorage {
 	pool;
 	operations;
@@ -10132,8 +10113,7 @@ var ScorerDefinitionsMySQL = class ScorerDefinitionsMySQL extends ScorerDefiniti
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/scores/index.ts
+// src/storage/domains/scores/index.ts
 function parseJSON(value) {
 	if (value === null || value === void 0) return void 0;
 	if (typeof value === "string") {
@@ -10428,8 +10408,7 @@ var ScoresMySQL = class ScoresMySQL extends ScoresStorage {
 		}), pagination);
 	}
 };
-//#endregion
-//#region src/storage/domains/skills/index.ts
+// src/storage/domains/skills/index.ts
 const SNAPSHOT_FIELDS$1 = [
 	"name",
 	"description",
@@ -10989,8 +10968,7 @@ var SkillsMySQL = class SkillsMySQL extends SkillsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/tool-provider-connections/index.ts
+// src/storage/domains/tool-provider-connections/index.ts
 function normaliseScope(raw) {
 	const value = raw == null ? "per-author" : String(raw);
 	if (value === "shared") return "shared";
@@ -11215,8 +11193,7 @@ var ToolProviderConnectionsMySQL = class ToolProviderConnectionsMySQL extends To
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workflow-definitions/index.ts
+// src/storage/domains/workflow-definitions/index.ts
 function rowToDefinition(row) {
 	const transformed = transformFromSqlRow({
 		tableName: TABLE_WORKFLOW_DEFINITIONS,
@@ -11357,8 +11334,8 @@ var WorkflowDefinitionsMySQL = class extends WorkflowDefinitionsStorage {
 		await this.pool.execute(`DELETE FROM ${formatTableName(TABLE_WORKFLOW_DEFINITIONS, this.database)} WHERE ${quoteIdentifier("id", "column name")} = ?`, [id]);
 	}
 };
-//#endregion
-//#region src/storage/domains/workflows/index.ts
+// 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);
@@ -11442,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;
 	}
@@ -11726,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")})`, [
@@ -11755,6 +11755,7 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 					transformToSqlValue(createdAtValue),
 					transformToSqlValue(updatedAtValue)
 				]);
+				return true;
 			});
 		} catch (error) {
 			if (isRunFenceConflictError(error)) throw error;
@@ -11769,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({
@@ -11877,8 +11894,7 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workspaces/index.ts
+// src/storage/domains/workspaces/index.ts
 const SNAPSHOT_FIELDS = [
 	"name",
 	"description",
@@ -12396,8 +12412,7 @@ var WorkspacesMySQL = class WorkspacesMySQL extends WorkspacesStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/index.ts
+// src/storage/index.ts
 function validateConfig(config) {
 	if ("connectionString" in config) {
 		if (!config.connectionString || typeof config.connectionString !== "string") throw new Error("MySQLStore: connectionString must be a non-empty string.");
@@ -12670,7 +12685,6 @@ var MySQLStore = class extends MastraCompositeStore {
 		await this.pool.end();
 	}
 };
-//#endregion
 export { AgentsMySQL, BlobsMySQL, DatasetsMySQL, ExperimentsMySQL, MCPClientsMySQL, MCPServersMySQL, MemoryMySQL, MySQLStore, ObservabilityMySQL, PromptBlocksMySQL, ScorerDefinitionsMySQL, ScoresMySQL, SkillsMySQL, StoreOperationsMySQL, WorkflowsMySQL, WorkspacesMySQL };
 
 //# sourceMappingURL=index.js.map