PostgreSQL能否在单查询内更新表后统计行数并复用结果？

完全可以在单条SQL里实现你的需求，核心是把统计步骤也放到CTE（WITH子句）中，确保统计的是更新后的table1数据。

你原来的SQL逻辑方向没问题，调整结构后能更清晰地保证结果准确：

WITH updated_table1 AS (
  -- 第一步：更新table1并返回更新的行数据
  UPDATE table1
  SET enabled = true
  WHERE id = 'efgh'
  RETURNING data
),
enabled_count AS (
  -- 第二步：统计更新后table1中enabled=true的总行数
  SELECT COUNT(*) AS total
  FROM table1
  WHERE enabled = true
)
-- 第三步：用前面的结果更新table2
UPDATE table2 t
SET data = u.data, count = c.total
FROM updated_table1 u, enabled_count c
WHERE t.id = 'abcd'
RETURNING t.count;

逻辑说明： #

• PostgreSQL的WITH子句会按顺序执行所有CTE块，再执行主查询。所以enabled_count统计的是已经完成更新后的table1数据，完全包含刚被更新的行。
• 如果table1数据量很大，全表统计COUNT(*)可能有性能问题，也可以结合更新前的历史count值（比如存在其他表或变量中）加上本次更新的行数（SELECT COUNT(*) FROM updated_table1）来计算，但这种方式需要确保没有其他并发修改。

内容的提问来源于stack exchange，提问作者Marcus Ruddick