@mastra/mssql 1.8.0 → 1.9.0-alpha.0

package/dist/index.js

@@ -1,5 +1,5 @@
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
-import { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, AgentsStorage, BackgroundTasksStorage, MastraCompositeStore, MemoryStorage, ObservabilityStorage, SPAN_SCHEMA, ScoresStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TraceStatus, WORKFLOW_DEFINITIONS_SCHEMA, WorkflowDefinitionsStorage, WorkflowsStorage, calculatePagination, createStorageErrorId, getDefaultValue, listTracesArgsSchema, matchesExpectedWorkflowStatus, normalizePerPage, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, AgentsStorage, BackgroundTasksStorage, MastraCompositeStore, MemoryStorage, ObservabilityStorage, SPAN_SCHEMA, ScoresStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TraceStatus, WORKFLOW_DEFINITIONS_SCHEMA, WorkflowDefinitionsStorage, WorkflowsStorage, calculatePagination, createStorageErrorId, executeRetentionPrune, getDefaultValue, listTracesArgsSchema, matchesExpectedWorkflowStatus, normalizePerPage, resolveRetentionTargets, retentionCutoffMs, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import sql from "mssql";
 import { randomUUID } from "crypto";
 import { parseSqlIdentifier } from "@mastra/core/utils";
@@ -215,6 +215,14 @@ var MssqlDB = class extends MastraBase {
 			}, error);
 		}
 	}
+	async pruneBatch({ tableName, column, cutoff, limit }) {
+		if (!/^[A-Za-z_][A-Za-z0-9_]*$/.test(column)) throw new Error(`Invalid retention identifier: ${tableName}.${column}`);
+		const fullTableName = getTableName$1({
+			indexName: tableName,
+			schemaName: getSchemaName$1(this.schemaName)
+		});
+		return (await this.pool.request().input("limit", sql.Int, limit).input("cutoff", sql.DateTime2, cutoff).query(`DELETE TOP (@limit) FROM ${fullTableName} WHERE [${column}] < @cutoff`)).rowsAffected[0] ?? 0;
+	}
 	async clearTable({ tableName }) {
 		const fullTableName = getTableName$1({
 			indexName: tableName,
@@ -3127,6 +3135,11 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 //#endregion
 //#region src/storage/domains/observability/index.ts
 var ObservabilityMSSQL = class ObservabilityMSSQL extends ObservabilityStorage {
+	static retentionTables = { spans: {
+		table: TABLE_SPANS,
+		column: "startedAt",
+		indexed: true
+	} };
 	pool;
 	db;
 	schema;
@@ -3149,6 +3162,24 @@ var ObservabilityMSSQL = class ObservabilityMSSQL extends ObservabilityStorage {
 		this.skipDefaultIndexes = skipDefaultIndexes;
 		this.indexes = indexes?.filter((idx) => ObservabilityMSSQL.MANAGED_TABLES.includes(idx.table));
 	}
+	async prune(policies, options) {
+		return executeRetentionPrune({
+			domain: "observability",
+			targets: resolveRetentionTargets({
+				policies,
+				descriptor: ObservabilityMSSQL.retentionTables,
+				order: ["spans"]
+			}),
+			options,
+			cutoffFor: (target, now) => new Date(retentionCutoffMs(target.policy, now)),
+			deleteBatch: (target, cutoff, limit) => this.db.pruneBatch({
+				tableName: target.table,
+				column: target.column,
+				cutoff,
+				limit
+			})
+		});
+	}
 	async init() {
 		if (this.needsConnect) {
 			await this.pool.connect();
@@ -4801,7 +4832,8 @@ var MSSQLStore = class extends MastraCompositeStore {
 		super({
 			id: config.id,
 			name: "MSSQLStore",
-			disableInit: config.disableInit
+			disableInit: config.disableInit,
+			retention: config.retention
 		});
 		try {
 			this.schema = config.schemaName || "dbo";