feat(ArchiveUploader.upload_with_chunking_wrapper): retry once on error
This commit is contained in:
parent
c72ba43c55
commit
9e8807d12a
2 changed files with 15 additions and 3 deletions
|
@ -48,11 +48,11 @@ class ArchiveUploader
|
|||
params = blob_default_params(filepath).merge(byte_size: File.size(filepath),
|
||||
checksum: Digest::SHA256.file(filepath).hexdigest)
|
||||
blob = ActiveStorage::Blob.create_before_direct_upload!(**params)
|
||||
if syscall_to_custom_uploader(blob)
|
||||
if retryable_syscall_to_custom_uploader(blob)
|
||||
return blob
|
||||
else
|
||||
blob.purge
|
||||
fail "custom archive attachment failed, should it be retried ?"
|
||||
fail "custom archive attachment failed twice, retry later"
|
||||
end
|
||||
end
|
||||
|
||||
|
@ -73,6 +73,18 @@ class ArchiveUploader
|
|||
@namespaced_object_key ||= "archives/#{Date.today.strftime("%Y-%m-%d")}/#{SecureRandom.uuid}"
|
||||
end
|
||||
|
||||
def retryable_syscall_to_custom_uploader(blob)
|
||||
limit_to_retry = 1
|
||||
begin
|
||||
syscall_to_custom_uploader(blob)
|
||||
rescue
|
||||
if limit_to_retry > 0
|
||||
limit_to_retry = limit_to_retry - 1
|
||||
retry
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
def syscall_to_custom_uploader(blob)
|
||||
system(ENV.fetch('ACTIVE_STORAGE_BIG_FILE_UPLOADER_WITH_ENCRYPTION_PATH').to_s, filepath, blob.key, exception: true)
|
||||
end
|
||||
|
|
|
@ -82,7 +82,7 @@ describe ProcedureArchiveService do
|
|||
it 'does not retry more than once' do
|
||||
expect(uploader).to receive(:syscall_to_custom_uploader).with(anything).twice.and_raise(StandardError, "BOOM")
|
||||
expect { uploader.send(:upload_with_chunking_wrapper) }
|
||||
.to raise_error(StandardError, "BOOM")
|
||||
.to raise_error(RuntimeError, "custom archive attachment failed twice, retry later")
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue