@mastra/mssql 1.10.5 → 1.10.6-alpha.1

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-mssql
 description: Documentation for @mastra/mssql. Use when working with @mastra/mssql APIs, configuration, or implementation.
 metadata:
   package: "@mastra/mssql"
-  version: "1.10.5"
+  version: "1.10.6-alpha.1"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.10.5",
+  "version": "1.10.6-alpha.1",
   "package": "@mastra/mssql",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-storage-retention.md

@@ -294,7 +294,7 @@ const storage = new MongoDBStore({
 
 ## Reclaiming disk
 
-`prune()` deletes rows but doesn't shrink the database file. On SQLite/libSQL the freed pages go on a freelist and are reused by future writes, so the file stops growing: for most users this alone solves the unbounded-growth problem.
+`prune()` deletes rows but doesn't shrink the database file. On SQLite/libSQL the freed pages go on a freelist for reuse by future writes. Reusing these pages prevents unbounded file growth for most users.
 
 Handing that free space back to the OS is a separate concern that Mastra doesn't manage. If you specifically need to shrink the file, run the underlying database's compaction (for example `VACUUM` on self-hosted libSQL) yourself in a maintenance window. A full `VACUUM` locks the file and needs roughly twice the file size in free disk. On PostgreSQL, autovacuum reclaims dead tuples for reuse automatically. A manual `VACUUM FULL` is only needed if you must return disk to the OS.
 

package/dist/index.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-//#region \0rolldown/runtime.js
+// \0rolldown/runtime.js
 var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -20,7 +20,6 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 	value: mod,
 	enumerable: true
 }) : target, mod));
-//#endregion
 let _mastra_core_error = require("@mastra/core/error");
 let _mastra_core_storage = require("@mastra/core/storage");
 let mssql = require("mssql");
@@ -29,7 +28,7 @@ let _mastra_core_utils = require("@mastra/core/utils");
 let _mastra_core_base = require("@mastra/core/base");
 let _mastra_core_agent = require("@mastra/core/agent");
 let _mastra_core_evals = require("@mastra/core/evals");
-//#region src/storage/db/utils.ts
+// src/storage/db/utils.ts
 function getSchemaName$1(schema) {
 	return schema ? `[${(0, _mastra_core_utils.parseSqlIdentifier)(schema, "schema name")}]` : void 0;
 }
@@ -38,8 +37,7 @@ function getTableName$1({ indexName, schemaName }) {
 	const quotedSchemaName = schemaName;
 	return quotedSchemaName ? `${quotedSchemaName}.${quotedIndexName}` : quotedIndexName;
 }
-//#endregion
-//#region src/storage/db/index.ts
+// src/storage/db/index.ts
 /**
 * Resolves MssqlDomainConfig to pool and schema.
 * Domain classes create their own MssqlDB instance from the returned pool.
@@ -1057,8 +1055,7 @@ MIGRATION REQUIRED: Duplicate spans detected in ${duplicateInfo.tableName}\n====
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/utils.ts
+// src/storage/domains/utils.ts
 function getSchemaName(schema) {
 	return schema ? `[${(0, _mastra_core_utils.parseSqlIdentifier)(schema, "schema name")}]` : void 0;
 }
@@ -1180,8 +1177,7 @@ function transformFromSqlRow({ tableName, sqlRow }) {
 	});
 	return result;
 }
-//#endregion
-//#region src/storage/domains/agents/index.ts
+// src/storage/domains/agents/index.ts
 var AgentsMSSQL = class AgentsMSSQL extends _mastra_core_storage.AgentsStorage {
 	pool;
 	schema;
@@ -1769,8 +1765,7 @@ var AgentsMSSQL = class AgentsMSSQL extends _mastra_core_storage.AgentsStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/background-tasks/index.ts
+// src/storage/domains/background-tasks/index.ts
 function serializeJson(v) {
 	if (typeof v === "object" && v != null) return JSON.stringify(v);
 	return v ?? null;
@@ -2131,8 +2126,7 @@ var BackgroundTasksMSSQL = class BackgroundTasksMSSQL extends _mastra_core_stora
 		return Number(result.recordset[0]?.count ?? 0);
 	}
 };
-//#endregion
-//#region src/storage/db/run-fencing.ts
+// src/storage/db/run-fencing.ts
 /**
 * Epoch milliseconds on the database clock. SQL Server evaluates
 * `SYSUTCDATETIME()` once when a statement starts, so a statement that waits
@@ -2191,16 +2185,23 @@ function withRunFence(pool, claimsTable, fence, operation, write) {
 * duplicate key if they don't. On retry the row exists and the claim
 * serializes behind its row lock.
 */
