@mastra/mysql 0.8.7 → 0.8.8-alpha.0

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;