demarches-normaliennes/app/jobs/virus_scanner_job.rb

36 lines
1 KiB
Ruby
Raw Normal View History

class VirusScannerJob < ApplicationJob
# If by the time the job runs the blob has been deleted, ignore the error
discard_on ActiveRecord::RecordNotFound
# If the file is deleted during the scan, ignore the error
discard_on ActiveStorage::FileNotFoundError
# If for some reason the file appears invalid, retry for a while
retry_on(ActiveStorage::IntegrityError, attempts: 5, wait: 5.seconds) do |job, _error|
blob = job.arguments.first
metadata = {
virus_scan_result: ActiveStorage::VirusScanner::INTEGRITY_ERROR,
scanned_at: Time.zone.now
}
2021-04-06 18:00:54 +02:00
merge_and_update_metadata(blob, metadata)
end
def perform(blob)
if blob.virus_scanner.done? then return end
metadata = extract_metadata_via_virus_scanner(blob)
2021-04-06 18:00:54 +02:00
VirusScannerJob.merge_and_update_metadata(blob, metadata)
end
def extract_metadata_via_virus_scanner(blob)
ActiveStorage::VirusScanner.new(blob).metadata
end
2021-04-06 18:00:54 +02:00
private
def self.merge_and_update_metadata(blob, metadata)
blob.update!(metadata: blob.metadata.merge(metadata))
end
end