Ruby 数据库实现的分布式锁

807715333 · 2024年04月19日 · 最后由 hellorails 回复于 2024年05月22日 · 473 次阅读

请问下各位大佬，目前有基于数据库实现分布式锁的 gem 吗看了下只找到了一个 redlock，基于 redis 实现的但是项目中的版本比较旧，使用 redlock 的话需要升级 ruby 和 rails 版本，怕影响太大

crazyjacky #1 2024年04月20日

require 'active_record'
require 'mutex_m'

class ScheduleLock

  def initialize
    @locked = false
    @mutex = Mutex.new
  end

  def lock(name)
    @mutex.synchronize do
      ActiveRecord::Base.transaction do
        lock = ActiveLock.find_or_create_by!(name: name)
        host_name = `hostname`.strip
        if lock.locked?
          Rails.logger.error "Lock #{name} is already locked by #{lock.host}"
          return
        else
          lock.update!(locked: true, host: host_name)
          @locked = true
        end
      end
    end
  end

  def unlock(name)
    @mutex.synchronize do
      ActiveRecord::Base.transaction do
        lock = ActiveLock.find_or_create_by!(name: name)
        if lock.locked?
          lock.update!(locked: false)
          @locked = false
        end
      end
    end
  end

  def with_lock(name)
    lock(name)
    return unless @locked

    begin
      yield
    ensure
      unlock(name)
    end

  end
end