60 lines
2.0 KiB
C#
60 lines
2.0 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Logging;
|
|
using Microsoft.Extensions.Options;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace FastGithub.Scanner
|
|
{
|
|
/// <summary>
|
|
/// 域名与ip关系工厂
|
|
/// </summary>
|
|
[Service(ServiceLifetime.Singleton)]
|
|
sealed class GithubLookupFacotry
|
|
{
|
|
private readonly IEnumerable<IGithubLookupProvider> providers;
|
|
private readonly IOptionsMonitor<GithubLookupFactoryOptions> options;
|
|
private readonly ILogger<GithubLookupFacotry> logger;
|
|
|
|
/// <summary>
|
|
/// 域名与ip关系工厂
|
|
/// </summary>
|
|
/// <param name="providers"></param>
|
|
/// <param name="options"></param>
|
|
public GithubLookupFacotry(
|
|
IEnumerable<IGithubLookupProvider> providers,
|
|
IOptionsMonitor<GithubLookupFactoryOptions> options,
|
|
ILogger<GithubLookupFacotry> logger)
|
|
{
|
|
this.providers = providers.OrderBy(item => item.Order);
|
|
this.options = options;
|
|
this.logger = logger;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 查找域名与ip关系
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public async Task<IEnumerable<DomainAddress>> LookupAsync(CancellationToken cancellationToken)
|
|
{
|
|
this.logger.LogInformation($"开始查找各域名的ip..");
|
|
var hashSet = new HashSet<DomainAddress>();
|
|
var domains = this.options.CurrentValue.Domains;
|
|
|
|
foreach (var provider in this.providers)
|
|
{
|
|
var domainAddresses = await provider.LookupAsync(domains, cancellationToken);
|
|
foreach (var item in domainAddresses)
|
|
{
|
|
hashSet.Add(item);
|
|
}
|
|
}
|
|
|
|
this.logger.LogInformation($"查找到{hashSet.Count}条域名ip记录");
|
|
return hashSet;
|
|
}
|
|
}
|
|
}
|