@mastra/spanner 1.6.4 → 1.6.5-alpha.0

package/dist/index.js

@@ -9078,19 +9078,34 @@ var ObservabilitySpanner = class ObservabilitySpanner extends ObservabilityStora
 		}
 	}
 	async batchDeleteTraces(args) {
+		this.assertUnscopedBatchDeleteTraces(args);
 		try {
 			if (args.traceIds.length === 0) return;
-			const tableName = quoteIdent(TABLE_SPANS, "table name");
 			const params = {};
 			const placeholders = args.traceIds.map((id, i) => {
 				const name = `t${i}`;
 				params[name] = id;
 				return `@${name}`;
 			});
-			await this.db.runDml({
-				sql: `DELETE FROM ${tableName} WHERE ${quoteIdent("traceId", "column name")} IN (${placeholders.join(", ")})`,
-				params
-			});
+			const traceIdFilter = `${quoteIdent("traceId", "column name")} IN (${placeholders.join(", ")})`;
+			await this.db.runWithAbortRetry(() => this.database.runTransactionAsync(async (tx) => {
+				try {
+					await tx.runUpdate({
+						sql: `DELETE FROM ${quoteIdent(TABLE_SPANS, "table name")} WHERE ${traceIdFilter}`,
+						params
+					});
+					if (!this.disableMetrics) await tx.runUpdate({
+						sql: `DELETE FROM ${quoteIdent(TABLE_AI_METRICS, "table name")} WHERE ${traceIdFilter}`,
+						params
+					});
+					await tx.commit();
+				} catch (error) {
+					await tx.rollback().catch((rollbackError) => {
+						throw new AggregateError([error, rollbackError], "Transaction and rollback both failed");
+					});
+					throw error;
+				}
+			}));
 		} catch (error) {
 			throw new MastraError({
 				id: createStorageErrorId("SPANNER", "BATCH_DELETE_TRACES", "FAILED"),