1 files changed, 10 insertions, 1 deletions
diff --git a/lib/gitlab/repository_cache.rb b/lib/gitlab/repository_cache.rb
index dc8b2467f72..8de2c2fe772 100644
--- a/lib/gitlab/repository_cache.rb
+++ b/lib/gitlab/repository_cache.rb
@@ -5,7 +5,7 @@ module Gitlab
   class RepositoryCache
     attr_reader :repository, :namespace, :backend
 
-    def initialize(repository, extra_namespace: nil, backend: Rails.cache)
+    def initialize(repository, extra_namespace: nil, backend: self.class.store)
       @repository = repository
       @namespace = "#{repository.full_path}"
       @namespace += ":#{repository.project.id}" if repository.project
@@ -48,5 +48,14 @@ module Gitlab
 
       value
     end
+
+    def self.store
+      if Feature.enabled?(:use_primary_and_secondary_stores_for_repository_cache) ||
+          Feature.enabled?(:use_primary_store_as_default_for_repository_cache)
+        Gitlab::Redis::RepositoryCache.cache_store
+      else
+        Rails.cache
+      end
+    end
   end
 end