Contents

module Btspm
  module EagerLoadable

    def eager_load(collection, *associations)
      EagerLoader.load(collection, *associations)
    end

    class EagerLoader
      def initialize(coll, *assocs)
        @collection = coll
        @associations = assocs
      end

      def self.load(collection, *associations)
        preloader = new(collection, *associations)
        preloader.load
        collection
      end

      def load
        preloader = ActiveRecord::Associations::Preloader.new
        preloader.preload(collection, associations)
      end

      private

      def true_collection?
        !@collection.is_a?(ActiveRecord::Base)
      end

      def collection
        true_collection? ? @collection : [@collection]
      end

      attr_reader :associations
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
btspm-0.0.4	lib/btspm/eager_loadable.rb
btspm-0.0.3	lib/btspm/eager_loadable.rb
btspm-0.0.2	lib/btspm/eager_loadable.rb
btspm-0.0.1	lib/btspm/eager_loadable.rb