diff options
Diffstat (limited to 'activerecord/lib/active_record/associations')
3 files changed, 12 insertions, 4 deletions
diff --git a/activerecord/lib/active_record/associations/association_proxy.rb b/activerecord/lib/active_record/associations/association_proxy.rb index c62c042ab2..a4fc85b15e 100644 --- a/activerecord/lib/active_record/associations/association_proxy.rb +++ b/activerecord/lib/active_record/associations/association_proxy.rb @@ -74,6 +74,15 @@ module ActiveRecord @owner.send(:extract_options_from_args!, args) end + def set_belongs_to_association_for(record) + if @reflection.options[:as] + record["#{@reflection.options[:as]}_id"] = @owner.id unless @owner.new_record? + record["#{@reflection.options[:as]}_type"] = ActiveRecord::Base.send(:class_name_of_active_record_descendant, @owner.class).to_s + else + record[@reflection.primary_key_name] = @owner.id unless @owner.new_record? + end + end + def merge_options_from_reflection!(options) options.reverse_merge!( :group => @reflection.options[:group], diff --git a/activerecord/lib/active_record/associations/has_many_association.rb b/activerecord/lib/active_record/associations/has_many_association.rb index 6800f363dc..b861fde55d 100644 --- a/activerecord/lib/active_record/associations/has_many_association.rb +++ b/activerecord/lib/active_record/associations/has_many_association.rb @@ -13,7 +13,7 @@ module ActiveRecord else load_target record = @reflection.klass.new(attributes) - record[@reflection.primary_key_name] = @owner.id unless @owner.new_record? + set_belongs_to_association_for(record) @target << record record end @@ -140,7 +140,7 @@ module ActiveRecord end def insert_record(record) - record[@reflection.primary_key_name] = @owner.id + set_belongs_to_association_for(record) record.save end diff --git a/activerecord/lib/active_record/associations/has_one_association.rb b/activerecord/lib/active_record/associations/has_one_association.rb index 4dbc0f6441..d07323a869 100644 --- a/activerecord/lib/active_record/associations/has_one_association.rb +++ b/activerecord/lib/active_record/associations/has_one_association.rb @@ -42,8 +42,7 @@ module ActiveRecord @target = nil else raise_on_type_mismatch(obj) - - obj[@reflection.primary_key_name] = @owner.id unless @owner.new_record? + set_belongs_to_association_for(obj) @target = (AssociationProxy === obj ? obj.target : obj) end |