Class: WithAdvisoryLock::Base

Inherits:
Object
  • Object
show all
Defined in:
lib/with_advisory_lock/base.rb

Direct Known Subclasses

Flock, MySQL, PostgreSQL

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(connection, lock_name, options) ⇒ Base

Returns a new instance of Base.



24
25
26
27
28
29
30
31
32
33
 
# File 'lib/with_advisory_lock/base.rb', line 24

def initialize(connection, lock_name, options)
  options = {timeout_seconds: options} unless options.respond_to?(:fetch)
  options.assert_valid_keys :timeout_seconds, :shared, :transaction

  @connection = connection
  @lock_name = lock_name
  @timeout_seconds = options.fetch(:timeout_seconds, nil)
  @shared = options.fetch(:shared, false)
  @transaction = options.fetch(:transaction, false)
end

Instance Attribute Details

#connection ⇒ Object (readonly)

Returns the value of attribute connection.



22
23
24
 
# File 'lib/with_advisory_lock/base.rb', line 22

def connection
  @connection
end

#lock_name ⇒ Object (readonly)

Returns the value of attribute lock_name.



22
23
24
 
# File 'lib/with_advisory_lock/base.rb', line 22

def lock_name
  @lock_name
end

#shared ⇒ Object (readonly)

Returns the value of attribute shared.



22
23
24
 
# File 'lib/with_advisory_lock/base.rb', line 22

def shared
  @shared
end

#timeout_seconds ⇒ Object (readonly)

Returns the value of attribute timeout_seconds.



22
23
24
 
# File 'lib/with_advisory_lock/base.rb', line 22

def timeout_seconds
  @timeout_seconds
end

#transaction ⇒ Object (readonly)

Returns the value of attribute transaction.



22
23
24
 
# File 'lib/with_advisory_lock/base.rb', line 22

def transaction
  @transaction
end

Class Method Details

.lock_stack ⇒ Object 



43
44
45
46
 
# File 'lib/with_advisory_lock/base.rb', line 43

def self.lock_stack
  # access doesn't need to be synchronized as it is only accessed by the current thread.
  Thread.current[:with_advisory_lock_stack] ||= []
end

Instance Method Details

#already_locked? ⇒ Boolean

Returns:

  • (Boolean) 


49
50
51
 
# File 'lib/with_advisory_lock/base.rb', line 49

def already_locked?
  lock_stack.include? lock_stack_item
end

#lock_stack_item ⇒ Object 



39
40
41
 
# File 'lib/with_advisory_lock/base.rb', line 39

def lock_stack_item
  @lock_stack_item ||= LockStackItem.new(lock_str, shared)
end

#lock_str ⇒ Object 



35
36
37
 
# File 'lib/with_advisory_lock/base.rb', line 35

def lock_str
  @lock_str ||= "#{ENV['WITH_ADVISORY_LOCK_PREFIX'].to_s}#{lock_name.to_s}"
end

#stable_hashcode(input) ⇒ Object 



63
64
65
66
67
68
69
70
71
 
# File 'lib/with_advisory_lock/base.rb', line 63

def stable_hashcode(input)
  if input.is_a? Numeric
    input.to_i
  else
    # Ruby MRI's String#hash is randomly seeded as of Ruby 1.9 so
    # make sure we use a deterministic hash.
    Zlib.crc32(input.to_s)
  end
end

#unique_column_name ⇒ Object

Prevent AR from caching results improperly



100
101
102
 
# File 'lib/with_advisory_lock/base.rb', line 100

def unique_column_name
  "t#{SecureRandom.hex}"
end

#with_advisory_lock_if_needed(&block) ⇒ Object 



53
54
55
56
57
58
59
60
61
 
# File 'lib/with_advisory_lock/base.rb', line 53

def with_advisory_lock_if_needed(&block)
  if already_locked?
    Result.new(true, yield)
  elsif timeout_seconds == 0
    yield_with_lock(&block)
  else
    yield_with_lock_and_timeout(&block)
  end
end

#yield_with_lock ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/with_advisory_lock/base.rb', line 84

def yield_with_lock
  if try_lock
    begin
      lock_stack.push(lock_stack_item)
      result = block_given? ? yield : nil
      Result.new(true, result)
    ensure
      lock_stack.pop
      release_lock
    end
  else
    FAILED_TO_LOCK
  end
end

#yield_with_lock_and_timeout(&block) ⇒ Object 



73
74
75
76
77
78
79
80
81
82
 
# File 'lib/with_advisory_lock/base.rb', line 73

def yield_with_lock_and_timeout(&block)
  give_up_at = Time.now + @timeout_seconds if @timeout_seconds
  while @timeout_seconds.nil? || Time.now < give_up_at do
    r = yield_with_lock(&block)
    return r if r.lock_was_acquired?
    # Randomizing sleep time may help reduce contention.
    sleep(rand(0.05..0.15))
  end
  FAILED_TO_LOCK
end