aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb
diff options
context:
space:
mode:
Diffstat (limited to 'activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb')
-rw-r--r--activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb21
1 files changed, 19 insertions, 2 deletions
diff --git a/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb b/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb
index 917a7fa5e1..663655213a 100644
--- a/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb
+++ b/activerecord/lib/active_record/associations/has_and_belongs_to_many_association.rb
@@ -15,6 +15,7 @@ module ActiveRecord
def create(attributes = {})
# Can't use Base.create because the foreign key may be a protected attribute.
+ ensure_owner_is_not_new
if attributes.is_a?(Array)
attributes.collect { |attr| create(attr) }
else
@@ -23,6 +24,18 @@ module ActiveRecord
record
end
end
+
+ def create!(attributes = {})
+ # Can't use Base.create! because the foreign key may be a protected attribute.
+ ensure_owner_is_not_new
+ if attributes.is_a?(Array)
+ attributes.collect { |attr| create(attr) }
+ else
+ record = build(attributes)
+ insert_record(record, true) unless @owner.new_record?
+ record
+ end
+ end
def find_first
load_target.first
@@ -75,9 +88,13 @@ module ActiveRecord
load_target.size
end
- def insert_record(record)
+ def insert_record(record, force=true)
if record.new_record?
- return false unless record.save
+ if force
+ record.save!
+ else
+ return false unless record.save
+ end
end
if @reflection.options[:insert_sql]