PostgreSQL中能否将SELECT ... FOR UPDATE与UPDATE ... RETURNING合并为单查询？

完全可以把这两个查询合并成一个，既保留行锁定逻辑，又避免重复筛选，以下是两种适配不同场景的实现方式：

场景1：无需严格锁定顺序（最简方案） #

直接使用带RETURNING的UPDATE语句即可，PostgreSQL在执行UPDATE时会自动锁定匹配的行，无需额外的SELECT FOR UPDATE：

UPDATE table_name
SET id = id + 1
WHERE id IN (ids)
RETURNING *;

这条语句会一次性完成筛选目标行→锁定→更新→返回结果的全部流程，完全替代原来的两条查询，效率更高。

场景2：需要按指定顺序锁定（避免死锁） #

如果你的业务中存在多事务并发修改这些ID行的情况，为了避免死锁，需要严格按照ids的顺序锁定行，可以用CTE（公共表表达式）先锁定行再更新：

WITH locked_rows AS (
    SELECT id
    FROM table_name
    WHERE id IN (ids)
    ORDER BY array_position(ARRAY[ids], id)  -- 按ids数组的顺序排序锁定
    FOR UPDATE
)
UPDATE table_name
SET id = id + 1
FROM locked_rows
WHERE table_name.id = locked_rows.id
RETURNING *;

这里通过array_position函数确保锁定顺序和你传入的ids数组顺序一致，避免因锁定顺序不一致引发死锁，同时只做一次筛选，更新后直接返回结果。

内容的提问来源于stack exchange，提问作者Marco Groot