55 lines
1.1 KiB
Ruby
55 lines
1.1 KiB
Ruby
# frozen_string_literal: true
|
|
#
|
|
module Gitlab
|
|
module Diff
|
|
class StatsCache
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
EXPIRATION = 1.week
|
|
# The DiffStats#as_json representation is tied to the Gitaly protobuf version
|
|
VERSION = Gem.loaded_specs['gitaly'].version.to_s
|
|
|
|
def initialize(cachable_key:)
|
|
@cachable_key = cachable_key
|
|
end
|
|
|
|
def read
|
|
strong_memoize(:cached_values) do
|
|
content = cache.fetch(key)
|
|
|
|
next unless content
|
|
|
|
stats = content.map { |stat| Gitaly::DiffStats.new(stat) }
|
|
|
|
Gitlab::Git::DiffStatsCollection.new(stats)
|
|
end
|
|
end
|
|
|
|
def write_if_empty(stats)
|
|
return if cache.exist?(key)
|
|
return unless stats
|
|
|
|
cache.write(key, stats.map(&:to_h).as_json, expires_in: EXPIRATION)
|
|
clear_memoization(:cached_values)
|
|
end
|
|
|
|
def clear
|
|
cache.delete(key)
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :cachable_key
|
|
|
|
def cache
|
|
Rails.cache
|
|
end
|
|
|
|
def key
|
|
strong_memoize(:redis_key) do
|
|
['diff_stats', cachable_key, VERSION].join(":")
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|