PostgreSQL 和锁定

Question

希望一些比我更聪明的 DBA 可以帮助我找到一个好的解决方案来完成我需要做的事情。

为了便于讨论，我们假设我有一个名为“work”的表，其中包含一些列，其中一列代表给定客户端对该行工作的所有权。场景是，我将连接 2 个客户端并轮询表以查找要完成的工作，当出现一行（或某些行）时，选择这些行的第一个客户端也会更新它们以暗示所有权，即更新将删除这些行，使其不会返回到任何其他客户端的选择。我的问题是，在这种情况下，我可以使用哪种锁定来防止 2 个客户端同时访问表，并且它们都通过 select 返回相同的行？

Answer 1

带有 RETURNING 子句的 UPDATE 语句就是执行此操作的方法。

UPDATE table
SET ownership = owner
RETURNING ( column list );

参考文献：

类似问题

文档

Answer 2

我的问题是，在这种情况下，我可以使用哪种锁定来防止 2 个客户端同时访问表，并且它们都通过 select 返回相同的行？

这里不需要锁定。

在 UPDATE 中，只需指定您只希望脚本在所有者仍为 null 时获取任务的所有权（假设这就是您标记未分配任务的方式）。这应该有效：

UPDATE foo SET owner = ? WHERE id = ? AND owner = ? WHERE owner IS NULL

如果修改的行数等于您预期的数量（或者 RETURNING 子句返回 @Ketema 建议的结果），那么您成功地获得了所有权。

---

虚假编辑，因为我在提交此答案之前注意到您的评论：

例如：2 个客户端同时发出该查询，他们没有机会操作相同的行？

正确的。您可能想阅读 MVCC。在事务外部运行这些语句将执行正确的操作。事务内的行为会有所不同。