在 Postgresql 中锁定事务（一系列选择和更新）

Question

我有一个表和几个线程（系统），它们从表中读取特定列并更新它。所以我的查询如下：

Begin;
SELECT col FROM tbl WHERE <condition1>
UPDATE tbl SET col = col + 1 WHERE <condition1>
Commit;

我想确保在任何时候只有一个线程可以选择和更新该列。什么是最好的解决方案？

PS：据我所知，在 SQL Server 中使用事务会锁定事务中的所有资源，但我不知道 PostgreSQL。

谢谢！

Answer 1

SQL 中最高级的锁是行锁。

无法锁定列，数据库总是在整行（该行的所有列）上加锁。

你可以使用a SELECT .. FOR UPDATE clause:

Begin;
SELECT col FROM tbl WHERE <condition1> FOR UPDATE;
UPDATE tbl SET col = col + 1 WHERE <condition1> ;
Commit;

SELECT .. FOR UPDATE 的工作方式与普通的 select 类似（它从表中检索行），但另外它会锁定所有检索到的行，防止它们被其他事务修改，直到当前事务结束。

你也可以使用RETURNING clause in UPDATE statement：

Begin;
UPDATE tbl SET col = col + 1 WHERE <condition1> 
   RETURNING col - 1 AS col;
Commit;

UPDATE 语句总是在修改的记录上放置一个写锁。

RETURNING 子句导致返回值，和普通的SELECT 一样，但是它返回修改后的行值（更新后的行值），因此在上面的示例中使用了col - 1 表达式来计算增加之前的列值。

请看一下这个演示 --> http://www.sqlfiddle.com/#!15/4d0f4/3

第二种方式（UPDATE+RETURNING子句）的优点是记录只被访问一次，相对于第一种方式，当行必须先被SELECT语句选中时（第一次访问），那么它使用UPDATE 语句更新（第二次访问）。

Answer 2

kordirko 已正确指出 SELECT ... FOR UPDATE 是用于此目的的基本功能。

请注意，它不会使用谓词锁。所以如果你SELECT id FROM sometable WHERE id = 42 FOR UPDATE 并且没有id = 42 的行，没有锁定。没有在 ID 42 上创建“保留”。其他人可以在您的 SELECT 和随后的 INSERT 之间创建它。

如果您想解决这个问题，您可能希望查看SERIALIZABLE 隔离模式。我认为您习惯的 MS SQL Server 默认为 SERIALIZABLE 隔离。

您可以改为：

BEGIN ISOLATION LEVEL SERIALIZABLE;
SELECT col FROM tbl WHERE <condition1>
UPDATE tbl SET col = col + 1 WHERE <condition1>
COMMIT;

我强烈建议设置一些并发测试来证明这当然可以按您的预期工作。