64 lines
1.3 KiB
Ruby
64 lines
1.3 KiB
Ruby
|
class Upload < ActiveRecord::Base
|
||
|
|
# Upper limit for foreground checksum processing
|
||
|
|
CHECKSUM_THRESHOLD = 100.megabytes
|
||
|
|
|
||
|
|
belongs_to :model, polymorphic: true
|
||
|
|
|
||
|
|
validates :size, presence: true
|
||
|
|
validates :path, presence: true
|
||
|
|
validates :model, presence: true
|
||
|
|
validates :uploader, presence: true
|
||
|
|
|
||
|
|
before_save :calculate_checksum, if: :foreground_checksum?
|
||
|
|
after_commit :schedule_checksum, unless: :foreground_checksum?
|
||
|
|
|
||
|
|
def self.remove_path(path)
|
||
|
|
where(path: path).destroy_all
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.record(uploader)
|
||
|
|
remove_path(uploader.relative_path)
|
||
|
|
|
||
|
|
create(
|
||
|
|
size: uploader.file.size,
|
||
|
|
path: uploader.relative_path,
|
||
|
|
model: uploader.model,
|
||
|
|
uploader: uploader.class.to_s
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
def absolute_path
|
||
|
|
return path unless relative_path?
|
||
|
|
|
||
|
|
uploader_class.absolute_path(self)
|
||
|
|
end
|
||
|
|
|
||
|
|
def calculate_checksum
|
||
|
|
return unless exist?
|
||
|
|
|
||
|
|
self.checksum = Digest::SHA256.file(absolute_path).hexdigest
|
||
|
|
end
|
||
|
|
|
||
|
|
def exist?
|
||
|
|
File.exist?(absolute_path)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def foreground_checksum?
|
||
|
|
size <= CHECKSUM_THRESHOLD
|
||
|
|
end
|
||
|
|
|
||
|
|
def schedule_checksum
|
||
|
|
UploadChecksumWorker.perform_async(id)
|
||
|
|
end
|
||
|
|
|
||
|
|
def relative_path?
|
||
|
|
!path.start_with?('/')
|
||
|
|
end
|
||
|
|
|
||
|
|
def uploader_class
|
||
|
|
Object.const_get(uploader)
|
||
|
|
end
|
||
|
|
end
|