1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
# frozen_string_literal: true
module WithAdvisoryLock
module Concern
extend ActiveSupport::Concern
delegate :with_advisory_lock, :with_advisory_lock!, :advisory_lock_exists?, to: 'self.class'
class_methods do
def with_advisory_lock(lock_name, options = {}, &block)
result = with_advisory_lock_result(lock_name, options, &block)
result.lock_was_acquired? ? result.result : false
end
def with_advisory_lock!(lock_name, options = {}, &block)
result = with_advisory_lock_result(lock_name, options, &block)
raise WithAdvisoryLock::FailedToAcquireLock, lock_name unless result.lock_was_acquired?
result.result
end
def with_advisory_lock_result(lock_name, options = {}, &block)
impl = impl_class.new(connection, lock_name, options)
impl.with_advisory_lock_if_needed(&block)
end
def advisory_lock_exists?(lock_name)
impl = impl_class.new(connection, lock_name, 0)
impl.already_locked? || !impl.yield_with_lock.lock_was_acquired?
end
def current_advisory_lock
lock_stack_key = WithAdvisoryLock::Base.lock_stack.first
lock_stack_key && lock_stack_key[0]
end
def current_advisory_locks
WithAdvisoryLock::Base.lock_stack.map(&:name)
end
private
def impl_class
adapter = WithAdvisoryLock::DatabaseAdapterSupport.new(connection)
if adapter.postgresql?
WithAdvisoryLock::PostgreSQL
elsif adapter.mysql?
WithAdvisoryLock::MySQL
else
WithAdvisoryLock::Flock
end
end
end
end
end
|
