lib/bullet/ext/object.rb in bullet-4.14.7 vs lib/bullet/ext/object.rb in bullet-4.14.8

@@ -2,10 +2,12 @@
   def bullet_key
     "#{self.class}:#{self.primary_key_value}"
   end
 
   def primary_key_value
-    if self.class.respond_to?(:primary_key) && self.class.primary_key
+    if self.class.respond_to?(:primary_keys) && self.class.primary_keys
+      self.class.primary_keys.map { |primary_key| self.send primary_key }.join(',')
+    elsif self.class.respond_to?(:primary_key) && self.class.primary_key
       self.send self.class.primary_key
     else
       self.id
     end
   end