@mastra/dsql 1.5.2 → 1.5.3-alpha.0

package/dist/index.cjs

@@ -669,8 +669,9 @@ var DsqlDB = class extends _mastra_core_base.MastraBase {
 			indexName: tableName,
 			schemaName: getSchemaName$1(this.schemaName)
 		});
-		return (await this.client.query(`DELETE FROM ${table} WHERE ("traceId", "spanId") IN (
-        SELECT "traceId", "spanId" FROM ${table} WHERE "${column}" < $1 ORDER BY "${column}" LIMIT $2
+		const key = tableName === _mastra_core_storage.TABLE_SPANS ? "\"traceId\", \"spanId\"" : "\"runId\"";
+		return (await this.client.query(`DELETE FROM ${table} WHERE (${key}) IN (
+        SELECT ${key} FROM ${table} WHERE "${column}" < $1 ORDER BY "${column}" LIMIT $2
       )`, [cutoff, limit])).rowCount ?? 0;
 	}
 	async clearTable({ tableName }) {
@@ -1897,6 +1898,17 @@ var MemoryDSQL = class MemoryDSQL extends _mastra_core_storage.MemoryStorage {
 		_mastra_core_storage.TABLE_MESSAGES,
 		_mastra_core_storage.TABLE_RESOURCES
 	];
+	/**
+	* Tables `prune()` can clean up. Run fences anchor on `retiredAt` (epoch ms), which stays NULL until the
+	* execution holding the fence settles, so a running execution's fence is
+	* never pruned.
+	*/
+	static retentionTables = { runFences: {
+		table: _mastra_core_storage.TABLE_MEMORY_RUN_FENCES,
+		column: "retiredAt",
+		indexed: false,
+		anchorType: "epoch-ms"
+	} };
 	constructor(config) {
 		super();
 		const { client, schemaName, skipDefaultIndexes, indexes } = resolveDsqlConfig(config);
@@ -1972,6 +1984,24 @@ var MemoryDSQL = class MemoryDSQL extends _mastra_core_storage.MemoryStorage {
 			this.logger?.warn?.(`Failed to create custom index ${indexDef.name}:`, error);
 		}
 	}
+	async prune(policies, options) {
+		return (0, _mastra_core_storage.executeRetentionPrune)({
+			domain: "memory",
+			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
+				policies,
+				descriptor: MemoryDSQL.retentionTables,
+				order: ["runFences"]
+			}),
+			options,
+			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
+			deleteBatch: (target, cutoff, limit) => this.#db.pruneBatch({
+				tableName: target.table,
+				column: target.column,
+				cutoff,
+				limit
+			})
+		});
+	}
 	async dangerouslyClearAll() {
 		await this.#db.clearTable({ tableName: _mastra_core_storage.TABLE_MESSAGES });
 		await this.#db.clearTable({ tableName: _mastra_core_storage.TABLE_THREADS });
@@ -3884,6 +3914,17 @@ var WorkflowsDSQL = class WorkflowsDSQL extends _mastra_core_storage.WorkflowsSt
 	#indexes;
 	/** Tables managed by this domain */
 	static MANAGED_TABLES = [_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT];
+	/**
+	* Tables `prune()` can clean up. Run ownership rows anchor on `updatedAt` (epoch ms), which every
+	* claim, renewal and release refreshes, so a live execution's row is never
+	* older than its renew interval.
+	*/
+	static retentionTables = { runOwnership: {
+		table: _mastra_core_storage.TABLE_WORKFLOW_RUN_OWNERS,
+		column: "updatedAt",
+		indexed: false,
+		anchorType: "epoch-ms"
+	} };
 	constructor(config) {
 		super();
 		const { client, schemaName, skipDefaultIndexes, indexes } = resolveDsqlConfig(config);
@@ -3940,6 +3981,24 @@ var WorkflowsDSQL = class WorkflowsDSQL extends _mastra_core_storage.WorkflowsSt
 			this.logger?.warn?.(`Failed to create custom index ${indexDef.name}:`, error);
 		}
 	}
+	async prune(policies, options) {
+		return (0, _mastra_core_storage.executeRetentionPrune)({
+			domain: "workflows",
+			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
+				policies,
+				descriptor: WorkflowsDSQL.retentionTables,
+				order: ["runOwnership"]
+			}),
+			options,
+			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
+			deleteBatch: (target, cutoff, limit) => this.#db.pruneBatch({
+				tableName: target.table,
+				column: target.column,
+				cutoff,
+				limit
+			})
+		});
+	}
 	async dangerouslyClearAll() {
 		await this.#db.clearTable({ tableName: _mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT });
 		await this.#db.clearTable({ tableName: _mastra_core_storage.TABLE_WORKFLOW_RUN_OWNERS });