竞争条件的PostgreSQL和ActiveRecord子选择

Question

我在使用 PostgreSQL 的 ActiveRecord 中遇到了竞争条件，我正在读取一个值，然后递增它并插入一条新记录：

num = Foo.where(bar_id: 42).maximum(:number)
Foo.create!({
  bar_id: 42,
  number: num + 1
}) 

按比例，多个线程将同时读取并写入number的相同值。将其封装在事务中并不能修复竞争条件，因为SELECT不会锁定表。我不能使用自动递增，因为number不是唯一的，它只在给定特定的bar_id时才是唯一的。我看到3种可能的修复方法：

> < li>

显式使用postgres锁(行级锁？)

使用唯一约束并在失败时重试（恶心！）

覆盖保存以使用子选择，即。

< code >插入foo (bar_id，number)值(42，(从foo中选择MAX(number) 1，其中bar _ id = 42))；

所有这些解决方案看起来好像我要重新实现< code>ActiveRecord::Base#save的大部分内容！有没有更简单的方法？

更新:我以为我用< code>Foo.lock找到了答案(真)。其中(bar_id: 42)。最大值(:number)，但它使用了聚合查询中不允许的< code > SELECT FOR update

更新 2：我们的 DBA 刚刚通知我，即使我们可以做 插入 INTO foo（bar_id，数字）值（42，（从 foo 中选择 MAX（数字） 1，其中 bar_id = 42））; 这并不能解决任何问题，因为 SELECT 在与 INSERT 不同的锁中运行

Answer 1

您的选项包括：

>

在＜code＞SERIALIZABLE＜/code＞隔离中运行。由于序列化失败，相互依赖的事务将在提交时中止。你会收到很多错误日志垃圾邮件，你会做很多重试，但它会可靠地工作。

正如您所提到的，定义一个< code>UNIQUE约束，并在失败时重试。与上述问题相同。

如果有父对象，则可以选择...FOR 在执行最大查询之前更新父对象。在这种情况下，您将从栏中选择 1，其中 bar_id = $1 进行更新。您正在使用 bar 作为该bar_id的所有 foo的锁。然后，您可以知道继续是安全的，只要执行计数器增量的每个查询都可靠地执行此操作。这可以很好地工作。

这仍然会为每个调用执行聚合查询，这（按下一个选项）是不必要的，但至少不会像上面的选项那样垃圾邮件发送错误日志。

使用计数器表。这就是我要做的。要么在bar中，要么在bar_foo_counter这样的侧表中，使用获取行ID

UPDATE bar_foo_counter SET counter = counter + 1
WHERE bar_id = $1 RETURNING counter

或者，如果您的框架无法处理 RETURNING，则效率较低的选项：

SELECT counter FROM bar_foo_counter
WHERE bar_id = $1 FOR UPDATE;

UPDATE bar_foo_counter SET counter = $1;

然后，在同一事务中，将生成的计数器行用于编号。提交时，bar_id的计数器表行将被解锁，以便下一个查询使用。如果回滚，更改将被丢弃。

我建议使用计数器方法，为计数器使用专用的边表，而不是在bar中添加列。这对模型更为简洁，意味着您可以在bar中创建更少的更新膨胀，这可以降低对bar的查询速度。