@mastra/dynamodb 1.4.2 → 1.4.3-alpha.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-dynamodb
 description: Documentation for @mastra/dynamodb. Use when working with @mastra/dynamodb APIs, configuration, or implementation.
 metadata:
   package: "@mastra/dynamodb"
-  version: "1.4.2"
+  version: "1.4.3-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.4.2",
+  "version": "1.4.3-alpha.0",
   "package": "@mastra/dynamodb",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -1687,9 +1687,52 @@ async function goWithRunFence(service, check, write) {
 		await backoff(attempt);
 	}
 }
+/**
+* Deletes up to `limit` items of a run claim entity whose epoch-ms `attribute`
+* is older than `cutoff`. These entities have no index on the anchor, so this
+* scans for them. Each delete repeats the cutoff as its condition, so an item
+* claimed or raised again since the scan is kept.
+*/
+async function pruneRunClaims(service, entityName, attribute, cutoff, limit) {
+	const entity = service.entities[entityName];
+	const olderThanCutoff = (attr, op) => op.lt(attr[attribute], cutoff);
+	let deleted = 0;
+	let cursor = null;
+	do {
+		const page = await entity.scan.where(olderThanCutoff).go({
+			cursor,
+			limit: limit - deleted
+		});
+		for (const item of page.data) {
+			try {
+				await retryOnTransactionConflict(() => entity.delete({
+					entity: entityName,
+					run_id: item.run_id
+				}).where(olderThanCutoff).go());
+				deleted++;
+			} catch (error) {
+				if (!isConditionalCheckFailed(error)) throw error;
+			}
+			if (deleted >= limit) return deleted;
+		}
+		cursor = page.cursor;
+	} while (cursor);
+	return deleted;
+}
 //#endregion
 //#region src/storage/domains/memory/index.ts
-var MemoryStorageDynamoDB = class extends _mastra_core_storage.MemoryStorage {
+var MemoryStorageDynamoDB = class MemoryStorageDynamoDB extends _mastra_core_storage.MemoryStorage {
+	/**
+	* Tables `prune()` can clean up. Run fences anchor on `retiredAt` (epoch ms), which stays unset 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"
+	} };
 	supportsPartialThreadUpdate = true;
 	service;
 	ttlConfig;
@@ -1702,6 +1745,19 @@ var MemoryStorageDynamoDB = class extends _mastra_core_storage.MemoryStorage {
 	supportsRunFencing() {
 		return true;
 	}
+	async prune(policies, options) {
+		return (0, _mastra_core_storage.executeRetentionPrune)({
+			domain: "memory",
+			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
+				policies,
+				descriptor: MemoryStorageDynamoDB.retentionTables,
+				order: ["runFences"]
+			}),
+			options,
+			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
+			deleteBatch: (_target, cutoff, limit) => pruneRunClaims(this.service, "memory_run_fence", "retiredAt", Number(cutoff), limit)
+		});
+	}
 	async dangerouslyClearAll() {
 		await deleteTableData(this.service, _mastra_core_storage.TABLE_THREADS);
 		await deleteTableData(this.service, _mastra_core_storage.TABLE_MESSAGES);
@@ -2833,7 +2889,18 @@ function toRunOwnershipRecord(item) {
 }
 const MAX_RETRIES = 5;
 const BASE_DELAY_MS = 50;
-var WorkflowStorageDynamoDB = class extends _mastra_core_storage.WorkflowsStorage {
+var WorkflowStorageDynamoDB = class WorkflowStorageDynamoDB extends _mastra_core_storage.WorkflowsStorage {
+	/**
+	* Tables `prune()` can clean up. Run ownership records anchor on `updatedAt` (epoch ms), which every
+	* claim, renewal and release refreshes, so a live execution's record 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"
+	} };
 	service;
 	ttlConfig;
 	constructor(config) {
@@ -2848,6 +2915,19 @@ var WorkflowStorageDynamoDB = class extends _mastra_core_storage.WorkflowsStorag
 	supportsRunFencing() {
 		return true;
 	}
+	async prune(policies, options) {
+		return (0, _mastra_core_storage.executeRetentionPrune)({
+			domain: "workflows",
+			targets: (0, _mastra_core_storage.resolveRetentionTargets)({
+				policies,
+				descriptor: WorkflowStorageDynamoDB.retentionTables,
+				order: ["runOwnership"]
+			}),
+			options,
+			cutoffFor: (target, now) => (0, _mastra_core_storage.retentionCutoffMs)(target.policy, now),
+			deleteBatch: (_target, cutoff, limit) => pruneRunClaims(this.service, "workflow_run_owner", "updatedAt", Number(cutoff), limit)
+		});
+	}
 	async dangerouslyClearAll() {
 		await deleteTableData(this.service, _mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT);
 		await deleteTableData(this.service, _mastra_core_storage.TABLE_WORKFLOW_RUN_OWNERS);