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

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/mysql
 
+## 0.7.1-alpha.2
+
+### Patch Changes
+
+- Cut warm initialization from around 110 client-server round trips to single digits with an init-scoped schema snapshot, on top of the column-probe casing fix that removed the ALTER TABLE storm. Three information_schema reads at the start of init() now answer table, column, and index existence locally; createTable, alterTable, createIndex, and hasColumn consult the snapshot and maintain it as objects are created, and the memory domain's raw CREATE INDEX for idx_om_lookup_key consults it too instead of raising and swallowing ER_DUP_KEYNAME on every boot. The snapshot lives for exactly the init window and is cleared in a finally, so runtime callers keep querying the live catalog. Measured on docker mysql:9.7: warm init 109 to 111 round trips down to 7 (6 excluding measurement scaffolding), cold init 253 down to 153 or 154 across runs, with an identical cold-init table and index census before and after. ([#21634](https://github.com/mastra-ai/mastra/pull/21634))
+
+- Updated dependencies [[`6db7a5d`](https://github.com/mastra-ai/mastra/commit/6db7a5dd3dd2b6f7ef75dcd804fcffef5fa83963), [`0cdc5dc`](https://github.com/mastra-ai/mastra/commit/0cdc5dc69024957815da4f51acc4119eb4f447d7)]:
+  - @mastra/core@1.60.0-alpha.12
+
 ## 0.7.1-alpha.1
 
 ### Patch Changes

package/dist/index.cjs

@@ -6,6 +6,46 @@ let _mastra_core_utils = require("@mastra/core/utils");
 let crypto$1 = require("crypto");
 let _mastra_core_agent = require("@mastra/core/agent");
 let _mastra_core_storage_domains_skills = require("@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 `\`${(0, _mastra_core_utils.parseSqlIdentifier)(value, context)}\``;
@@ -145,12 +185,38 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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;
 	}
@@ -166,6 +232,8 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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 = ? )";
@@ -243,13 +311,23 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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 _mastra_core_error.MastraError({
 				id: "MYSQL_STORE_CREATE_TABLE_FAILED",
@@ -305,17 +383,22 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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",
@@ -341,6 +424,7 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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);
 		}
@@ -572,17 +656,22 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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];
@@ -604,6 +693,7 @@ var StoreOperationsMySQL = class extends _mastra_core_storage.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 _mastra_core_error.MastraError({
@@ -4836,10 +4926,14 @@ var MemoryMySQL = class MemoryMySQL extends _mastra_core_storage.MemoryStorage {
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.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();
@@ -10358,6 +10452,7 @@ function parseConnectionString(connectionString, overrides) {
 }
 var MySQLStore = class extends _mastra_core_storage.MastraCompositeStore {
 	pool;
+	operations;
 	stores;
 	constructor(config) {
 		super({
@@ -10372,6 +10467,7 @@ var MySQLStore = class extends _mastra_core_storage.MastraCompositeStore {
 			pool: this.pool,
 			database
 		});
+		this.operations = operations;
 		const memory = new MemoryMySQL({
 			pool: this.pool,
 			operations,
@@ -10514,6 +10610,7 @@ var MySQLStore = class extends _mastra_core_storage.MastraCompositeStore {
 	async init() {
 		try {
 			(await this.pool.getConnection()).release();
+			await this.operations.loadInitSchemaSnapshot();
 			await super.init();
 		} catch (error) {
 			throw new _mastra_core_error.MastraError({
@@ -10521,6 +10618,8 @@ var MySQLStore = class extends _mastra_core_storage.MastraCompositeStore {
 				domain: _mastra_core_error.ErrorDomain.STORAGE,
 				category: _mastra_core_error.ErrorCategory.THIRD_PARTY
 			}, error);
+		} finally {
+			this.operations.clearInitSchemaSnapshot();
 		}
 	}
 	async close() {