debian-mirror-gitlab/app/services/protected_branches/cache_service.rb

93 lines
2.9 KiB
Ruby
Raw Normal View History

2022-08-27 11:52:29 +05:30
# frozen_string_literal: true
module ProtectedBranches
class CacheService < ProtectedBranches::BaseService
CACHE_ROOT_KEY = 'cache:gitlab:protected_branch'
TTL_UNSET = -1
CACHE_EXPIRE_IN = 1.day
CACHE_LIMIT = 1000
2023-01-13 00:05:48 +05:30
def fetch(ref_name, dry_run: false, &block)
2022-08-27 11:52:29 +05:30
record = OpenSSL::Digest::SHA256.hexdigest(ref_name)
2023-01-13 00:05:48 +05:30
with_redis do |redis|
2022-08-27 11:52:29 +05:30
cached_result = redis.hget(redis_key, record)
2023-01-13 00:05:48 +05:30
if cached_result.nil?
metrics.increment_cache_miss
else
metrics.increment_cache_hit
decoded_result = Gitlab::Redis::Boolean.decode(cached_result)
end
2022-08-27 11:52:29 +05:30
# If we're dry-running, don't break because we need to check against
# the real value to ensure the cache is working properly.
# If the result is nil we'll need to run the block, so don't break yet.
break decoded_result unless dry_run || decoded_result.nil?
2023-01-13 00:05:48 +05:30
calculated_value = metrics.observe_cache_generation(&block)
2022-08-27 11:52:29 +05:30
check_and_log_discrepancy(decoded_result, calculated_value, ref_name) if dry_run
redis.hset(redis_key, record, Gitlab::Redis::Boolean.encode(calculated_value))
# We don't want to extend cache expiration time
if redis.ttl(redis_key) == TTL_UNSET
redis.expire(redis_key, CACHE_EXPIRE_IN)
end
# If the cache record has too many elements, then something went wrong and
# it's better to drop the cache key.
if redis.hlen(redis_key) > CACHE_LIMIT
redis.unlink(redis_key)
end
calculated_value
end
end
def refresh
2023-01-13 00:05:48 +05:30
with_redis { |redis| redis.unlink(redis_key) }
2022-08-27 11:52:29 +05:30
end
private
2023-01-13 00:05:48 +05:30
def with_redis(&block)
Gitlab::Redis::Cache.with(&block) # rubocop:disable CodeReuse/ActiveRecord
end
2022-08-27 11:52:29 +05:30
def check_and_log_discrepancy(cached_value, real_value, ref_name)
return if cached_value.nil?
return if cached_value == real_value
encoded_ref_name = Gitlab::EncodingHelper.encode_utf8_with_replacement_character(ref_name)
log_error(
'class' => self.class.name,
'message' => "Cache mismatch '#{encoded_ref_name}': cached value: #{cached_value}, real value: #{real_value}",
2023-03-04 22:38:38 +05:30
'record_class' => project_or_group.class.name,
'record_id' => project_or_group.id,
'record_path' => project_or_group.full_path
2022-08-27 11:52:29 +05:30
)
end
def redis_key
2023-03-04 22:38:38 +05:30
@redis_key ||= if Feature.enabled?(:group_protected_branches)
[CACHE_ROOT_KEY, project_or_group.class.name, project_or_group.id].join(':')
else
[CACHE_ROOT_KEY, project_or_group.id].join(':')
end
2022-08-27 11:52:29 +05:30
end
2023-01-13 00:05:48 +05:30
def metrics
@metrics ||= Gitlab::Cache::Metrics.new(
caller_id: Gitlab::ApplicationContext.current_context_attribute(:caller_id),
cache_identifier: "#{self.class}#fetch",
feature_category: :source_code_management,
backing_resource: :cpu
)
end
2022-08-27 11:52:29 +05:30
end
end