@mastra/mysql 0.13.3-alpha.0 → 0.14.0

package/dist/index.js

@@ -85,13 +85,14 @@ function prepareWhereClause(filters) {
 			if (value === true) conditions.push(`${quoteIdentifier(fieldName, "column name")} IS NULL`);
 			else if (value === false) conditions.push(`${quoteIdentifier(fieldName, "column name")} IS NOT NULL`);
 		} else if (value === null) conditions.push(`${quoteIdentifier(key, "column name")} IS NULL`);
-		else if (Array.isArray(value)) if (value.length === 0) conditions.push("FALSE");
-		else {
-			const placeholders = value.map(() => "?").join(", ");
-			conditions.push(`${quoteIdentifier(key, "column name")} IN (${placeholders})`);
-			args.push(...value.map(transformToSqlValue));
-		}
-		else {
+		else if (Array.isArray(value)) {
+			if (value.length === 0) conditions.push("FALSE");
+			else {
+				const placeholders = value.map(() => "?").join(", ");
+				conditions.push(`${quoteIdentifier(key, "column name")} IN (${placeholders})`);
+				args.push(...value.map(transformToSqlValue));
+			}
+		} else {
 			conditions.push(`${quoteIdentifier(key, "column name")} = ?`);
 			args.push(transformToSqlValue(value));
 		}
@@ -156,13 +157,14 @@ function transformFromSqlRow({ tableName, sqlRow }) {
 	const result = {};
 	Object.entries(sqlRow).forEach(([key, value]) => {
 		const columnSchema = schema?.[key];
-		if (columnSchema?.type === "jsonb") if (typeof value === "string") try {
-			result[key] = JSON.parse(value);
-		} catch {
-			result[key] = value;
-		}
-		else result[key] = value;
-		else if (columnSchema?.type === "timestamp") result[key] = value === null ? null : parseDateTime(value);
+		if (columnSchema?.type === "jsonb") {
+			if (typeof value === "string") try {
+				result[key] = JSON.parse(value);
+			} catch {
+				result[key] = value;
+			}
+			else result[key] = value;
+		} else if (columnSchema?.type === "timestamp") result[key] = value === null ? null : parseDateTime(value);
 		else if (columnSchema?.type === "boolean") result[key] = value === true || value === 1 || value === "1";
 		else result[key] = value;
 	});
@@ -1564,15 +1566,19 @@ var BackgroundTasksMySQL = class BackgroundTasksMySQL extends BackgroundTasksSto
 			predicates.push(`${quoteIdentifier("status", "column name")} = ?`);
 			params.push(options.expectedStatus);
 		}
-		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates.push(`${quoteIdentifier("ownerId", "column name")} IS NULL`);
-		else {
-			predicates.push(`${quoteIdentifier("ownerId", "column name")} = ?`);
-			params.push(options.expectedOwnerId);
+		if (options?.expectedOwnerId !== void 0) {
+			if (options.expectedOwnerId === null) predicates.push(`${quoteIdentifier("ownerId", "column name")} IS NULL`);
+			else {
+				predicates.push(`${quoteIdentifier("ownerId", "column name")} = ?`);
+				params.push(options.expectedOwnerId);
+			}
 		}
-		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} IS NULL`);
-		else {
-			predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} = ?`);
-			params.push(transformToSqlValue(options.expectedLeaseExpiresAt));
+		if (options?.expectedLeaseExpiresAt !== void 0) {
+			if (options.expectedLeaseExpiresAt === null) predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} IS NULL`);
+			else {
+				predicates.push(`${quoteIdentifier("leaseExpiresAt", "column name")} = ?`);
+				params.push(transformToSqlValue(options.expectedLeaseExpiresAt));
+			}
 		}
 		const predicateSql = predicates.length > 0 ? ` AND ${predicates.join(" AND ")}` : "";
 		const [result] = await this.pool.execute(`UPDATE ${formatTableName(TABLE_BACKGROUND_TASKS)} SET ${setClauses.join(", ")} WHERE ${quoteIdentifier("id", "column name")} = ?${predicateSql}`, params);
@@ -2339,6 +2345,10 @@ var DatasetsMySQL = class DatasetsMySQL extends DatasetsStorage {
 			const { page, perPage: perPageInput } = args.pagination;
 			const filterParts = [];
 			const filterArgs = [];
+			if (args.filters?.datasetIds !== void 0) {
+				filterParts.push(args.filters.datasetIds.length > 0 ? `\`id\` IN (${args.filters.datasetIds.map(() => "?").join(",")})` : "1 = 0");
+				filterArgs.push(...args.filters.datasetIds);
+			}
 			if (args.filters?.organizationId !== void 0) {
 				filterParts.push(`${quoteIdentifier("organizationId", "column name")} = ?`);
 				filterArgs.push(args.filters.organizationId);
@@ -6129,10 +6139,11 @@ async function assertRunFence(q, claimsTable, fence, operation) {
 	const [rows] = await q.execute(`SELECT generation, ${quoteIdentifier("ownerId", "column name")} AS ownerId FROM ${claimsTable}
      WHERE ${quoteIdentifier("runId", "column name")} = ? LOCK IN SHARE MODE`, [fence.runId]);
 	const row = rows[0];
-	if (!matchesRunFence(row ? {
+	const current = row ? {
 		generation: Number(row.generation),
 		ownerId: String(row.ownerId)
-	} : null, fence)) throw new RunFenceConflictError(fence, operation);
+	} : null;
+	if (!matchesRunFence(current, fence)) throw new RunFenceConflictError(fence, operation);
 }
 /** Runs `fn` in a transaction on its own pooled connection. */
 async function inTransaction(pool, fn) {
@@ -6466,13 +6477,14 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 		for (const indexDef of this.#indexes) await this.operations.createIndex(indexDef);
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: MemoryMySQL.retentionTables,
+			order: ["runFences"]
+		});
 		return executeRetentionPrune({
 			domain: "memory",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: MemoryMySQL.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -8400,13 +8412,14 @@ var ObservabilityMySQL = class ObservabilityMySQL extends ObservabilityStorage {
 		this.#indexes = indexes?.filter((idx) => ObservabilityMySQL.MANAGED_TABLES.includes(idx.table));
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: ObservabilityMySQL.retentionTables,
+			order: ["spans"]
+		});
 		return executeRetentionPrune({
 			domain: "observability",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: ObservabilityMySQL.retentionTables,
-				order: ["spans"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => new Date(retentionCutoffMs(target.policy, now)),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
@@ -8773,22 +8786,22 @@ var ObservabilityMySQL = class ObservabilityMySQL extends ObservabilityStorage {
 					case TraceStatus.RUNNING:
 						conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
 						break;
-					case TraceStatus.SUCCESS:
-						conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NOT NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
-						break;
+					case TraceStatus.SUCCESS: conditions.push(`${quoteIdentifier("endedAt", "column name")} IS NOT NULL AND ${quoteIdentifier("error", "column name")} IS NULL`);
 				}
-				if (filters.hasChildError !== void 0) if (filters.hasChildError) conditions.push(`EXISTS (
+				if (filters.hasChildError !== void 0) {
+					if (filters.hasChildError) conditions.push(`EXISTS (
               SELECT 1 FROM ${tbl} c
               WHERE c.${quoteIdentifier("traceId", "column name")} = ${tbl}.${quoteIdentifier("traceId", "column name")}
                 AND c.${quoteIdentifier("parentSpanId", "column name")} IS NOT NULL
                 AND c.${quoteIdentifier("error", "column name")} IS NOT NULL
             )`);
-				else conditions.push(`NOT EXISTS (
+					else conditions.push(`NOT EXISTS (
               SELECT 1 FROM ${tbl} c
               WHERE c.${quoteIdentifier("traceId", "column name")} = ${tbl}.${quoteIdentifier("traceId", "column name")}
                 AND c.${quoteIdentifier("parentSpanId", "column name")} IS NOT NULL
                 AND c.${quoteIdentifier("error", "column name")} IS NOT NULL
             )`);
+				}
 			}
 			const whereClause = conditions.length > 0 ? ` WHERE ${conditions.join(" AND ")}` : "";
 			const sortField = quoteIdentifier(orderBy.field, "column name");
@@ -9524,15 +9537,19 @@ var SchedulesMySQL = class SchedulesMySQL extends SchedulesStorage {
 			conditions.push(`JSON_EXTRACT(${quoteIdentifier("target", "column name")}, '$.workflowId') = ?`);
 			params.push(filter.workflowId);
 		}
-		if (filter?.ownerType !== void 0) if (filter.ownerType === null) conditions.push(`${quoteIdentifier("owner_type", "column name")} IS NULL`);
-		else {
-			conditions.push(`${quoteIdentifier("owner_type", "column name")} = ?`);
-			params.push(filter.ownerType);
+		if (filter?.ownerType !== void 0) {
+			if (filter.ownerType === null) conditions.push(`${quoteIdentifier("owner_type", "column name")} IS NULL`);
+			else {
+				conditions.push(`${quoteIdentifier("owner_type", "column name")} = ?`);
+				params.push(filter.ownerType);
+			}
 		}
-		if (filter?.ownerId !== void 0) if (filter.ownerId === null) conditions.push(`${quoteIdentifier("owner_id", "column name")} IS NULL`);
-		else {
-			conditions.push(`${quoteIdentifier("owner_id", "column name")} = ?`);
-			params.push(filter.ownerId);
+		if (filter?.ownerId !== void 0) {
+			if (filter.ownerId === null) conditions.push(`${quoteIdentifier("owner_id", "column name")} IS NULL`);
+			else {
+				conditions.push(`${quoteIdentifier("owner_id", "column name")} = ?`);
+				params.push(filter.ownerId);
+			}
 		}
 		const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
 		const [rows] = await this.pool.execute(`SELECT * FROM ${formatTableName(TABLE_SCHEDULES)} ${where} ORDER BY ${quoteIdentifier("created_at", "column name")} ASC`, params);
@@ -11467,13 +11484,14 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 		await this.createCustomIndexes();
 	}
 	async prune(policies, options) {
+		const targets = resolveRetentionTargets({
+			policies,
+			descriptor: WorkflowsMySQL.retentionTables,
+			order: ["runOwnership"]
+		});
 		return executeRetentionPrune({
 			domain: "workflows",
-			targets: resolveRetentionTargets({
-				policies,
-				descriptor: WorkflowsMySQL.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({