@mastra/mssql 1.7.5 → 1.8.0-alpha.0

package/dist/index.js

@@ -2486,6 +2486,65 @@ var MemoryMSSQL = class MemoryMSSQL extends MemoryStorage {
 			}, error);
 		}
 	}
+	async updateThreadResourceId({ threadId, resourceId }) {
+		const threadsTable = getTableName({
+			indexName: TABLE_THREADS,
+			schemaName: getSchemaName(this.schema)
+		});
+		const messagesTable = getTableName({
+			indexName: TABLE_MESSAGES,
+			schemaName: getSchemaName(this.schema)
+		});
+		const tx = this.pool.transaction();
+		try {
+			await tx.begin();
+			const selectReq = tx.request();
+			selectReq.input("threadId", threadId);
+			const row = (await selectReq.query(`SELECT id, [resourceId], title, metadata, [createdAt], [updatedAt]
+         FROM ${threadsTable} WITH (UPDLOCK, HOLDLOCK)
+         WHERE id = @threadId`)).recordset[0];
+			if (!row) throw new Error(`Thread "${threadId}" not found`);
+			const normalized = {
+				id: row.id,
+				resourceId: row.resourceId,
+				title: row.title,
+				metadata: typeof row.metadata === "string" ? JSON.parse(row.metadata) : row.metadata,
+				createdAt: row.createdAt,
+				updatedAt: row.updatedAt
+			};
+			if (row.resourceId === resourceId) {
+				await tx.commit();
+				return normalized;
+			}
+			const now = /* @__PURE__ */ new Date();
+			const updateThreadReq = tx.request();
+			updateThreadReq.input("resourceId", resourceId);
+			updateThreadReq.input("updatedAt", now);
+			updateThreadReq.input("threadId", threadId);
+			await updateThreadReq.query(`UPDATE ${threadsTable} SET [resourceId] = @resourceId, [updatedAt] = @updatedAt WHERE id = @threadId`);
+			const updateMessagesReq = tx.request();
+			updateMessagesReq.input("resourceId", resourceId);
+			updateMessagesReq.input("threadId", threadId);
+			await updateMessagesReq.query(`UPDATE ${messagesTable} SET [resourceId] = @resourceId WHERE [thread_id] = @threadId`);
+			await tx.commit();
+			return {
+				...normalized,
+				resourceId,
+				updatedAt: now
+			};
+		} catch (error) {
+			await tx.rollback().catch(() => {});
+			throw new MastraError({
+				id: createStorageErrorId("MSSQL", "UPDATE_THREAD_RESOURCE_ID", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: {
+					threadId,
+					resourceId
+				}
+			}, error);
+		}
+	}
 	_sortMessages(messages, field, direction) {
 		const mult = direction === "ASC" ? 1 : -1;
 		return messages.sort((a, b) => {