mirror of
https://github.com/discourse/discourse.git
synced 2025-01-24 05:13:57 +08:00
142571bba0
* `rescue nil` is a really bad pattern to use in our code base. We should rescue errors that we expect the code to throw and not rescue everything because we're unsure of what errors the code would throw. This would reduce the amount of pain we face when debugging why something isn't working as expexted. I've been bitten countless of times by errors being swallowed as a result during debugging sessions.
66 lines
1.8 KiB
Ruby
66 lines
1.8 KiB
Ruby
class HandleChunkUpload
|
|
|
|
def initialize(chunk, params = {})
|
|
@chunk = chunk
|
|
@params = params
|
|
end
|
|
|
|
def self.check_chunk(chunk, params)
|
|
HandleChunkUpload.new(chunk, params).check_chunk
|
|
end
|
|
|
|
def self.upload_chunk(chunk, params)
|
|
HandleChunkUpload.new(chunk, params).upload_chunk
|
|
end
|
|
|
|
def self.merge_chunks(chunk, params)
|
|
HandleChunkUpload.new(chunk, params).merge_chunks
|
|
end
|
|
|
|
def check_chunk
|
|
# check whether the chunk has already been uploaded
|
|
has_chunk_been_uploaded = File.exists?(@chunk) && File.size(@chunk) == @params[:current_chunk_size]
|
|
# 200 = exists, 404 = not uploaded yet
|
|
status = has_chunk_been_uploaded ? 200 : 404
|
|
end
|
|
|
|
def upload_chunk
|
|
# path to chunk file
|
|
dir = File.dirname(@chunk)
|
|
# ensure directory exists
|
|
FileUtils.mkdir_p(dir) unless Dir.exists?(dir)
|
|
# save chunk to the directory
|
|
File.open(@chunk, "wb") { |f| f.write(@params[:file].tempfile.read) }
|
|
end
|
|
|
|
def merge_chunks
|
|
upload_path = @params[:upload_path]
|
|
tmp_upload_path = @params[:tmp_upload_path]
|
|
model = @params[:model]
|
|
identifier = @params[:identifier]
|
|
filename = @params[:filename]
|
|
tmp_directory = @params[:tmp_directory]
|
|
|
|
# delete destination files
|
|
File.delete(upload_path)
|
|
File.delete(tmp_upload_path)
|
|
|
|
# merge all the chunks
|
|
File.open(tmp_upload_path, "a") do |file|
|
|
(1..@chunk).each do |chunk_number|
|
|
# path to chunk
|
|
chunk_path = model.chunk_path(identifier, filename, chunk_number)
|
|
# add chunk to file
|
|
file << File.open(chunk_path).read
|
|
end
|
|
end
|
|
|
|
# rename tmp file to final file name
|
|
FileUtils.mv(tmp_upload_path, upload_path, force: true)
|
|
|
|
# remove tmp directory
|
|
FileUtils.rm_rf(tmp_directory)
|
|
end
|
|
|
|
end
|