ditkrg/active_model_serializers
4
0
Fork 0
mirror of https://github.com/ditkrg/active_model_serializers.git synced 2026-01-22 22:06:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #176 from hmsimple/id-methods

Browse Source 
Use pluck(:id) or _id methods instead of associated objects
This commit is contained in:
Steve Klabnik 2012-12-20 11:55:57 -08:00
commit 89660d1c5d
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
lib/active_model/serializer/associations.rb
View File

@ -109,9 +109,12 @@ module ActiveModel
def serialize_ids
# Use pluck or select_columns if available
# return collection.ids if collection.respond_to?(:ids)
associated_object.map do |item|
item.read_attribute_for_serialization(:id)
if !option(:include) && associated_object.respond_to?(:pluck)
associated_object.pluck(:id)
else
associated_object.map do |item|
item.read_attribute_for_serialization(:id)
end
end
end
end
@ -161,15 +164,19 @@ module ActiveModel
end
def serialize_ids
object = associated_object
if object && polymorphic?
{
:type => polymorphic_key,
:id => object.read_attribute_for_serialization(:id)
}
elsif object
object.read_attribute_for_serialization(:id)
if polymorphic?
if associated_object
{
:type => polymorphic_key,
:id => associated_object.read_attribute_for_serialization(:id)
}
else
nil
end
elsif source_serializer.object.respond_to?("#{name}_id")
source_serializer.object.send("#{name}_id")
elsif associated_object
associated_object.read_attribute_for_serialization(:id)
else
nil
end