Django 事务的乐观“锁定”

Question

我有一个函数fn() 需要以原子方式执行一些数据库工作，这些工作依赖于某些数据集在其执行期间不会发生变化（大多数情况下都是如此）。

在 Django 中实现这一点的正确方法是什么？基本上我想做这样的事情：

def ensure_fn_runs_successfully():
  # While commit unsuccessful, keep trying
  while not fn():
    pass

@transaction.atomic
def fn():
  data = read_data_that_must_not_change()

  ... do some operations with the data and perform database operations ...

  # Assume it returns true if commit was successful, otherwise false
  return commit_only_if_the_data_actually_didnt_change()

@transaction.atomic 解决了部分问题（数据库应该只看到fn 运行之前或fn 成功运行之后的状态），但我不确定是否存在一个好的原语来做commit_only_if_the_data_actually_didnt_change，如果失败则重试操作。

要验证数据没有改变，只需检查查询返回的项目数是否与函数开始时相同即可；但是，我不知道是否有任何原语可以让您同时做出检查和提交决定/没有竞争条件。

Answer 1

如果您在一个事务块中，唯一可以改变您正在读取的数据的是同一事务块中的其他操作。因此，只要fn() 不对data 进行任何更改，就可以保证数据不会更改，除非fn() 更改它。这就是交易要解决的问题。

如果data 可以在fn() 的范围内发生变化，只需跟踪它发生变化的地方或跟踪最终结果。

@transaction.atomic
def fn():
  data = read_data_that_must_not_change()
  original_data = copy.copy(data)
  ... do some operations with the data and perform database operations ...

  # Assume it returns true if commit was successful, otherwise false
  if data != original_data:
    raise Exception('Oh no!  Data changed!') 
    # raising in exception is how you prevent transaction.atomic
    # from committing
  return commit_only_if_the_data_actually_didnt_change()

然后像这样处理你的while循环中的异常：

while True:
    try:
        fn()
        break
    except:
        time.sleep(10) # ten second cool off
        pass