1 files changed, 13 insertions, 0 deletions
diff --git a/lib/gitlab/repository_archive_rate_limiter.rb b/lib/gitlab/repository_archive_rate_limiter.rb
new file mode 100644
index 00000000000..31a3dc34bf6
--- /dev/null
+++ b/lib/gitlab/repository_archive_rate_limiter.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module RepositoryArchiveRateLimiter
+    def check_archive_rate_limit!(current_user, project, &block)
+      return unless Feature.enabled?(:archive_rate_limit)
+
+      threshold = current_user ? nil : 100
+
+      check_rate_limit!(:project_repositories_archive, scope: [project, current_user], threshold: threshold, &block)
+    end
+  end
+end