@mastra/mysql 0.13.1 → 0.13.2

package/dist/index.js

@@ -6282,6 +6282,17 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 		TABLE_MESSAGES,
 		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: TABLE_MEMORY_RUN_FENCES,
+		column: "retiredAt",
+		indexed: false,
+		anchorType: "epoch-ms"
+	} };
 	constructor({ pool, operations, skipDefaultIndexes, indexes }) {
 		super();
 		this.pool = pool;
@@ -6454,6 +6465,24 @@ var MemoryMySQL = class MemoryMySQL extends MemoryStorage {
 		if (!this.#indexes || this.#indexes.length === 0) return;
 		for (const indexDef of this.#indexes) await this.operations.createIndex(indexDef);
 	}
+	async prune(policies, options) {
+		return executeRetentionPrune({
+			domain: "memory",
+			targets: resolveRetentionTargets({
+				policies,
+				descriptor: MemoryMySQL.retentionTables,
+				order: ["runFences"]
+			}),
+			options,
+			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
+			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
+				tableName: target.table,
+				column: target.column,
+				cutoff,
+				limit
+			})
+		});
+	}
 	async dangerouslyClearAll() {
 		await this.pool.execute(`DELETE FROM ${formatTableName(TABLE_MESSAGES)}`);
 		await this.pool.execute(`DELETE FROM ${formatTableName(TABLE_THREADS)}`);
@@ -11359,6 +11388,17 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 	/** Tables managed by this domain */
 	static MANAGED_TABLES = [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: TABLE_WORKFLOW_RUN_OWNERS,
+		column: "updatedAt",
+		indexed: false,
+		anchorType: "epoch-ms"
+	} };
+	/**
 	* Returns default index definitions for the workflows domain tables.
 	* Currently no default indexes are defined for workflows.
 	*/
@@ -11426,6 +11466,24 @@ var WorkflowsMySQL = class WorkflowsMySQL extends WorkflowsStorage {
 		await this.createDefaultIndexes();
 		await this.createCustomIndexes();
 	}
+	async prune(policies, options) {
+		return executeRetentionPrune({
+			domain: "workflows",
+			targets: resolveRetentionTargets({
+				policies,
+				descriptor: WorkflowsMySQL.retentionTables,
+				order: ["runOwnership"]
+			}),
+			options,
+			cutoffFor: (target, now) => retentionCutoffMs(target.policy, now),
+			deleteBatch: (target, cutoff, limit) => this.operations.pruneBatch({
+				tableName: target.table,
+				column: target.column,
+				cutoff,
+				limit
+			})
+		});
+	}
 	async dangerouslyClearAll() {
 		await this.pool.execute(`DELETE FROM ${formatTableName(TABLE_WORKFLOW_SNAPSHOT)}`);
 		await this.pool.execute(`DELETE FROM ${RUN_OWNERS_TABLE}`);