Contents

# -*- encoding : utf-8 -*-
module SecondLevelCache
  module ActiveRecord
    module Associations
      class Preloader
        module Association
          extend ActiveSupport::Concern

          included do
            class_eval do
              alias_method_chain :records_for, :second_level_cache
            end
          end

          def records_for_with_second_level_cache(ids)
            return records_for_without_second_level_cache(ids) unless klass.second_level_cache_enabled?

            map_cache_keys = ids.map{|id| klass.second_level_cache_key(id)}
            records_from_cache = ::SecondLevelCache.cache_store.read_multi(*map_cache_keys)
            # NOTICE 
            # Rails.cache.read_multi return hash that has keys only hitted.
            # eg. Rails.cache.read_multi(1,2,3) => {2 => hit_value, 3 => hit_value} 
            hitted_ids = records_from_cache.map{|key, _| key.split("/")[2].to_i}
            missed_ids = ids.map{|x| x.to_i} - hitted_ids

            ::SecondLevelCache::Config.logger.info "missed ids -> #{missed_ids.inspect} | hitted ids -> #{hitted_ids.inspect}"

            if missed_ids.empty?
              RecordMarshal.load_multi(records_from_cache.values)
            else
              records_from_db = records_for_without_second_level_cache(missed_ids)
              records_from_db.map{|record| write_cache(record); record} + RecordMarshal.load_multi(records_from_cache.values)
            end
          end

          private

          def write_cache(record)
            record.write_second_level_cache
          end
        end
      end
    end 
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
second_level_cache-2.1.0.rc2	lib/second_level_cache/active_record/preloader.rb
second_level_cache-2.1.0.rc1	lib/second_level_cache/active_record/preloader.rb