强制使用自身反向代理进行升级检测

This commit is contained in:
陈国伟 2021-07-22 09:36:59 +08:00
parent e9a8ce8ece
commit e8a3c5eb84
4 changed files with 58 additions and 8 deletions

View File

@ -8,4 +8,8 @@
<PackageReference Include="Microsoft.Extensions.Hosting" Version="5.0.0" /> <PackageReference Include="Microsoft.Extensions.Hosting" Version="5.0.0" />
</ItemGroup> </ItemGroup>
<ItemGroup>
<ProjectReference Include="..\FastGithub.Core\FastGithub.Core.csproj" />
</ItemGroup>
</Project> </Project>

View File

@ -0,0 +1,43 @@
using System;
using System.Net;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;
namespace FastGithub.Upgrade
{
/// <summary>
/// 本机反向代理的httpHandler
/// </summary>
sealed class ReverseProxyHttpHandler : DelegatingHandler
{
/// <summary>
/// 本机反向代理的httpHandler
/// </summary>
public ReverseProxyHttpHandler()
{
this.InnerHandler = new HttpClientHandler();
}
/// <summary>
/// 替换为Loopback
/// </summary>
/// <param name="request"></param>
/// <param name="cancellationToken"></param>
/// <returns></returns>
protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
{
var uri = request.RequestUri;
if (uri != null && uri.HostNameType == UriHostNameType.Dns)
{
var domain = uri.Host;
var builder = new UriBuilder(uri) { Host = IPAddress.Loopback.ToString() };
request.RequestUri = builder.Uri;
request.Headers.Host = domain;
}
return await base.SendAsync(request, cancellationToken);
}
}
}

View File

@ -33,7 +33,7 @@ namespace FastGithub.Upgrade
/// <returns></returns> /// <returns></returns>
protected override async Task ExecuteAsync(CancellationToken stoppingToken) protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{ {
var maxTryCount = 3; var maxTryCount = 5;
for (var i = 1; i <= maxTryCount; i++) for (var i = 1; i <= maxTryCount; i++)
{ {
try try

View File

@ -59,7 +59,10 @@ namespace FastGithub.Upgrade
/// <returns></returns> /// <returns></returns>
public async Task<GithubRelease?> GetLastedReleaseAsync(CancellationToken cancellationToken) public async Task<GithubRelease?> GetLastedReleaseAsync(CancellationToken cancellationToken)
{ {
using var httpClient = new HttpClient { Timeout = TimeSpan.FromSeconds(30d) }; using var httpClient = new HttpClient(new ReverseProxyHttpHandler())
{
Timeout = TimeSpan.FromSeconds(30d),
};
httpClient.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("*/*")); httpClient.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("*/*"));
httpClient.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(nameof(FastGithub), "1.0")); httpClient.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(nameof(FastGithub), "1.0"));