From b93d09dbc59b3b85e7208cd57c92be9d86fd51df Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Tue, 24 Sep 2013 11:28:28 -0700
Subject: push preloaded test up to the factory method so we can eliminate
 conditionals from the individual preloaded classes

---
 .../lib/active_record/associations/preloader.rb       | 19 +++++++++++++++++--
 .../associations/preloader/association.rb             |  4 +---
 .../associations/preloader/through_association.rb     |  6 +++---
 3 files changed, 21 insertions(+), 8 deletions(-)

(limited to 'activerecord/lib/active_record')

diff --git a/activerecord/lib/active_record/associations/preloader.rb b/activerecord/lib/active_record/associations/preloader.rb
index b99ae35809..8a975468f2 100644
--- a/activerecord/lib/active_record/associations/preloader.rb
+++ b/activerecord/lib/active_record/associations/preloader.rb
@@ -133,7 +133,7 @@ module ActiveRecord
       def preloaders_for_one(association, records, scope)
         grouped_records(association, records).flat_map do |reflection, klasses|
           klasses.map do |rhs_klass, rs|
-            loader = preloader_for(reflection).new(rhs_klass, rs, reflection, scope)
+            loader = preloader_for(reflection, rs).new(rhs_klass, rs, reflection, scope)
             loader.run self
             loader
           end
@@ -173,7 +173,22 @@ module ActiveRecord
         end
       end
 
-      def preloader_for(reflection)
+      class NullPreloader
+        attr_reader :owners
+
+        def initialize(klass, owners, reflection, preload_scope)
+          @owners = owners
+        end
+
+        def run(preloader); end
+        def loaded?; false; end
+      end
+
+      def preloader_for(reflection, owners)
+        if owners.first.association(reflection.name).loaded?
+          return NullPreloader
+        end
+
         case reflection.macro
         when :has_many
           reflection.options[:through] ? HasManyThrough : HasMany
diff --git a/activerecord/lib/active_record/associations/preloader/association.rb b/activerecord/lib/active_record/associations/preloader/association.rb
index 9607afeea2..cdf4a27341 100644
--- a/activerecord/lib/active_record/associations/preloader/association.rb
+++ b/activerecord/lib/active_record/associations/preloader/association.rb
@@ -21,9 +21,7 @@ module ActiveRecord
         end
 
         def run(preloader)
-          unless owners.first.association(reflection.name).loaded?
-            preload(preloader)
-          end
+          preload(preloader)
         end
 
         def preload(preloader)
diff --git a/activerecord/lib/active_record/associations/preloader/through_association.rb b/activerecord/lib/active_record/associations/preloader/through_association.rb
index 16a48c1db2..0c0501fd58 100644
--- a/activerecord/lib/active_record/associations/preloader/through_association.rb
+++ b/activerecord/lib/active_record/associations/preloader/through_association.rb
@@ -49,13 +49,13 @@ module ActiveRecord
           @associated_records_by_owner = through_records.each_with_object({}) { |(lhs,center),records_by_owner|
             pl_to_middle = center.group_by { |record| middle_to_pl[record] }
 
-            records_by_owner[lhs] = pl_to_middle.flat_map do |preloader, middles|
+            records_by_owner[lhs] = pl_to_middle.flat_map do |pl, middles|
               rhs_records = middles.flat_map { |r|
                 r.send(source_reflection.name)
               }.compact
 
-              if preloader && preloader.loaded?
-                loaded_records = preloader.preloaded_records
+              if pl && pl.loaded?
+                loaded_records = pl.preloaded_records
                 i = 0
                 record_index = loaded_records.each_with_object({}) { |r,indexes|
                   indexes[r] = i
-- 
cgit v1.2.3