active_model_serializers/lib/active_model_serializers/cached_serializer.rb

module ActiveModelSerializers
  class CachedSerializer
    def initialize(serializer)
      @cached_serializer = serializer
      @klass             = @cached_serializer.class
    end

    def cache_check(adapter_instance)
      if cached?
        @klass._cache.fetch(cache_key, @klass._cache_options) do
          yield
        end
      elsif fragment_cached?
        FragmentCache.new(adapter_instance, @cached_serializer, adapter_instance.instance_options).fetch
      else
        yield
      end
    end

    def cached?
      @klass.cache_enabled?
    end

    def fragment_cached?
      @klass.fragment_cache_enabled?
    end

    def cache_key
      return @cache_key if defined?(@cache_key)

      parts = []
      parts << object_cache_key
      parts << @klass._cache_digest unless @klass._cache_options && @klass._cache_options[:skip_digest]
      @cache_key = parts.join('/')
    end

    def object_cache_key
      object_time_safe = @cached_serializer.object.updated_at
      object_time_safe = object_time_safe.strftime('%Y%m%d%H%M%S%9N') if object_time_safe.respond_to?(:strftime)
      @klass._cache_key ? "#{@klass._cache_key}/#{@cached_serializer.object.id}-#{object_time_safe}" : @cached_serializer.object.cache_key
    end

    # find all cache_key for the collection_serializer
    # @param collection_serializer
    # @param include_tree
    # @return [Array] all cache_key of collection_serializer
    def self.object_cache_keys(serializers, include_tree)
      cache_keys = []

      serializers.each do |serializer|
        cache_keys << object_cache_key(serializer)

        serializer.associations(include_tree).each do |association|
          if association.serializer.respond_to?(:each)
            association.serializer.each do |sub_serializer|
              cache_keys << object_cache_key(sub_serializer)
            end
          else
            cache_keys << object_cache_key(association.serializer)
          end
        end
      end

      cache_keys.compact.uniq
    end

    # @return [String, nil] the cache_key of the serializer or nil
    def self.object_cache_key(serializer)
      return unless serializer.present? && serializer.object.present?

      cached_serializer = new(serializer)
      cached_serializer.cached? ? cached_serializer.cache_key : nil
    end
  end
end