1 files changed, 21 insertions, 0 deletions
diff --git a/lib/gitlab/utils/strong_memoize.rb b/lib/gitlab/utils/strong_memoize.rb
index eb44b7ddd95..2b3841b8f09 100644
--- a/lib/gitlab/utils/strong_memoize.rb
+++ b/lib/gitlab/utils/strong_memoize.rb
@@ -35,6 +35,27 @@ module Gitlab
         end
       end
 
+      # Works the same way as "strong_memoize" but takes
+      # a second argument - expire_in. This allows invalidate
+      # the data after specified number of seconds
+      def strong_memoize_with_expiration(name, expire_in)
+        key = ivar(name)
+        expiration_key = "#{key}_expired_at"
+
+        if instance_variable_defined?(expiration_key)
+          expire_at = instance_variable_get(expiration_key)
+          clear_memoization(name) if Time.current > expire_at
+        end
+
+        if instance_variable_defined?(key)
+          instance_variable_get(key)
+        else
+          value = instance_variable_set(key, yield)
+          instance_variable_set(expiration_key, Time.current + expire_in)
+          value
+        end
+      end
+
       def strong_memoize_with(name, *args)
         container = strong_memoize(name) { {} }