@mastra/dsql 1.2.2 → 1.3.0-alpha.1

package/dist/index.js

@@ -125,16 +125,21 @@ var PoolAdapter = class {
 		const client = await this.$pool.connect();
 		try {
 			await client.query("BEGIN");
-			const result = await callback(new TransactionClient(client));
-			await client.query("COMMIT");
-			return result;
-		} catch (error) {
+			const txClient = new TransactionClient(client);
 			try {
-				await client.query("ROLLBACK");
-			} catch (rollbackError) {
-				console.error("Transaction rollback failed:", rollbackError);
+				const result = await callback(txClient);
+				await txClient.drain();
+				await client.query("COMMIT");
+				return result;
+			} catch (error) {
+				await txClient.drain().catch(() => void 0);
+				try {
+					await client.query("ROLLBACK");
+				} catch (rollbackError) {
+					console.error("Transaction rollback failed:", rollbackError);
+				}
+				throw error;
 			}
-			throw error;
 		} finally {
 			client.release();
 		}
@@ -142,41 +147,85 @@ var PoolAdapter = class {
 };
 /**
 * Transaction client that wraps a PoolClient for executing queries within a transaction.
+*
+* Query methods are serialized through a tail promise (same pattern as
+* PinnedClientAdapter). Callers such as memory.updateMessages historically
+* did `queries.push(t.none(...))` then `await t.batch(queries)` — each
+* `t.none()` is async and starts `client.query` immediately, so by the time
+* batch runs, N queries are already in flight on one PoolClient. pg@8 queues
+* those internally and emits a DeprecationWarning; pg@9 will throw.
+* (#20820)
 */
 var TransactionClient = class {
 	client;
+	/**
+	* Serialization tail. Without this gate, concurrent t.none()/t.query()
+	* from Promise.all / batch land on the same PoolClient at once.
+	*/
+	#tail = Promise.resolve();
+	#error;
 	constructor(client) {
 		this.client = client;
 	}
-	async none(query, values) {
-		await this.client.query(query, values);
-		return null;
+	#enqueue(fn) {
+		const next = this.#tail.then(fn);
+		this.#tail = next.then(() => void 0, (error) => {
+			this.#error ??= { value: error };
+		});
+		return next;
 	}
-	async one(query, values) {
-		const result = await this.client.query(query, values);
-		if (result.rows.length === 0) throw new Error(`No data returned from query: ${truncateQuery(query)}`);
-		if (result.rows.length > 1) throw new Error(`Multiple rows returned when one was expected: ${truncateQuery(query)}`);
-		return result.rows[0];
+	/**
+	* Wait until every enqueued query has settled and surface the first failure.
+	* PoolAdapter calls this before COMMIT/ROLLBACK so those control
+	* statements never overlap in-flight work on the same client.
+	*/
+	async drain() {
+		await this.#tail;
+		if (this.#error) {
+			const { value } = this.#error;
+			this.#error = void 0;
+			throw value;
+		}
+	}
+	none(query, values) {
+		return this.#enqueue(async () => {
+			await this.client.query(query, values);
+			return null;
+		});
 	}
-	async oneOrNone(query, values) {
-		const result = await this.client.query(query, values);
-		if (result.rows.length === 0) return null;
-		if (result.rows.length > 1) throw new Error(`Multiple rows returned when one or none was expected: ${truncateQuery(query)}`);
-		return result.rows[0];
+	one(query, values) {
+		return this.#enqueue(async () => {
+			const result = await this.client.query(query, values);
+			if (result.rows.length === 0) throw new Error(`No data returned from query: ${truncateQuery(query)}`);
+			if (result.rows.length > 1) throw new Error(`Multiple rows returned when one was expected: ${truncateQuery(query)}`);
+			return result.rows[0];
+		});
 	}
-	async any(query, values) {
-		return (await this.client.query(query, values)).rows;
+	oneOrNone(query, values) {
+		return this.#enqueue(async () => {
+			const result = await this.client.query(query, values);
+			if (result.rows.length === 0) return null;
+			if (result.rows.length > 1) throw new Error(`Multiple rows returned when one or none was expected: ${truncateQuery(query)}`);
+			return result.rows[0];
+		});
 	}
