@mastra/dsql 1.5.4-alpha.0 → 1.5.4

package/dist/index.cjs

@@ -935,7 +935,8 @@ var DsqlDB = class extends _mastra_core_base.MastraBase {
 			const uniqueStr = unique ? "UNIQUE " : "";
 			const methodStr = method !== "btree" ? `USING ${method} ` : "";
 			const columnsStr = columns.map((col) => {
-				const quotedCol = `"${(0, _mastra_core_utils.parseSqlIdentifier)(col.replace(/\s+(DESC|ASC)$/i, "").trim(), "column name")}"`;
+				const colName = col.replace(/\s+(DESC|ASC)$/i, "").trim();
+				const quotedCol = `"${(0, _mastra_core_utils.parseSqlIdentifier)(colName, "column name")}"`;
 				return opclass ? `${quotedCol} ${opclass}` : quotedCol;
 			}).join(", ");
 			const whereStr = where ? ` WHERE ${where}` : "";
@@ -1851,10 +1852,11 @@ function isRetriableRunFenceWrite(error) {
 */
 async function assertRunFence(t, claimsTable, fence, operation) {
 	const row = await t.oneOrNone(`SELECT generation, "ownerId" FROM ${claimsTable} WHERE "runId" = $1 FOR UPDATE`, [fence.runId]);
-	if (!(0, _mastra_core_storage.matchesRunFence)(row && {
+	const current = row && {
 		generation: Number(row.generation),
 		ownerId: row.ownerId
-	}, fence)) throw new _mastra_core_storage.RunFenceConflictError(fence, operation);
+	};
+	if (!(0, _mastra_core_storage.matchesRunFence)(current, fence)) throw new _mastra_core_storage.RunFenceConflictError(fence, operation);
 }
 /**
 * Runs `write` directly when it carries no fence, or in a transaction behind
@@ -1985,13 +1987,14 @@ var MemoryDSQL = class MemoryDSQL extends _mastra_core_storage.MemoryStorage {
 		}
 	}
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: MemoryDSQL.retentionTables,
+			order: ["runFences"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "memory",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: MemoryDSQL.retentionTables,
-				order: ["runFences"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.#db.pruneBatch({
@@ -2971,13 +2974,14 @@ var ObservabilityDSQL = class ObservabilityDSQL extends _mastra_core_storage.Obs
 		this.#indexes = indexes?.filter((idx) => ObservabilityDSQL.MANAGED_TABLES.includes(idx.table));
 	}
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: ObservabilityDSQL.retentionTables,
+			order: ["spans"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "observability",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: ObservabilityDSQL.retentionTables,
-				order: ["spans"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => new Date((0, _mastra_core_storage.retentionCutoffMs)(target.policy, now)),
 			deleteBatch: (target, cutoff, limit) => this.#db.pruneBatch({
@@ -3352,18 +3356,18 @@ var ObservabilityDSQL = class ObservabilityDSQL extends _mastra_core_storage.Obs
 					case _mastra_core_storage.TraceStatus.RUNNING:
 						conditions.push(`r."endedAtZ" IS NULL AND r."error" IS NULL`);
 						break;
-					case _mastra_core_storage.TraceStatus.SUCCESS:
-						conditions.push(`r."endedAtZ" IS NOT NULL AND r."error" IS NULL`);
-						break;
+					case _mastra_core_storage.TraceStatus.SUCCESS: conditions.push(`r."endedAtZ" IS NOT NULL AND r."error" 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 ${tableName} c
               WHERE c."traceId" = r."traceId" AND c."error" IS NOT NULL
             )`);
-				else conditions.push(`NOT EXISTS (
+					else conditions.push(`NOT EXISTS (
               SELECT 1 FROM ${tableName} c
               WHERE c."traceId" = r."traceId" AND c."error" IS NOT NULL
             )`);
+				}
 			}
 			const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
 			const orderClause = `ORDER BY r."${`${orderBy?.field ?? "startedAt"}Z`}" ${orderBy?.direction ?? "DESC"}`;
@@ -3982,13 +3986,14 @@ var WorkflowsDSQL = class WorkflowsDSQL extends _mastra_core_storage.WorkflowsSt
 		}
 	}
 	async prune(policies, options) {
+		const targets = (0, _mastra_core_storage.resolveRetentionTargets)({
+			policies,
+			descriptor: WorkflowsDSQL.retentionTables,
+			order: ["runOwnership"]
+		});
 		return (0, _mastra_core_storage.executeRetentionPrune)({
 			domain: "workflows",
-			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
-				policies,
-				descriptor: WorkflowsDSQL.retentionTables,
-				order: ["runOwnership"]
-			}),
+			targets,
 			options,
 			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
 			deleteBatch: (target, cutoff, limit) => this.#db.pruneBatch({
@@ -4354,11 +4359,13 @@ var WorkflowsDSQL = class WorkflowsDSQL extends _mastra_core_storage.WorkflowsSt
 				values.push(status);
 				paramIndex++;
 			}
-			if (resourceId) if (await this.#db.hasColumn(_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT, "resourceId")) {
-				conditions.push(`"resourceId" = $${paramIndex}`);
-				values.push(resourceId);
-				paramIndex++;
-			} else console.warn(`[${_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT}] resourceId column not found. Skipping resourceId filter.`);
+			if (resourceId) {
+				if (await this.#db.hasColumn(_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT, "resourceId")) {
+					conditions.push(`"resourceId" = $${paramIndex}`);
+					values.push(resourceId);
+					paramIndex++;
+				} else console.warn(`[${_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT}] resourceId column not found. Skipping resourceId filter.`);
+			}
 			if (fromDate) {
 				conditions.push(`"createdAt" >= $${paramIndex}`);
 				values.push(fromDate instanceof Date ? fromDate.toISOString() : fromDate);
@@ -4466,7 +4473,7 @@ var DSQLStore = class extends _mastra_core_storage.MastraStorage {
 	createPool(config) {
 		if (!isHostConfig(config)) throw new Error("DSQLStore: Invalid configuration for creating pool.");
 		const region = getEffectiveRegion(config);
-		return new pg.Pool({
+		const poolConfig = {
 			host: config.host,
 			user: config.user ?? "admin",
 			database: config.database ?? "postgres",
@@ -4479,7 +4486,8 @@ var DSQLStore = class extends _mastra_core_storage.MastraStorage {
 			maxLifetimeSeconds: config.maxLifetimeSeconds ?? DSQL_POOL_DEFAULTS.maxLifetimeSeconds,
 			connectionTimeoutMillis: config.connectionTimeoutMillis ?? DSQL_POOL_DEFAULTS.connectionTimeoutMillis,
 			allowExitOnIdle: config.allowExitOnIdle ?? DSQL_POOL_DEFAULTS.allowExitOnIdle
-		});
+		};
+		return new pg.Pool(poolConfig);
 	}
 	async init() {
 		if (this.isInitialized) return;