public async Task<FileInfo> DownloadAsync(
SourceImage image,
ImageEntry entry,
DownloadTracker tracker,
DatabaseTracker dbTracker)
{
var file = image.File;
var started = false;
long downloaded = 0;
try
{
var response = await this.Client.GetObjectAsync(new GetObjectRequest()
{
BucketName = this.BucketName,
Key = image.Key
});
var tmp = Path.GetTempFileName();
long transferred = 0;
response.WriteObjectProgressEvent += (sender, p) =>
{
if (!started)
{
tracker.Start();
}
started = true;
downloaded = p.TransferredBytes;
tracker.Downloaded(downloaded - transferred);
transferred = downloaded;
};
await response.WriteResponseStreamToFileAsync(tmp, false, CancellationToken.None);
if (this.DebugMode)
{
File.Delete(tmp);
}
else
{
if (!this.SafeMove(tmp, file.FullName))
{
image.MarkFailed();
dbTracker.Update(entry, ImageStatus.Failed);
tracker.Fail();
return null;
}
}
this.Logger.LogDebug("Download Complete: {0} -> {1}", image.Key, file?.FullName ?? "-");
image.MarkCompleted();
dbTracker.Update(entry, ImageStatus.Completed);
if (started)
{
tracker.Complete();
}
return file;
}
catch (Exception ec)
{
this.Logger.LogDebug("Download Failed: {0} -> {1}", image.Key, file?.FullName ?? "-");
this.Logger.LogDebug("Exception: {0}", ec.ToString());
image.MarkFailed();
dbTracker.Update(entry, ImageStatus.Failed);
if (started)
{
tracker.Fail();
tracker.Downloaded(image.TotalBytes - downloaded);
}
return null;
}
}
