@mastra/mysql 0.8.7 → 0.9.0-alpha.1

package/dist/index.js

@@ -2414,23 +2414,6 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 	}
 	async _doUpdateItem(args) {
 		this.#rejectToolMocks(args.toolMocks);
-		const existing = await this.getItemById({ id: args.id });
-		if (!existing) throw new MastraError({
-			id: "MYSQL_UPDATE_ITEM_NOT_FOUND",
-			domain: ErrorDomain.STORAGE,
-			category: ErrorCategory.USER,
-			details: { itemId: args.id }
-		});
-		if (existing.datasetId !== args.datasetId) throw new MastraError({
-			id: "MYSQL_UPDATE_ITEM_DATASET_MISMATCH",
-			domain: ErrorDomain.STORAGE,
-			category: ErrorCategory.USER,
-			details: {
-				itemId: args.id,
-				expectedDatasetId: args.datasetId,
-				actualDatasetId: existing.datasetId
-			}
-		});
 		const connection = await this.pool.getConnection();
 		try {
 			await connection.beginTransaction();
@@ -2439,6 +2422,36 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const tableDatasetsName = formatTableName(TABLE_DATASETS);
 			const tableItemsName = formatTableName(TABLE_DATASET_ITEMS);
 			const tableVersionsName = formatTableName(TABLE_DATASET_VERSIONS);
+			await connection.execute(`SELECT id FROM ${tableDatasetsName} WHERE id = ? FOR UPDATE`, [args.datasetId]);
+			const [itemRows] = await connection.execute(`SELECT * FROM ${tableItemsName} WHERE id = ? AND validTo IS NULL AND isDeleted = 0`, [args.id]);
+			const itemRow = itemRows[0];
+			const existing = itemRow ? this.mapItem(itemRow) : null;
+			if (!existing) throw new MastraError({
+				id: "MYSQL_UPDATE_ITEM_NOT_FOUND",
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.USER,
+				details: { itemId: args.id }
+			});
+			if (existing.datasetId !== args.datasetId) throw new MastraError({
+				id: "MYSQL_UPDATE_ITEM_DATASET_MISMATCH",
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.USER,
+				details: {
+					itemId: args.id,
+					expectedDatasetId: args.datasetId,
+					actualDatasetId: existing.datasetId
+				}
+			});
+			if (existing.metadata?.__purged === true) throw new MastraError({
+				id: "DATASET_ITEM_PURGED",
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.USER,
+				details: {
+					datasetId: args.datasetId,
+					itemId: args.id
+				},
+				text: `Purged dataset item cannot be updated: ${args.id}`
+			});
 			const mergedInput = args.input ?? existing.input;
 			const mergedGroundTruth = args.groundTruth ?? existing.groundTruth;
 			const mergedExpectedTrajectory = args.expectedTrajectory ?? existing.expectedTrajectory;
@@ -2510,18 +2523,6 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 		}
 	}
 	async _doDeleteItem({ id, datasetId }) {
-		const existing = await this.getItemById({ id });
-		if (!existing) return;
-		if (existing.datasetId !== datasetId) throw new MastraError({
-			id: "MYSQL_DELETE_ITEM_DATASET_MISMATCH",
-			domain: ErrorDomain.STORAGE,
-			category: ErrorCategory.USER,
-			details: {
-				itemId: id,
-				expectedDatasetId: datasetId,
-				actualDatasetId: existing.datasetId
-			}
-		});
 		const connection = await this.pool.getConnection();
 		try {
 			await connection.beginTransaction();
@@ -2530,6 +2531,24 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const tableDatasetsName = formatTableName(TABLE_DATASETS);
 			const tableItemsName = formatTableName(TABLE_DATASET_ITEMS);
 			const tableVersionsName = formatTableName(TABLE_DATASET_VERSIONS);
+			await connection.execute(`SELECT id FROM ${tableDatasetsName} WHERE id = ? FOR UPDATE`, [datasetId]);
+			const [itemRows] = await connection.execute(`SELECT * FROM ${tableItemsName} WHERE id = ? AND validTo IS NULL AND isDeleted = 0`, [id]);
+			const itemRow = itemRows[0];
+			const existing = itemRow ? this.mapItem(itemRow) : null;
+			if (!existing) {
+				await connection.commit();
+				return;
+			}
+			if (existing.datasetId !== datasetId) throw new MastraError({
+				id: "MYSQL_DELETE_ITEM_DATASET_MISMATCH",
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.USER,
+				details: {
+					itemId: id,
+					expectedDatasetId: datasetId,
+					actualDatasetId: existing.datasetId
+				}
+			});
 			await connection.execute(`UPDATE ${tableDatasetsName} SET \`version\` = \`version\` + 1 WHERE id = ?`, [datasetId]);
 			const [datasetRows] = await connection.execute(`SELECT \`version\`, \`organizationId\`, \`projectId\` FROM ${tableDatasetsName} WHERE id = ?`, [datasetId]);
 			const parentRow = datasetRows[0];
@@ -2898,12 +2917,6 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			category: ErrorCategory.USER,
 			details: { datasetId: input.datasetId }
 		});
