5819ca1a24
One should really use a separate table instead of using polymorphic associations. See https://gitlab.com/gitlab-org/gitlab-ce/merge_requests/11168 for more information.
63 lines
1.3 KiB
Ruby
63 lines
1.3 KiB
Ruby
class Upload < ActiveRecord::Base
|
|
# Upper limit for foreground checksum processing
|
|
CHECKSUM_THRESHOLD = 100.megabytes
|
|
|
|
belongs_to :model, polymorphic: true # rubocop:disable Cop/PolymorphicAssociations
|
|
|
|
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
|