use _ids method instead of pluck

rails-api · Dec 21, 2012 · be005c6 · be005c6
1 parent a74fc6c
commit be005c6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/active_model/serializer/associations.rb b/lib/active_model/serializer/associations.rb
@@ -109,8 +109,10 @@ def serializables
         def serialize_ids
           # Use pluck or select_columns if available
           # return collection.ids if collection.respond_to?(:ids)
-          if !option(:include) && associated_object.respond_to?(:pluck)
-            associated_object.pluck(:id)
+          ids_key = "#{key.to_s.singularize}_ids"
+
+          if !option(:include) && source_serializer.object.respond_to?(ids_key)
+            source_serializer.object.send(ids_key)
           else
             associated_object.map do |item|
               item.read_attribute_for_serialization(:id)