diff options
Diffstat (limited to 'lib/gitlab/instrumentation/rate_limiting_gates.rb')
-rw-r--r-- | lib/gitlab/instrumentation/rate_limiting_gates.rb | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/lib/gitlab/instrumentation/rate_limiting_gates.rb b/lib/gitlab/instrumentation/rate_limiting_gates.rb index 01b362f0cf4..960b6995030 100644 --- a/lib/gitlab/instrumentation/rate_limiting_gates.rb +++ b/lib/gitlab/instrumentation/rate_limiting_gates.rb @@ -5,11 +5,9 @@ module Gitlab class RateLimitingGates GATES = :rate_limiting_gates - InstrumentationStorage = ::Gitlab::Instrumentation::Storage - class << self def track(key) - if InstrumentationStorage.active? + if ::Gitlab::SafeRequestStore.active? gates_set << key end end @@ -27,7 +25,7 @@ module Gitlab private def gates_set - InstrumentationStorage[GATES] ||= Set.new + ::Gitlab::SafeRequestStore[GATES] ||= Set.new end end end |