Postgres.js 事务处理中的多语句执行问题解析

Postgres.js 是一个流行的 Node.js PostgreSQL 客户端库，提供了简洁的 API 来处理数据库操作。在实际开发中，我们经常会遇到需要在事务中执行多个 SQL 语句的场景。本文将深入探讨 Postgres.js 中处理多语句执行的各种方法及其背后的原理。

问题背景

在 PostgreSQL 协议层面，当使用参数化查询时，不允许在单个查询中包含多个 SQL 语句。这是 PostgreSQL 协议的一个限制。开发者通常会尝试以下两种方式：

1. 使用分号分隔的多语句查询
2. 使用事务块中的多个独立查询

多语句查询的限制

尝试执行如下代码会导致错误：

await postgresClient.begin(async sql => {
  const q = `INSERT INTO table1; INSERT INTO table2`;
  const res = await sql.unsafe(q, [...], { prepare: false });
});

即使设置了 prepare: false，这种写法仍然会失败，因为 PostgreSQL 协议不允许在参数化查询中包含多个语句。

正确的解决方案

方案一：使用简单查询

如果确实需要在一个查询中执行多个语句，可以使用 .simple() 方法：

await sql`INSERT INTO table1; INSERT INTO table2`.simple();

需要注意的是，这种方式需要开发者自行确保 SQL 注入的安全性，因为参数不会被自动转义。

方案二：使用事务管道

更推荐的做法是使用事务管道，这可以保持原子性的同时执行多个查询：

const result = await sql.begin(async sql => {
  await sql`SELECT * FROM table1 WHERE uid = ${uid} FOR UPDATE`;
  return Promise.all([
    sql`INSERT INTO table1 ...`,
    sql`INSERT INTO table2 ...`
  ]);
});

这种方式的优势在于：

1. 保持了事务的原子性
2. 每个查询都可以安全地使用参数化
3. 查询会被自动优化执行

方案三：保持执行顺序的管道

如果需要保持查询的执行顺序，可以使用如下方式：

const result = await sql.begin(async sql => {
  const results = [];
  results.push(await sql`INSERT INTO table1 ...`);
  results.push(await sql`INSERT INTO table2 ...`);
  return results;
});

技术原理

PostgreSQL 协议在处理查询时有两种模式：

1. 简单查询协议：可以包含多个语句，但不能使用参数绑定
2. 扩展查询协议：支持参数绑定，但每个查询只能包含一个语句

Postgres.js 会根据查询的构成自动选择合适的协议。当使用参数化查询时，会自动使用扩展查询协议，因此不能包含多个语句。

最佳实践建议

1. 优先使用事务管道（方案二）来处理多个相关操作
2. 只有在确实需要且能确保安全的情况下才使用简单查询
3. 考虑使用 CTE (Common Table Expressions) 将多个操作合并到一个查询中
4. 对于复杂业务逻辑，可以结合应用层处理和多步查询

通过理解这些原理和方法，开发者可以更高效地使用 Postgres.js 处理复杂的数据库操作场景。