-		const currentItems = [];
-		for (const itemId of input.itemIds) {
-			const item = await this.getItemById({ id: itemId });
-			if (item && item.datasetId === input.datasetId) currentItems.push(item);
-		}
-		if (currentItems.length === 0) return;
 		const connection = await this.pool.getConnection();
 		try {
 			await connection.beginTransaction();
@@ -2912,6 +2925,18 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const tableDatasetsName = formatTableName(TABLE_DATASETS);
 			const tableItemsName = formatTableName(TABLE_DATASET_ITEMS);
 			const tableVersionsName = formatTableName(TABLE_DATASET_VERSIONS);
+			await connection.execute(`SELECT id FROM ${tableDatasetsName} WHERE id = ? FOR UPDATE`, [input.datasetId]);
+			const currentItems = [];
+			for (const itemId of input.itemIds) {
+				const [itemRows] = await connection.execute(`SELECT * FROM ${tableItemsName} WHERE id = ? AND validTo IS NULL AND isDeleted = 0`, [itemId]);
+				const row = itemRows[0];
+				const item = row ? this.mapItem(row) : null;
+				if (item && item.datasetId === input.datasetId) currentItems.push(item);
+			}
+			if (currentItems.length === 0) {
+				await connection.commit();
+				return;
+			}
 			await connection.execute(`UPDATE ${tableDatasetsName} SET \`version\` = \`version\` + 1 WHERE id = ?`, [input.datasetId]);
 			const [versionRows] = await connection.execute(`SELECT \`version\` FROM ${tableDatasetsName} WHERE id = ?`, [input.datasetId]);
 			const newVersion = versionRows[0]?.version;
@@ -6349,6 +6374,62 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 			}, error);
 		}
 	}
+	/**
+	* Atomically reassign a thread and all of its messages to a different resource.
+	*
+	* Runs inside a single transaction and takes a `SELECT ... FOR UPDATE` row lock on the
+	* thread, so overlapping transfers of the same thread serialize and can never interleave
+	* the thread update with the message update. Either both the thread and every message move
+	* to the new resource, or neither does — there is no split-ownership window. The thread's
+	* `createdAt` is preserved. Callers are responsible for authorizing the reassignment.
+	*/
+	async updateThreadResourceId({ threadId, resourceId }) {
+		const connection = await this.pool.getConnection();
+		try {
+			await connection.beginTransaction();
+			const [rows] = await connection.execute(`SELECT * FROM ${formatTableName(TABLE_THREADS)} WHERE ${quoteIdentifier("id", "column name")} = ? FOR UPDATE`, [threadId]);
+			const row = rows[0];
+			if (!row) throw new MastraError({
+				id: createStorageErrorId("MYSQL", "UPDATE_THREAD_RESOURCE_ID", "NOT_FOUND"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.USER,
+				text: `Thread "${threadId}" not found`,
+				details: { threadId }
+			});
+			const thread = this.mapThread(row);
+			if (thread.resourceId === resourceId) {
+				await connection.commit();
+				return thread;
+			}
+			const updatedAt = /* @__PURE__ */ new Date();
+			await connection.execute(`UPDATE ${formatTableName(TABLE_THREADS)} SET ${quoteIdentifier("resourceId", "column name")} = ?, ${quoteIdentifier("updatedAt", "column name")} = ? WHERE ${quoteIdentifier("id", "column name")} = ?`, [
+				resourceId,
+				transformToSqlValue(updatedAt),
+				threadId
+			]);
+			await connection.execute(`UPDATE ${formatTableName(TABLE_MESSAGES)} SET ${quoteIdentifier("resourceId", "column name")} = ? WHERE ${quoteIdentifier("thread_id", "column name")} = ?`, [resourceId, threadId]);
+			await connection.commit();
+			return {
+				...thread,
+				resourceId,
+				updatedAt
+			};
+		} catch (error) {
+			await connection.rollback();
+			if (error instanceof MastraError) throw error;
+			throw new MastraError({
+				id: createStorageErrorId("MYSQL", "UPDATE_THREAD_RESOURCE_ID", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: {
+					threadId,
+					resourceId
+				}
+			}, error);
+		} finally {
+			connection.release();
+		}
+	}
 	async listThreads(args) {
 		const { page = 0, perPage: perPageInput, orderBy, filter } = args;
 		const { field, direction } = this.parseOrderBy(orderBy, "DESC");