@mastra/mysql 0.7.1-alpha.0 → 0.7.1-alpha.2

package/dist/index.js

@@ -5,6 +5,46 @@ import { parseSqlIdentifier } from "@mastra/core/utils";
 import { randomUUID } from "crypto";
 import { MessageList } from "@mastra/core/agent";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
+//#region src/storage/db/schema-snapshot.ts
+/** Builds the `table.index` presence key the snapshot uses. */
+function indexKey(table, index) {
+	return `${table.toLowerCase()}.${index.toLowerCase()}`;
+}
+const lower = (value) => String(value).toLowerCase();
+/**
+* Reads the catalog for `schemaName` in three queries. Returns null when no
+* schema name is available (a pool with no default database): correctness over
+* optimization, callers fall back to today's per-probe behavior.
+*/
+async function loadSchemaSnapshot(pool, schemaName) {
+	if (!schemaName) return null;
+	const [[tableRows], [columnRows], [indexRows]] = await Promise.all([
+		pool.execute(`SELECT table_name FROM information_schema.tables WHERE table_schema = ?`, [schemaName]),
+		pool.execute(`SELECT table_name, column_name FROM information_schema.columns WHERE table_schema = ?`, [schemaName]),
+		pool.execute(`SELECT DISTINCT table_name, index_name FROM information_schema.statistics WHERE table_schema = ?`, [schemaName])
+	]);
+	const tables = /* @__PURE__ */ new Set();
+	for (const row of tableRows ?? []) tables.add(lower(row.table_name ?? row.TABLE_NAME));
+	const columns = /* @__PURE__ */ new Map();
+	for (const row of columnRows ?? []) {
+		const table = lower(row.table_name ?? row.TABLE_NAME);
+		let set = columns.get(table);
+		if (!set) {
+			set = /* @__PURE__ */ new Set();
+			columns.set(table, set);
+		}
+		set.add(lower(row.column_name ?? row.COLUMN_NAME));
+	}
+	const indexes = /* @__PURE__ */ new Set();
+	for (const row of indexRows ?? []) indexes.add(indexKey(lower(row.table_name ?? row.TABLE_NAME), lower(row.index_name ?? row.INDEX_NAME)));
+	return {
+		schemaName,
+		tables,
+		columns,
+		indexes
+	};
+}
+//#endregion
 //#region src/storage/domains/utils.ts
 function quoteIdentifier(value, context) {
 	return `\`${parseSqlIdentifier(value, context)}\``;
@@ -144,12 +184,38 @@ var StoreOperationsMySQL = class extends StoreOperations {
 	pool;
 	database;
 	resolvedDatabase;
+	/**
+	* Init-scoped catalog snapshot (see db/schema-snapshot.ts). Installed by
+	* MySQLStore.init() for exactly the init window and cleared in its finally;
+	* null at runtime so non-init callers keep probing the live catalog.
+	*/
+	schemaSnapshot = null;
 	constructor({ pool, database }) {
 		super();
 		this.pool = pool;
 		this.database = database;
 		this.resolvedDatabase = database ?? null;
 	}
+	/**
+	* Loads and installs the init-scoped snapshot. A null load (no default
+	* database) or a failed load leaves the snapshot uninstalled, so init
+	* proceeds with today's per-probe behavior: correctness over optimization.
+	*/
+	async loadInitSchemaSnapshot() {
+		try {
+			this.schemaSnapshot = await loadSchemaSnapshot(this.pool, await this.getDatabase());
+		} catch (error) {
+			console.warn("Failed to load init schema snapshot, falling back to per-object probing:", error);
+			this.schemaSnapshot = null;
+		}
+	}
+	clearInitSchemaSnapshot() {
+		this.schemaSnapshot = null;
+	}
+	/** Read by sibling domains (memory's raw index DDL) so snapshot state lives in exactly one place. */
+	getInitSchemaSnapshot() {
+		return this.schemaSnapshot;
+	}
 	getPool() {
 		return this.pool;
 	}
@@ -165,6 +231,8 @@ var StoreOperationsMySQL = class extends StoreOperations {
 		return this.resolvedDatabase ?? void 0;
 	}
 	async hasColumn(table, column) {
+		const snapshotColumns = this.schemaSnapshot?.columns.get(table.toLowerCase());
+		if (snapshotColumns) return snapshotColumns.has(column.toLowerCase());
 		const db = await this.getDatabase();
 		const params = [table, column];
 		let sql = "SELECT COUNT(*) as count FROM information_schema.columns WHERE table_name = ? AND (column_name = ? OR column_name = ? )";
@@ -242,13 +310,23 @@ var StoreOperationsMySQL = class extends StoreOperations {
 		return false;
 	}
 	async createTable({ tableName, schema }) {
+		const snapshot = this.schemaSnapshot;
+		if (snapshot?.tables.has(tableName.toLowerCase())) return;
 		const connection = await this.pool.getConnection();
 		try {
-			const db = await this.getDatabase();
-			const [t_rows] = await connection.query("SELECT COUNT(*) AS count FROM information_schema.tables WHERE table_schema = ? AND table_name = ?", [db ?? "", tableName]);
-			if (Array.isArray(t_rows) && t_rows.length > 0 && t_rows[0].count > 0) return;
+			if (!snapshot) {
+				const db = await this.getDatabase();
+				const [t_rows] = await connection.query("SELECT COUNT(*) AS count FROM information_schema.tables WHERE table_schema = ? AND table_name = ?", [db ?? "", tableName]);
+				if (Array.isArray(t_rows) && t_rows.length > 0 && t_rows[0].count > 0) return;
+			}
 			const sql = this.getCreateTableSQL(tableName, schema);
-			await connection.execute(sql);
+			const [result] = await connection.execute(sql);
+			const created = result?.warningStatus === 0;
+			if (snapshot && created) {
+				const table = tableName.toLowerCase();
+				snapshot.tables.add(table);
+				snapshot.columns.set(table, new Set(Object.keys(schema).map((column) => column.toLowerCase())));
+			}
 		} catch (error) {
 			throw new MastraError({
 				id: "MYSQL_STORE_CREATE_TABLE_FAILED",
@@ -304,17 +382,22 @@ var StoreOperationsMySQL = class extends StoreOperations {
 		const tableName = formatTableName(table, this.database);
 		const indexName = quoteIdentifier(name, "index name");
 		try {
-			const db = await this.getDatabase() ?? "";
-			const [existing] = await this.pool.execute(`SELECT 1 FROM information_schema.STATISTICS 
+			const snapshot = this.schemaSnapshot;
+			if (snapshot) {
+				if (snapshot.indexes.has(indexKey(table, name))) return;
+			} else {
+				const db = await this.getDatabase() ?? "";
+				const [existing] = await this.pool.execute(`SELECT 1 FROM information_schema.STATISTICS 
          WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? AND INDEX_NAME = ?
          LIMIT 1`, [
-				db,
-				table,
-				name
-			]);
-			if (existing.length > 0) return;
+					db,
+					table,
+					name
+				]);
+				if (existing.length > 0) return;
+			}
 			const [columnMeta] = await this.pool.execute(`SELECT COLUMN_NAME, DATA_TYPE FROM information_schema.COLUMNS
-         WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ?`, [db, table]);
+         WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ?`, [await this.getDatabase() ?? "", table]);
 			const dataTypeByColumn = new Map(columnMeta.map((row) => [String(row.COLUMN_NAME).toLowerCase(), String(row.DATA_TYPE).toLowerCase()]));
 			const PREFIX_TYPES = /* @__PURE__ */ new Set([
 				"tinytext",
@@ -340,6 +423,7 @@ var StoreOperationsMySQL = class extends StoreOperations {
 			}).join(", ");
 			const sql = `CREATE ${unique ? "UNIQUE " : ""}INDEX ${indexName} ON ${tableName} (${columnsStr})`;
 			await this.pool.execute(sql);
+			this.schemaSnapshot?.indexes.add(indexKey(table, name));
 		} catch (error) {
 			console.warn(`Failed to create index ${name}:`, error);
 		}
@@ -571,17 +655,22 @@ var StoreOperationsMySQL = class extends StoreOperations {
 	}
 	async alterTable({ tableName, schema, ifNotExists }) {
 		if (!ifNotExists.length) return;
-		const db = await this.getDatabase();
-		const [tableRows] = await this.pool.execute("SELECT COUNT(*) AS count FROM information_schema.tables WHERE table_schema = ? AND table_name = ?", [db ?? "", tableName]);
-		if (!(Array.isArray(tableRows) && tableRows.length > 0 && tableRows[0].count > 0)) return;
-		const params = [tableName];
-		let sql = "SELECT column_name FROM information_schema.columns WHERE table_name = ?";
-		if (db) {
-			sql += " AND table_schema = ?";
-			params.push(db);
+		const snapshotColumns = this.schemaSnapshot?.columns.get(tableName.toLowerCase());
+		let existing;
+		if (snapshotColumns) existing = snapshotColumns;
+		else {
+			const db = await this.getDatabase();
+			const [tableRows] = await this.pool.execute("SELECT COUNT(*) AS count FROM information_schema.tables WHERE table_schema = ? AND table_name = ?", [db ?? "", tableName]);
+			if (!(Array.isArray(tableRows) && tableRows.length > 0 && tableRows[0].count > 0)) return;
+			const params = [tableName];
+			let sql = "SELECT column_name FROM information_schema.columns WHERE table_name = ?";
+			if (db) {
+				sql += " AND table_schema = ?";
+				params.push(db);
+			}
+			const [rows] = await this.pool.execute(sql, params);
+			existing = new Set((rows || []).map((row) => String(row.column_name ?? row.COLUMN_NAME).toLowerCase()));
 		}
-		const [rows] = await this.pool.execute(sql, params);
-		const existing = new Set((rows || []).map((row) => String(row.column_name ?? row.COLUMN_NAME).toLowerCase()));
 		for (const columnName of ifNotExists) {
 			if (existing.has(columnName.toLowerCase())) continue;
 			const column = schema[columnName];
@@ -603,6 +692,7 @@ var StoreOperationsMySQL = class extends StoreOperations {
 			const alterSql = `ALTER TABLE ${formatTableName(tableName, this.database)} ADD COLUMN ${parts.join(" ")}`;
 			try {
 				await this.pool.execute(alterSql);
+				existing.add(columnName.toLowerCase());
 			} catch (error) {
 				if (error?.code === "ER_DUP_FIELDNAME") continue;
 				throw new MastraError({
@@ -759,7 +849,8 @@ var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 				"requestContextSchema",
 				"workspace",
 				"skills",
-				"skillsFormat"
+				"skillsFormat",
+				"durable"
 			]
 		});
 		await this.createDefaultIndexes();
@@ -1051,6 +1142,7 @@ var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 					workspace: input.workspace ?? null,
 					skills: input.skills ?? null,
 					skillsFormat: input.skillsFormat ?? null,
+					durable: input.durable ?? null,
 					changedFields: input.changedFields ?? null,
 					changeMessage: input.changeMessage ?? null,
 					createdAt: now
@@ -1266,6 +1358,7 @@ var AgentsMySQL = class AgentsMySQL extends AgentsStorage {
 			workspace: this.safeParseJSON(row.workspace),
 			skills: this.safeParseJSON(row.skills),
 			skillsFormat: row.skillsFormat,
+			durable: this.safeParseJSON(row.durable),
 			changedFields: this.safeParseJSON(row.changedFields),
 			changeMessage: row.changeMessage ?? void 0,
 			createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt)
@@ -4832,10 +4925,14 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 			schema: TABLE_SCHEMAS[TABLE_MESSAGES],
 			ifNotExists: ["resourceId"]
 		});
-		if (omSchema) try {
-			await this.pool.execute(`CREATE INDEX idx_om_lookup_key ON ${OM_TABLE_QUOTED} (${quoteIdentifier("lookupKey", "column name")}(191))`);
-		} catch (err) {
-			if (err?.errno !== 1061) throw err;
+		if (omSchema) {
+			const snapshot = this.operations.getInitSchemaSnapshot();
+			if (!snapshot?.indexes.has(indexKey(OM_TABLE, "idx_om_lookup_key"))) try {
+				await this.pool.execute(`CREATE INDEX idx_om_lookup_key ON ${OM_TABLE_QUOTED} (${quoteIdentifier("lookupKey", "column name")}(191))`);
+				snapshot?.indexes.add(indexKey(OM_TABLE, "idx_om_lookup_key"));
+			} catch (err) {
+				if (err?.errno !== 1061) throw err;
+			}
 		}
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
@@ -10354,6 +10451,7 @@ function parseConnectionString(connectionString, overrides) {
 }
 var MySQLStore = class extends MastraCompositeStore {
 	pool;
+	operations;
 	stores;
 	constructor(config) {
 		super({
@@ -10368,6 +10466,7 @@ var MySQLStore = class extends MastraCompositeStore {
 			pool: this.pool,
 			database
 		});
+		this.operations = operations;
 		const memory = new MemoryMySQL({
 			pool: this.pool,
 			operations,
@@ -10510,6 +10609,7 @@ var MySQLStore = class extends MastraCompositeStore {
 	async init() {
 		try {
 			(await this.pool.getConnection()).release();
+			await this.operations.loadInitSchemaSnapshot();
 			await super.init();
 		} catch (error) {
 			throw new MastraError({
@@ -10517,6 +10617,8 @@ var MySQLStore = class extends MastraCompositeStore {
 				domain: ErrorDomain.STORAGE,
 				category: ErrorCategory.THIRD_PARTY
 			}, error);
+		} finally {
+			this.operations.clearInitSchemaSnapshot();
 		}
 	}
 	async close() {