如何防止选择和更新之间的竞争条件

Question

我正在做分类帐模块。在这个过程中，我必须按顺序完成这些任务

1. 从源中选择余额（table 1 row1）
2. 从目标中选择余额（table 1 row2）
3. 用一些逻辑修改余额
4. 更新源的余额（table 1 row1）
5. 更新目标余额（table 1 row2）
6. 提交更改
7. 将事务插入到事件表中。

在多线程环境中，线程在前一个线程更新和提交之前获得余额。在Postgres中，锁被强加给正在被访问的行，直到线程提交。在这种情况下，我看到了一种病态。

尝试对整个模式使用同步块。没有帮助。不能使用SELECT from UPDATE，因为更改余额的逻辑取决于其他操作。

以下是一些日志，在 [1] 线程转到更新部分之前，其他线程收集现有余额。

[DEBUG] [2016-12-15 10:49:53,893] [1] - Src Ledger_book_idLB001
[DEBUG] [2016-12-15 10:49:53,893] [1] - Src Balance2500.0
[DEBUG] [2016-12-15 10:49:53,897] [1] - Dest Ledger_book_idLB002
[DEBUG] [2016-12-15 10:49:53,897] [1] - Dest Balance0.0
[DEBUG] [2016-12-15 10:49:53,898] [15] - Src Ledger_book_idLB001
[DEBUG] [2016-12-15 10:49:53,898] [15] - Src Balance2500.0
[DEBUG] [2016-12-15 10:49:53,899] [16] - Src Ledger_book_idLB001
[DEBUG] [2016-12-15 10:49:53,899] [16] - Src Balance2500.0
[DEBUG] [2016-12-15 10:49:53,900] [16] - Dest Ledger_book_idLB002
[DEBUG] [2016-12-15 10:49:53,900] [15] - Dest Ledger_book_idLB002

任何帮助都值得赞赏:)如果对情况有任何具体疑问，请发表评论。

Answer 1

从文档中

当锁定子句出现在子选择中时，锁定的行是子查询返回给外部查询的行。这可能涉及比单独检查子查询所建议的更少的行，因为来自外部查询的条件可能用于优化子查询的执行。

所以你可以这样做...

您还应该阅读CTE的页面，尤其是DATA-MODIFYING

WITH s1 AS (
  SELECT balance FROM source
  FOR UPDATE
), s2 AS (
  SELECT balance destination
  FOR UPDATE
), u1 AS (
  UPDATE source SET balance = ...
  WHERE... -- potential join to S2 if you need destination
  RETURNING -- whatever you need (if anything)
), u2 AS (
  UPDATE source SET balance = ...
  WHERE... -- potential join to S1 if you need source
  RETURNING -- whatever you need (if anything)
)
INSERT INTO transactions (foo,date) VALUES (bar,now());