Add latest changes from gitlab-org/gitlab@15-0-stable-eev15.0.0-rc42

1 files changed, 33 insertions, 0 deletions

diff --git a/lib/gitlab/instrumentation/rate_limiting_gates.rb b/lib/gitlab/instrumentation/rate_limiting_gates.rb
new file mode 100644
index 00000000000..960b6995030
--- /dev/null
+++ b/lib/gitlab/instrumentation/rate_limiting_gates.rb
@@ -0,0 +1,33 @@
+# 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