Skip to content

Commit

Permalink
中间件重命名
Browse files Browse the repository at this point in the history
  • Loading branch information
xljiulang committed Jun 11, 2021
1 parent a2ea783 commit c86e371
Showing 1 changed file with 64 additions and 0 deletions.
64 changes: 64 additions & 0 deletions FastGithub/Middlewares/HttpsScanMiddleware.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using System;
using System.Linq;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;

namespace FastGithub.Middlewares
{
sealed class HttpsScanMiddleware : IGithubMiddleware
{
private readonly IOptionsMonitor<GithubOptions> options;
private readonly ILogger<HttpsScanMiddleware> logger;

public HttpsScanMiddleware(
IOptionsMonitor<GithubOptions> options,
ILogger<HttpsScanMiddleware> logger)
{
this.options = options;
this.logger = logger;
}

public async Task InvokeAsync(GithubContext context, Func<Task> next)
{
try
{
var request = new HttpRequestMessage
{
Method = HttpMethod.Get,
RequestUri = new Uri($"https://{context.Address}"),
};
request.Headers.Host = context.Domain;

using var httpClient = new HttpClient(new HttpClientHandler
{
Proxy = null,
UseProxy = false,
ServerCertificateCustomValidationCallback = (_, _, _, _) => true
});

var startTime = DateTime.Now;
using var cancellationTokenSource = new CancellationTokenSource(this.options.CurrentValue.HttpsScanTimeout);
var response = await httpClient.SendAsync(request, HttpCompletionOption.ResponseHeadersRead, cancellationTokenSource.Token);
var server = response.EnsureSuccessStatusCode().Headers.Server;
if (server.Any(s => string.Equals("GitHub.com", s.Product?.Name, StringComparison.OrdinalIgnoreCase)))
{
context.HttpElapsed = DateTime.Now.Subtract(startTime);
this.logger.LogWarning(context.ToString());

await next();
}
}
catch (TaskCanceledException)
{
this.logger.LogInformation($"{context.Domain} {context.Address}连接超时");
}
catch (Exception ex)
{
this.logger.LogInformation($"{context.Domain} {context.Address} {ex.Message}");
}
}
}
}

0 comments on commit c86e371

Please sign in to comment.