Distributed Mutex with sidekiq-ent

distributed_mutex.rb

class DistributedMutex
  Blocked = Class.new(StandardError)

  DEFAULT_OPTIONS = {
    wait_timeout: 0,
    lock_timeout: 30,
    policy: :raise,
    ttl: 1.hour
  }

  def self.blocking(key, options = {}, &block)
    new(key, options, &block).blocking
  end

  attr_reader :key, :options, :block

  def initialize(key, options, &block)
    @options = options.reverse_merge(DEFAULT_OPTIONS)
    @key, @block = key, block
  end

  def blocking
    blocker.within_limit &block
  rescue Sidekiq::Limiter::OverLimit
    raise Blocked, key
  end

  private

  def blocker
    Sidekiq::Limiter.concurrent(
      "distributed-mutex-#{key}".parameterize, 1, options
    )
  end
end