fix count method signature for active_record5

- fix also count method signature for active_record5
flyerhzm · Jan 18, 2024 · c673436 · c673436
1 parent f063cad
commit c673436
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/bullet/active_record5.rb b/lib/bullet/active_record5.rb
@@ -273,7 +273,7 @@ def count_records
 
       ::ActiveRecord::Associations::CollectionProxy.prepend(
         Module.new do
-          def count
+          def count(column_name = nil)
             if Bullet.start? && !proxy_association.is_a?(::ActiveRecord::Associations::ThroughAssociation)
               Bullet::Detector::CounterCache.add_counter_cache(
                 proxy_association.owner,
@@ -284,7 +284,7 @@ def count
                 proxy_association.reflection.name
               )
             end
-            super
+            super(column_name)
           end
         end
       )