debian-mirror-gitlab/lib/gitlab/repository_cache_adapter.rb

268 lines
9.2 KiB
Ruby
Raw Normal View History

2018-12-13 13:39:08 +05:30
# frozen_string_literal: true
2018-03-27 19:54:05 +05:30
module Gitlab
module RepositoryCacheAdapter
extend ActiveSupport::Concern
2018-12-05 23:21:45 +05:30
include Gitlab::Utils::StrongMemoize
2018-03-27 19:54:05 +05:30
class_methods do
2018-12-05 23:21:45 +05:30
# Caches and strongly memoizes the method.
2018-03-27 19:54:05 +05:30
#
# This only works for methods that do not take any arguments.
2018-12-05 23:21:45 +05:30
#
# name - The name of the method to be cached.
# fallback - A value to fall back to if the repository does not exist, or
# in case of a Git error. Defaults to nil.
def cache_method(name, fallback: nil)
uncached_name = alias_uncached_method(name)
define_method(name) do
cache_method_output(name, fallback: fallback) do
__send__(uncached_name) # rubocop:disable GitlabSecurity/PublicSend
end
end
end
2018-03-27 19:54:05 +05:30
2019-12-04 20:38:33 +05:30
# Caches and strongly memoizes the method as a Redis Set.
#
# This only works for methods that do not take any arguments. The method
# should return an Array of Strings to be cached.
#
# In addition to overriding the named method, a "name_include?" method is
# defined. This uses the "SISMEMBER" query to efficiently check membership
# without needing to load the entire set into memory.
#
# name - The name of the method to be cached.
# fallback - A value to fall back to if the repository does not exist, or
# in case of a Git error. Defaults to nil.
#
# It is not safe to use this method prior to the release of 12.3, since
# 12.2 does not correctly invalidate the redis set cache value. A mixed
# code environment containing both 12.2 and 12.3 nodes breaks, while a
# mixed code environment containing both 12.3 and 12.4 nodes will work.
def cache_method_as_redis_set(name, fallback: nil)
uncached_name = alias_uncached_method(name)
define_method(name) do
cache_method_output_as_redis_set(name, fallback: fallback) do
__send__(uncached_name) # rubocop:disable GitlabSecurity/PublicSend
end
end
# Attempt to determine whether a value is in the set of values being
# cached, by performing a redis SISMEMBERS query if appropriate.
#
# If the full list is already in-memory, we're better using it directly.
#
# If the cache is not yet populated, querying it directly will give the
# wrong answer. We handle that by querying the full list - which fills
# the cache - and using it directly to answer the question.
define_method("#{name}_include?") do |value|
2020-10-24 23:57:45 +05:30
ivar = "@#{name}_include"
memoized = instance_variable_get(ivar) || {}
2021-04-29 21:17:54 +05:30
lookup = proc { __send__(name).include?(value) } # rubocop:disable GitlabSecurity/PublicSend
2020-10-24 23:57:45 +05:30
next memoized[value] if memoized.key?(value)
memoized[value] =
2021-04-29 21:17:54 +05:30
if strong_memoized?(name)
lookup.call
2020-10-24 23:57:45 +05:30
else
2021-04-29 21:17:54 +05:30
result, exists = redis_set_cache.try_include?(name, value)
exists ? result : lookup.call
2020-10-24 23:57:45 +05:30
end
2019-12-04 20:38:33 +05:30
2020-10-24 23:57:45 +05:30
instance_variable_set(ivar, memoized)[value]
2019-12-04 20:38:33 +05:30
end
end
2018-12-05 23:21:45 +05:30
# Caches truthy values from the method. All values are strongly memoized,
# and cached in RequestStore.
#
# Currently only used to cache `exists?` since stale false values are
# particularly troublesome. This can occur, for example, when an NFS mount
# is temporarily down.
#
# This only works for methods that do not take any arguments.
#
# name - The name of the method to be cached.
def cache_method_asymmetrically(name)
uncached_name = alias_uncached_method(name)
2018-03-27 19:54:05 +05:30
define_method(name) do
2018-12-05 23:21:45 +05:30
cache_method_output_asymmetrically(name) do
__send__(uncached_name) # rubocop:disable GitlabSecurity/PublicSend
2018-03-27 19:54:05 +05:30
end
end
end
2018-12-05 23:21:45 +05:30
# Strongly memoizes the method.
#
# This only works for methods that do not take any arguments.
#
# name - The name of the method to be memoized.
# fallback - A value to fall back to if the repository does not exist, or
# in case of a Git error. Defaults to nil. The fallback value
# is not memoized.
def memoize_method(name, fallback: nil)
uncached_name = alias_uncached_method(name)
define_method(name) do
memoize_method_output(name, fallback: fallback) do
__send__(uncached_name) # rubocop:disable GitlabSecurity/PublicSend
end
end
end
# Prepends "_uncached_" to the target method name
#
# Returns the uncached method name
def alias_uncached_method(name)
uncached_name = :"_uncached_#{name}"
alias_method(uncached_name, name)
uncached_name
end
end
# RequestStore-backed RepositoryCache to be used. Should be overridden by
# the including class
def request_store_cache
raise NotImplementedError
2018-03-27 19:54:05 +05:30
end
# RepositoryCache to be used. Should be overridden by the including class
def cache
raise NotImplementedError
end
2019-12-04 20:38:33 +05:30
# RepositorySetCache to be used. Should be overridden by the including class
def redis_set_cache
raise NotImplementedError
end
2020-03-13 15:44:24 +05:30
# RepositoryHashCache to be used. Should be overridden by the including class
def redis_hash_cache
raise NotImplementedError
end
2018-11-08 19:23:39 +05:30
# List of cached methods. Should be overridden by the including class
def cached_methods
raise NotImplementedError
end
2018-12-05 23:21:45 +05:30
# Caches and strongly memoizes the supplied block.
2018-03-27 19:54:05 +05:30
#
2018-12-05 23:21:45 +05:30
# name - The name of the method to be cached.
# fallback - A value to fall back to if the repository does not exist, or
# in case of a Git error. Defaults to nil.
def cache_method_output(name, fallback: nil, &block)
memoize_method_output(name, fallback: fallback) do
cache.fetch(name, &block)
end
end
2019-12-04 20:38:33 +05:30
# Caches and strongly memoizes the supplied block as a Redis Set. The result
# will be provided as a sorted array.
#
# name - The name of the method to be cached.
# fallback - A value to fall back to if the repository does not exist, or
# in case of a Git error. Defaults to nil.
def cache_method_output_as_redis_set(name, fallback: nil, &block)
memoize_method_output(name, fallback: fallback) do
redis_set_cache.fetch(name, &block).sort
end
end
2018-12-05 23:21:45 +05:30
# Caches truthy values from the supplied block. All values are strongly
# memoized, and cached in RequestStore.
2018-03-27 19:54:05 +05:30
#
2018-12-05 23:21:45 +05:30
# Currently only used to cache `exists?` since stale false values are
# particularly troublesome. This can occur, for example, when an NFS mount
# is temporarily down.
2018-03-27 19:54:05 +05:30
#
2018-12-05 23:21:45 +05:30
# name - The name of the method to be cached.
def cache_method_output_asymmetrically(name, &block)
memoize_method_output(name) do
request_store_cache.fetch(name) do
cache.fetch_without_caching_false(name, &block)
2018-03-27 19:54:05 +05:30
end
end
end
2018-12-05 23:21:45 +05:30
# Strongly memoizes the supplied block.
#
# name - The name of the method to be memoized.
# fallback - A value to fall back to if the repository does not exist, or
# in case of a Git error. Defaults to nil. The fallback value is
# not memoized.
def memoize_method_output(name, fallback: nil, &block)
no_repository_fallback(name, fallback: fallback) do
strong_memoize(memoizable_name(name), &block)
end
end
# Returns the fallback value if the repository does not exist
def no_repository_fallback(name, fallback: nil, &block)
# Avoid unnecessary gRPC invocations
return fallback if fallback && fallback_early?(name)
yield
rescue Gitlab::Git::Repository::NoRepository
# Even if the `#exists?` check in `fallback_early?` passes, these errors
# might still occur (for example because of a non-existing HEAD). We
# want to gracefully handle this and not memoize anything.
fallback
end
2018-03-27 19:54:05 +05:30
# Expires the caches of a specific set of methods
def expire_method_caches(methods)
2018-12-05 23:21:45 +05:30
methods.each do |name|
unless cached_methods.include?(name.to_sym)
2020-11-24 15:15:51 +05:30
Gitlab::AppLogger.error "Requested to expire non-existent method '#{name}' for Repository"
2018-11-08 19:23:39 +05:30
next
end
2018-12-05 23:21:45 +05:30
cache.expire(name)
2018-03-27 19:54:05 +05:30
2018-12-05 23:21:45 +05:30
clear_memoization(memoizable_name(name))
2018-03-27 19:54:05 +05:30
end
2018-12-05 23:21:45 +05:30
2019-12-04 20:38:33 +05:30
expire_redis_set_method_caches(methods)
2020-03-13 15:44:24 +05:30
expire_redis_hash_method_caches(methods)
2018-12-05 23:21:45 +05:30
expire_request_store_method_caches(methods)
2018-03-27 19:54:05 +05:30
end
private
2018-12-05 23:21:45 +05:30
def memoizable_name(name)
"#{name.to_s.tr('?!', '')}"
end
def expire_request_store_method_caches(methods)
methods.each do |name|
request_store_cache.expire(name)
end
end
2019-12-04 20:38:33 +05:30
def expire_redis_set_method_caches(methods)
2020-04-08 14:13:33 +05:30
redis_set_cache.expire(*methods)
2019-12-04 20:38:33 +05:30
end
2020-03-13 15:44:24 +05:30
def expire_redis_hash_method_caches(methods)
2020-10-24 23:57:45 +05:30
redis_hash_cache.delete(*methods)
2020-03-13 15:44:24 +05:30
end
2018-12-05 23:21:45 +05:30
# All cached repository methods depend on the existence of a Git repository,
# so if the repository doesn't exist, we already know not to call it.
def fallback_early?(method_name)
# Avoid infinite loop
return false if method_name == :exists?
!exists?
2018-03-27 19:54:05 +05:30
end
end
end