-	async manyOrNone(query, values) {
+	any(query, values) {
+		return this.#enqueue(async () => {
+			return (await this.client.query(query, values)).rows;
+		});
+	}
+	manyOrNone(query, values) {
 		return this.any(query, values);
 	}
-	async many(query, values) {
-		const result = await this.client.query(query, values);
-		if (result.rows.length === 0) throw new Error(`No data returned from query: ${truncateQuery(query)}`);
-		return result.rows;
+	many(query, values) {
+		return this.#enqueue(async () => {
+			const result = await this.client.query(query, values);
+			if (result.rows.length === 0) throw new Error(`No data returned from query: ${truncateQuery(query)}`);
+			return result.rows;
+		});
 	}
-	async query(query, values) {
-		return this.client.query(query, values);
+	query(query, values) {
+		return this.#enqueue(() => this.client.query(query, values));
 	}
 	async batch(promises) {
 		return Promise.all(promises);
@@ -1976,6 +2025,7 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 				hasMore: perPageInput === false ? false : offset + perPage < total
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
 				id: createStorageErrorId("DSQL", "LIST_THREADS", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -1988,13 +2038,7 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException(mastraError);
-			return {
-				threads: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async saveThread({ thread }) {
@@ -2128,7 +2172,14 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			}, error);
 		});
 	}
-	async _getIncludedMessages({ include }) {
+	/**
+	* Fetches the messages named by `include` together with their surrounding context.
+	*
+	* @param include - Message ids to pin, each with an optional before/after window.
+	* @param resourceId - When set, restricts both the pinned messages and their context
+	* to that resource so an id from another resource returns nothing.
+	*/
+	async _getIncludedMessages({ include, resourceId }) {
 		if (!include || include.length === 0) return null;
 		const unionQueries = [];
 		const params = [];
@@ -2139,17 +2190,18 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 		});
 		for (const inc of include) {
 			const { id, withPreviousMessages = 0, withNextMessages = 0 } = inc;
+			const resourceCondition = resourceId ? ` AND "resourceId" = $${paramIdx + 3}` : "";
 			unionQueries.push(`
             SELECT * FROM (
               WITH target_thread AS (
-                SELECT thread_id FROM ${tableName} WHERE id = $${paramIdx}
+                SELECT thread_id FROM ${tableName} WHERE id = $${paramIdx}${resourceCondition}
               ),
               ordered_messages AS (
                 SELECT
                   *,
                   ROW_NUMBER() OVER (ORDER BY "createdAt" ASC) as row_num
                 FROM ${tableName}
-                WHERE thread_id = (SELECT thread_id FROM target_thread)
+                WHERE thread_id = (SELECT thread_id FROM target_thread)${resourceCondition}
               )
               SELECT
                 m.id,
@@ -2175,6 +2227,10 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
             `);
 			params.push(id, withPreviousMessages, withNextMessages);
 			paramIdx += 3;
+			if (resourceId) {
+				params.push(resourceId);
+				paramIdx += 1;
+			}
 		}
 		const finalQuery = unionQueries.join(" UNION ALL ") + " ORDER BY \"createdAt\" ASC";
 		const includedRows = await this.#db.client.manyOrNone(finalQuery, params);
@@ -2224,7 +2280,7 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException(mastraError);
-			return { messages: [] };
+			throw mastraError;
 		}
 	}
 	async listMessages(args) {
@@ -2300,7 +2356,10 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			};
 			const messageIds = new Set(messages.map((m) => m.id));
 			if (include && include.length > 0) {
-				const includeMessages = await this._getIncludedMessages({ include });
+				const includeMessages = await this._getIncludedMessages({
+					include,
+					resourceId
+				});
 				if (includeMessages) {
 					for (const includeMsg of includeMessages) if (!messageIds.has(includeMsg.id)) {
 						messages.push(includeMsg);
@@ -2331,6 +2390,7 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 				hasMore
 			};
 		} catch (error) {
+			if (error instanceof MastraError && error.category === ErrorCategory.USER) throw error;
 			const mastraError = new MastraError({
 				id: createStorageErrorId("DSQL", "LIST_MESSAGES", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -2342,13 +2402,7 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException(mastraError);
-			return {
-				messages: [],
-				total: 0,
-				page,
-				perPage: perPageForResponse,
-				hasMore: false
-			};
+			throw mastraError;
 		}
 	}
 	async saveMessages({ messages }) {