github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-05-31 08:37:16 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

make sure we only do the work once

Browse Source
This commit is contained in:
Régis Hanol
2013-06-15 11:52:40 +02:00
parent f94e4ffdcb
commit 8a98310cf9
6 changed files with 22 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
app/models/upload.rb
View File

@ -13,17 +13,18 @@ class Upload < ActiveRecord::Base
validates_presence_of :original_filename
def self.create_for(user_id, file)
# retrieve image info
image_info = FastImage.new(file.tempfile, raise_on_failure: true)
# compute image aspect ratio
width, height = ImageSizer.resize(*image_info.size)
# compute the sha
sha = Digest::SHA1.file(file.tempfile).hexdigest
# check if the file has already been uploaded
upload = Upload.where(sha: sha).first
# otherwise, create it
if upload.blank?
# retrieve image info
image_info = FastImage.new(file.tempfile, raise_on_failure: true)
# compute image aspect ratio
width, height = ImageSizer.resize(*image_info.size)
# create a db record (so we can use the id)
upload = Upload.create!({
user_id: user_id,
original_filename: file.original_filename,
@ -33,23 +34,20 @@ class Upload < ActiveRecord::Base
height: height,
url: ""
})
# make sure we're at the beginning of the file (FastImage is moving the pointer)
file.rewind
# store the file and update its url
upload.url = Upload.store_file(file, image_info, upload.id)
upload.url = Upload.store_file(file, sha, image_info, upload.id)
# save the url
upload.save
end
# return the uploaded file
upload
end
def self.store_file(file, image_info, upload_id)
return S3.store_file(file, image_info, upload_id) if SiteSetting.enable_s3_uploads?
return LocalStore.store_file(file, image_info, upload_id)
def self.store_file(file, sha, image_info, upload_id)
return S3.store_file(file, sha, image_info, upload_id) if SiteSetting.enable_s3_uploads?
return LocalStore.store_file(file, sha, image_info, upload_id)
end
end