@mastra/dsql 1.2.2 → 1.3.0-alpha.0

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 }) {
@@ -2224,7 +2268,7 @@ var MemoryDSQL = class MemoryDSQL extends MemoryStorage {
 			}, error);
 			this.logger?.error?.(mastraError.toString());
 			this.logger?.trackException(mastraError);
-			return { messages: [] };
+			throw mastraError;
 		}
 	}
 	async listMessages(args) {
@@ -2331,6 +2375,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 +2387,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 }) {