34 lines
556 B
Ruby
34 lines
556 B
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Instrumentation
|
|
class RateLimitingGates
|
|
GATES = :rate_limiting_gates
|
|
|
|
class << self
|
|
def track(key)
|
|
if ::Gitlab::SafeRequestStore.active?
|
|
gates_set << key
|
|
end
|
|
end
|
|
|
|
def gates
|
|
gates_set.to_a
|
|
end
|
|
|
|
def payload
|
|
{
|
|
GATES => gates
|
|
}
|
|
end
|
|
|
|
private
|
|
|
|
def gates_set
|
|
::Gitlab::SafeRequestStore[GATES] ||= Set.new
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|