Added competability with HABTM

tinchi · Sep 5, 2014 · 60ccb55 · 60ccb55
1 parent a07216c
commit 60ccb55
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/mongoid/magic_counter_cache.rb b/lib/mongoid/magic_counter_cache.rb
@@ -79,8 +79,11 @@ def counter_cache(*args, &block)
             end
           else
             relation = doc.send(name)
-            if relation && relation.class.fields.keys.include?(counter_name)
-              increment_association(relation, counter_name.to_sym, inc)
+            relation = [relation] unless relation.respond_to?(:each)
+            relation.each do |entry|
+              if entry && entry.class.fields.keys.include?(counter_name)
+                increment_association(entry, counter_name.to_sym, inc)
+              end
             end
           end
         end