1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00

Fix attaching with standard Rails associations.

Removes needless ivar caching (a Rails association handles that).

Inserts a reload and a nil assign, since the association proxy doesn't
seem to that it's been destroyed through `purge`.
This commit is contained in:
Kasper Timm Hansen 2017-07-23 21:41:16 +02:00
parent 212f925654
commit a4f36f957e
No known key found for this signature in database
GPG key ID: 191153215EDA53D8
3 changed files with 15 additions and 11 deletions

View file

@ -7,15 +7,15 @@ class ActiveStorage::Attached::Many < ActiveStorage::Attached
# You don't have to call this method to access the attachments' methods as
# they are all available at the model level.
def attachments
@attachments ||= record.public_send("#{name}_attachments")
record.public_send("#{name}_attachments")
end
# Associates one or several attachments with the current record, saving
# them to the database.
def attach(*attachables)
@attachments = attachments | Array(attachables).flatten.collect do |attachable|
record.public_send("#{name}_attachments=", attachments | Array(attachables).flat_map do |attachable|
ActiveStorage::Attachment.create!(record: record, name: name, blob: create_blob_from(attachable))
end
end)
end
# Checks the presence of attachments.
@ -34,7 +34,7 @@ class ActiveStorage::Attached::Many < ActiveStorage::Attached
def purge
if attached?
attachments.each(&:purge)
@attachments = nil
attachments.reload
end
end
@ -42,7 +42,6 @@ class ActiveStorage::Attached::Many < ActiveStorage::Attached
def purge_later
if attached?
attachments.each(&:purge_later)
@attachments = nil
end
end
end

View file

@ -7,13 +7,14 @@ class ActiveStorage::Attached::One < ActiveStorage::Attached
# You don't have to call this method to access the attachment's methods as
# they are all available at the model level.
def attachment
@attachment ||= record.public_send("#{name}_attachment")
record.public_send("#{name}_attachment")
end
# Associates a given attachment with the current record, saving it to the
# database.
def attach(attachable)
@attachment = ActiveStorage::Attachment.create!(record: record, name: name, blob: create_blob_from(attachable))
write_attachment \
ActiveStorage::Attachment.create!(record: record, name: name, blob: create_blob_from(attachable))
end
# Checks the presence of the attachment.
@ -32,7 +33,7 @@ class ActiveStorage::Attached::One < ActiveStorage::Attached
def purge
if attached?
attachment.purge
@attachment = nil
write_attachment nil
end
end
@ -40,7 +41,11 @@ class ActiveStorage::Attached::One < ActiveStorage::Attached
def purge_later
if attached?
attachment.purge_later
@attachment = nil
end
end
private
def write_attachment(attachment)
record.public_send("#{name}_attachment=", attachment)
end
end