1 files changed, 27 insertions, 0 deletions
diff --git a/lib/banzai/request_store_reference_cache.rb b/lib/banzai/request_store_reference_cache.rb
new file mode 100644
index 00000000000..426131442a2
--- /dev/null
+++ b/lib/banzai/request_store_reference_cache.rb
@@ -0,0 +1,27 @@
+module Banzai
+  module RequestStoreReferenceCache
+    def cached_call(request_store_key, cache_key, path: [])
+      if RequestStore.active?
+        cache = RequestStore[request_store_key] ||= Hash.new do |hash, key|
+          hash[key] = Hash.new { |h, k| h[k] = {} }
+        end
+
+        cache = cache.dig(*path) if path.any?
+
+        get_or_set_cache(cache, cache_key) { yield }
+      else
+        yield
+      end
+    end
+
+    def get_or_set_cache(cache, key)
+      if cache.key?(key)
+        cache[key]
+      else
+        value = yield
+        cache[key] = value if key.present?
+        value
+      end
+    end
+  end
+end