-async function claimTransaction(pool, fn) {
+function claimTransaction(pool, fn) {
+	return retryWriteRace(() => inTransaction(pool, fn));
+}
+/**
+* Runs `write` again when it lost a race to insert the same row: a deadlock
+* victim or a duplicate key. `write` must be a whole statement or
+* transaction, since a deadlock victim's transaction has already ended.
+*/
+async function retryWriteRace(write) {
 	for (let attempt = 1;; attempt++) try {
-		return await inTransaction(pool, fn);
+		return await write();
 	} catch (error) {
 		const number = error.number;
 		if (attempt >= MAX_CLAIM_ATTEMPTS || !(number === ERR_DEADLOCK_VICTIM || number === ERR_PRIMARY_KEY || number === ERR_UNIQUE_INDEX)) throw error;
 	}
 }
-//#endregion
-//#region src/storage/domains/memory/index.ts
+// src/storage/domains/memory/index.ts
 function bindMssqlMetadataParams(request, params) {
 	for (const [paramName, paramValue] of Object.entries(params)) request.input(paramName, paramValue);
 }
@@ -3360,8 +3361,7 @@ var MemoryMSSQL = class MemoryMSSQL extends _mastra_core_storage.MemoryStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/observability/index.ts
+// src/storage/domains/observability/index.ts
 var ObservabilityMSSQL = class ObservabilityMSSQL extends _mastra_core_storage.ObservabilityStorage {
 	static retentionTables = { spans: {
 		table: _mastra_core_storage.TABLE_SPANS,
@@ -4002,8 +4002,7 @@ var ObservabilityMSSQL = class ObservabilityMSSQL extends _mastra_core_storage.O
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/scores/index.ts
+// src/storage/domains/scores/index.ts
 /**
 * MSSQL-specific score row transformation.
 * Converts timestamp strings to Date objects.
@@ -4419,8 +4418,7 @@ var ScoresMSSQL = class ScoresMSSQL extends _mastra_core_storage.ScoresStorage {
 		}
 	}
 };
-//#endregion
-//#region src/storage/domains/workflow-definitions/index.ts
+// src/storage/domains/workflow-definitions/index.ts
 function parseJson(value, column, rowId) {
 	if (value == null) return null;
 	if (typeof value === "string") try {
@@ -4623,8 +4621,7 @@ var WorkflowDefinitionsMSSQL = class WorkflowDefinitionsMSSQL extends _mastra_co
 		await request.query(`DELETE FROM ${tableName} WHERE [id] = @id`);
 	}
 };
-//#endregion
-//#region src/storage/domains/workflows/index.ts
+// src/storage/domains/workflows/index.ts
 const RUN_OWNER_COLUMNS = `[generation], [ownerId], [leaseExpiresAt], ${DB_NOW_MS} AS [nowMs]`;
 function toRunOwnershipRecord(runId, row) {
 	const leaseExpiresAt = row.leaseExpiresAt === null ? null : Number(row.leaseExpiresAt);
@@ -4670,6 +4667,9 @@ var WorkflowsMSSQL = class WorkflowsMSSQL extends _mastra_core_storage.Workflows
 		this.skipDefaultIndexes = skipDefaultIndexes;
 		this.indexes = indexes?.filter((idx) => WorkflowsMSSQL.MANAGED_TABLES.includes(idx.table));
 	}
+	supportsGuardedSnapshotWrites() {
+		return true;
+	}
 	supportsConcurrentUpdates() {
 		return true;
 	}
@@ -4979,14 +4979,16 @@ var WorkflowsMSSQL = class WorkflowsMSSQL extends _mastra_core_storage.Workflows
 			}, error);
 		}
 	}
-	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, fence }) {
+	async persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, fence, expectedStatus }) {
 		const table = getTableName({
 			indexName: _mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT,
 			schemaName: getSchemaName(this.schema)
 		});
 		const now = (/* @__PURE__ */ new Date()).toISOString();
+		const expectedStatuses = expectedStatus === void 0 ? void 0 : Array.isArray(expectedStatus) ? expectedStatus : [expectedStatus];
+		const guard = expectedStatuses === void 0 ? "" : expectedStatuses.length === 0 ? " AND 1 = 0" : ` AND JSON_VALUE(target.[snapshot], '$.status') IN (${expectedStatuses.map((_, i) => `@expectedStatus${i}`).join(", ")})`;
 		try {
-			await withRunFence(this.pool, this.#runOwnersTable, (0, _mastra_core_storage.resolveRunFence)(this, fence, runId), "persistWorkflowSnapshot", async (q) => {
+			return await retryWriteRace(() => withRunFence(this.pool, this.#runOwnersTable, (0, _mastra_core_storage.resolveRunFence)(this, fence, runId), "persistWorkflowSnapshot", async (q) => {
 				const request = q.request();
 				request.input("workflow_name", workflowName);
 				request.input("run_id", runId);
@@ -4994,17 +4996,18 @@ var WorkflowsMSSQL = class WorkflowsMSSQL extends _mastra_core_storage.Workflows
 				request.input("snapshot", JSON.stringify(snapshot));
 				request.input("createdAt", mssql.default.DateTime2, new Date(now));
 				request.input("updatedAt", mssql.default.DateTime2, new Date(now));
+				expectedStatuses?.forEach((status, i) => request.input(`expectedStatus${i}`, status));
 				const mergeSql = `MERGE INTO ${table} AS target
             USING (SELECT @workflow_name AS workflow_name, @run_id AS run_id) AS src
             ON target.workflow_name = src.workflow_name AND target.run_id = src.run_id
-            WHEN MATCHED THEN UPDATE SET
+            WHEN MATCHED${guard} THEN UPDATE SET
               resourceId = @resourceId,
               snapshot = @snapshot,
               [updatedAt] = @updatedAt
             WHEN NOT MATCHED THEN INSERT (workflow_name, run_id, resourceId, snapshot, [createdAt], [updatedAt])
               VALUES (@workflow_name, @run_id, @resourceId, @snapshot, @createdAt, @updatedAt);`;
-				await request.query(mergeSql);
-			});
+				return ((await request.query(mergeSql)).rowsAffected[0] ?? 0) > 0;
+			}));
 		} catch (error) {
 			if ((0, _mastra_core_storage.isRunFenceConflictError)(error)) throw error;
 			throw new _mastra_core_error.MastraError({
@@ -5018,6 +5021,27 @@ var WorkflowsMSSQL = class WorkflowsMSSQL extends _mastra_core_storage.Workflows
 			}, error);
 		}
 	}
