Rails 乐观锁这次应该说清楚了吧

flydragon · May 11, 2019 · Last by hwang replied at October 13, 2020 · 1340 hits

概要

希望从实例到源代码,逐步解析乐观锁的工作流程,希望对你有所帮助。

实例

# 添加lock_version列
Store.attribute_names # => ["id", "name", "number", "lock_version", "created_at", "updated_at"]

s1 = Store.first
s2 = Store.first

s1.number += 1
s.save # 正常

s2.number += 1
s2.save # 异常
-- sql日志
UPDATE `stores` SET `number` = 3, `updated_at` = '2019-05-11 01:15:19', `lock_version` = 1 WHERE `stores`.`id` = 1 AND `stores`.`lock_version` = 0

实例分析

  1. 递增 lock_version 追加到更新字段后面 (lock_version = 1)
  2. 当前 lock_version 合并到更新条件后面 (lock_version = 0)
  3. 执行更新 sql,返回更新行数,理论上如果没有并发的话,受影响行数一定为 1。
    • 更新行数为 1,说明当前版本号最新,期间无并发操作。
    • 更新行数不为 1,说明当前版本号失效,有并发操作,抛出异常ActiveRecord::StaleObjectError

源代码分析

代码位置ActiveRecord::Locking::Optimistic

def _update_row(attribute_names, attempted_action = "update")
  return super unless locking_enabled?

  begin
    # 获取乐观锁版本控制列,默认为lock_version
    locking_column = self.class.locking_column
    previous_lock_value = read_attribute_before_type_cast(locking_column)
    # lock_version追加到更新字段中
    attribute_names << locking_column

    # lock_version + 1
    self[locking_column] += 1

    # 执行更新操作,返回受影响行数
    # 更新字段:attribute_names + lock_version
    # 查询条件:primary_key + lock_version
    affected_rows = self.class._update_record(
      attributes_with_values(attribute_names), # 更新字段
      self.class.primary_key => id_in_database, # 主键查询条件
      locking_column => previous_lock_value # 版本号查询条件
    )

    # 受影响行数不为1,则说明有并发,抛出异常
    if affected_rows != 1
      raise ActiveRecord::StaleObjectError.new(self, attempted_action)
    end

    affected_rows

  # If something went wrong, revert the locking_column value.
  rescue Exception
    self[locking_column] = previous_lock_value.to_i
    raise
  end
end

总结

更新时检查当前读取记录 lock_version 是否为最新,来判断是否有并发的存在。

  1. 乐观锁通过版本号来检测并发
  2. 检测的条件是,在当前记录版本号下执行更新受影响的行数是否为 1
  3. 出现并发抛出异常ActiveRecord::StaleObjectError
  4. 并发处理,需要手动处理(捕获异常-》并发处理)
  5. 处理方法包括重试,具体可参考Rails 中乐观锁与悲观锁的使用
flydragon closed this topic. 11 May 20:48
flydragon reopened this topic. 11 May 20:48
You need to Sign in before reply, if you don't have an account, please Sign up first.