call association also on through reflection

flyerhzm · Nov 17, 2023 · c35e9fb · c35e9fb
1 parent e147136
commit c35e9fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/lib/bullet/active_record4.rb b/lib/bullet/active_record4.rb
@@ -108,7 +108,7 @@ def construct_association(record, join, row)
 
           if Bullet.start?
             associations = [node.reflection.name]
-            if node.reflection.through_reflection?
+            if node.reflection.chain?
               associations << node.reflection.through_reflection.name
             end
             associations.each do |association|

diff --git a/lib/bullet/active_record41.rb b/lib/bullet/active_record41.rb
@@ -111,7 +111,7 @@ def construct_model(record, node, row, model_cache, id, aliases)
 
           if Bullet.start?
             associations = [node.reflection.name]
-            if node.reflection.through_reflection?
+            if node.reflection.chain?
               associations << node.reflection.through_reflection.name
             end
             associations.each do |association|

diff --git a/lib/bullet/active_record42.rb b/lib/bullet/active_record42.rb
@@ -153,7 +153,7 @@ def construct_model(record, node, row, model_cache, id, aliases)
 
           if Bullet.start?
             associations = [node.reflection.name]
-            if node.reflection.through_reflection?
+            if node.reflection.chain?
               associations << node.reflection.through_reflection.name
             end
             associations.each do |association|