Fun/Snap.Hutao
1
0
Fork 0
mirror of https://jihulab.com/DGP-Studio/Snap.Hutao.git synced 2025-11-19 21:02:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

single threaded update package download

Browse Source
This commit is contained in:
DismissedLight
2024-07-22 14:28:21 +08:00
parent f3934ce2cd
commit 04cd4e7137
1 changed files with 26 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/Snap.Hutao/Snap.Hutao/Service/Update/UpdateService.cs
View File

@@ -2,8 +2,8 @@
// Licensed under the MIT license. // Licensed under the MIT license.
using Snap.Hutao.Core; using Snap.Hutao.Core;
using Snap.Hutao.Core.IO;
using Snap.Hutao.Core.IO.Hashing; using Snap.Hutao.Core.IO.Hashing;
using Snap.Hutao.Core.IO.Http.Sharding;
using Snap.Hutao.Core.Setting; using Snap.Hutao.Core.Setting;
using Snap.Hutao.Service.Abstraction; using Snap.Hutao.Service.Abstraction;
using Snap.Hutao.Service.Notification; using Snap.Hutao.Service.Notification;
@@ -12,7 +12,6 @@ using Snap.Hutao.Web.Hutao.Response;
using System.Diagnostics; using System.Diagnostics;
using System.IO; using System.IO;
using System.Net.Http; using System.Net.Http;
using Windows.Storage;
namespace Snap.Hutao.Service.Update; namespace Snap.Hutao.Service.Update;
@@ -138,27 +137,35 @@ internal sealed partial class UpdateService : IUpdateService
using (HttpClient httpClient = scope.ServiceProvider.GetRequiredService<HttpClient>()) using (HttpClient httpClient = scope.ServiceProvider.GetRequiredService<HttpClient>())
{ {
string version = versionInformation.Version.ToString(); string version = versionInformation.Version.ToString();
foreach (string url in versionInformation.Urls) using (FileStream fileStream = File.Create(filePath))
{ {
HttpShardCopyWorkerOptions<UpdateStatus> options = new() foreach (string url in versionInformation.Urls)
{ {
HttpClient = httpClient, fileStream.Seek(0, SeekOrigin.Begin);
SourceUrl = url,
DestinationFilePath = filePath,
MaxDegreeOfParallelism = Math.Clamp(Environment.ProcessorCount, 2, 6),
StatusFactory = (bytesRead, totalBytes) => new UpdateStatus(version, bytesRead, totalBytes),
};
using (HttpShardCopyWorker<UpdateStatus> worker = await HttpShardCopyWorker<UpdateStatus>.CreateAsync(options).ConfigureAwait(false)) try
{ {
await worker.CopyAsync(progress, token).ConfigureAwait(false); using (HttpResponseMessage responseMessage = await httpClient.GetAsync(url, HttpCompletionOption.ResponseHeadersRead, token).ConfigureAwait(false))
} {
long totalBytes = responseMessage.Content.Headers.ContentLength ?? 0;
using (Stream webStream = await responseMessage.Content.ReadAsStreamAsync(token).ConfigureAwait(false))
{
StreamCopyWorker<UpdateStatus> worker = new(webStream, fileStream, bytesRead => new UpdateStatus(version, bytesRead, totalBytes));
string? remoteHash = versionInformation.Sha256; await worker.CopyAsync(progress).ConfigureAwait(false);
ArgumentNullException.ThrowIfNull(remoteHash);
if (await CheckUpdateCacheSHA256Async(filePath, remoteHash, token).ConfigureAwait(false)) string? remoteHash = versionInformation.Sha256;
{ ArgumentNullException.ThrowIfNull(remoteHash);
return true; if (await CheckUpdateCacheSHA256Async(filePath, remoteHash, token).ConfigureAwait(false))
{
return true;
}
}
}
}
catch
{
}
} }
} }
} }