gitlab-org--gitlab-foss/lib/gitlab/diff/stats_cache.rb

# 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