mirror of
https://github.com/discourse/discourse.git
synced 2024-11-23 06:04:11 +08:00
30990006a9
This reduces chances of errors where consumers of strings mutate inputs and reduces memory usage of the app. Test suite passes now, but there may be some stuff left, so we will run a few sites on a branch prior to merging
73 lines
1.9 KiB
Ruby
73 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
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]
|
|
identifier = @params[:identifier]
|
|
filename = @params[:filename]
|
|
tmp_directory = @params[:tmp_directory]
|
|
|
|
# delete destination files
|
|
begin
|
|
File.delete(upload_path)
|
|
File.delete(tmp_upload_path)
|
|
rescue Errno::ENOENT
|
|
end
|
|
|
|
# merge all the chunks
|
|
File.open(tmp_upload_path, "a") do |file|
|
|
(1..@chunk).each do |chunk_number|
|
|
# path to chunk
|
|
chunk_path = BackupRestore::LocalBackupStore.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
|
|
begin
|
|
FileUtils.rm_rf(tmp_directory)
|
|
rescue Errno::ENOENT
|
|
end
|
|
end
|
|
|
|
end
|