43 lines
930 B
Ruby
43 lines
930 B
Ruby
|
module Gitlab
|
||
|
|
module TemporarilyAllow
|
||
|
|
TEMPORARILY_ALLOW_MUTEX = Mutex.new
|
||
|
|
|
||
|
|
def temporarily_allow(key)
|
||
|
|
temporarily_allow_add(key, 1)
|
||
|
|
yield
|
||
|
|
ensure
|
||
|
|
temporarily_allow_add(key, -1)
|
||
|
|
end
|
||
|
|
|
||
|
|
def temporarily_allowed?(key)
|
||
|
|
if RequestStore.active?
|
||
|
|
temporarily_allow_request_store[key] > 0
|
||
|
|
else
|
||
|
|
TEMPORARILY_ALLOW_MUTEX.synchronize do
|
||
|
|
temporarily_allow_ivar[key] > 0
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def temporarily_allow_ivar
|
||
|
|
@temporarily_allow ||= Hash.new(0)
|
||
|
|
end
|
||
|
|
|
||
|
|
def temporarily_allow_request_store
|
||
|
|
RequestStore[:temporarily_allow] ||= Hash.new(0)
|
||
|
|
end
|
||
|
|
|
||
|
|
def temporarily_allow_add(key, value)
|
||
|
|
if RequestStore.active?
|
||
|
|
temporarily_allow_request_store[key] += value
|
||
|
|
else
|
||
|
|
TEMPORARILY_ALLOW_MUTEX.synchronize do
|
||
|
|
temporarily_allow_ivar[key] += value
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|