52 lines
2.0 KiB
C#
52 lines
2.0 KiB
C#
using FastGithub.Scanner;
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using System;
|
|
using System.Net.Http;
|
|
using System.Net.Http.Headers;
|
|
|
|
namespace FastGithub
|
|
{
|
|
/// <summary>
|
|
/// 服务注册扩展
|
|
/// </summary>
|
|
public static class ScannerServiceCollectionExtensions
|
|
{
|
|
/// <summary>
|
|
/// 注册程序集下所有服务下选项
|
|
/// </summary>
|
|
/// <param name="services"></param>
|
|
/// <param name="configuration">配置</param>
|
|
/// <returns></returns>
|
|
public static IServiceCollection AddGithubScanner(this IServiceCollection services, IConfiguration configuration)
|
|
{
|
|
var assembly = typeof(ScannerServiceCollectionExtensions).Assembly;
|
|
var defaultUserAgent = new ProductInfoHeaderValue(assembly.GetName().Name ?? nameof(FastGithub), assembly.GetName().Version?.ToString());
|
|
|
|
services
|
|
.AddHttpClient(nameof(FastGithub))
|
|
.SetHandlerLifetime(TimeSpan.FromMinutes(5d))
|
|
.ConfigureHttpClient(httpClient =>
|
|
{
|
|
httpClient.DefaultRequestHeaders.Accept.TryParseAdd("*/*");
|
|
httpClient.DefaultRequestHeaders.UserAgent.Add(defaultUserAgent);
|
|
})
|
|
.ConfigurePrimaryHttpMessageHandler(() => new HttpClientHandler
|
|
{
|
|
Proxy = null,
|
|
UseProxy = false,
|
|
AllowAutoRedirect = false
|
|
})
|
|
.AddHttpMessageHandler<LoopbackDnsHttpHandler>();
|
|
|
|
return services
|
|
.AddMemoryCache()
|
|
.AddServiceAndOptions(assembly, configuration)
|
|
.AddHostedService<GithubFullScanHostedService>()
|
|
.AddHostedService<GithubResultScanHostedService>()
|
|
.AddSingleton<IGithubScanResults>(appService => appService.GetRequiredService<GithubContextCollection>());
|
|
;
|
|
}
|
|
}
|
|
}
|