+	async loadWorkflowRunStatus({ workflowName, runId }) {
+		try {
+			const request = this.pool.request();
+			request.input("workflow_name", workflowName);
+			request.input("run_id", runId);
+			return (await request.query(`SELECT TOP 1 JSON_VALUE([snapshot], '$.status') AS status FROM ${getTableName({
+				indexName: _mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT,
+				schemaName: getSchemaName(this.schema)
+			})} WHERE workflow_name = @workflow_name AND run_id = @run_id`)).recordset?.[0]?.status ?? null;
+		} catch (error) {
+			throw new _mastra_core_error.MastraError({
+				id: (0, _mastra_core_storage.createStorageErrorId)("MSSQL", "LOAD_WORKFLOW_RUN_STATUS", "FAILED"),
+				domain: _mastra_core_error.ErrorDomain.STORAGE,
+				category: _mastra_core_error.ErrorCategory.THIRD_PARTY,
+				details: {
+					workflowName,
+					runId
+				}
+			}, error);
+		}
+	}
 	async loadWorkflowSnapshot({ workflowName, runId }) {
 		try {
 			const result = await this.db.load({
@@ -5161,8 +5185,7 @@ var WorkflowsMSSQL = class WorkflowsMSSQL extends _mastra_core_storage.Workflows
 		}
 	}
 };
-//#endregion
-//#region src/storage/index.ts
+// src/storage/index.ts
 /**
 * Type guard for pre-configured pool config
 */
@@ -5275,12 +5298,8 @@ var MSSQLStore = class extends _mastra_core_storage.MastraCompositeStore {
 		}
 	}
 	async _performInitializationAndStore() {
-		try {
-			await this.pool.connect();
-			return true;
-		} catch (err) {
-			throw err;
-		}
+		await this.pool.connect();
+		return true;
 	}
 	/**
 	* Closes the MSSQL connection pool.
@@ -5291,7 +5310,6 @@ var MSSQLStore = class extends _mastra_core_storage.MastraCompositeStore {
 		await this.pool.close();
 	}
 };
-//#endregion
 exports.AgentsMSSQL = AgentsMSSQL;
 exports.BackgroundTasksMSSQL = BackgroundTasksMSSQL;
 exports.MSSQLStore = MSSQLStore;