This commit is contained in:
陈国伟 2022-09-20 15:14:03 +08:00
parent bd70867481
commit 893e9729ca

View File

@ -6,6 +6,7 @@ using System.IO;
using System.Linq; using System.Linq;
using System.Net; using System.Net;
using System.Text.Json; using System.Text.Json;
using System.Text.Json.Serialization;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
@ -14,21 +15,25 @@ namespace FastGithub.DomainResolve
/// <summary> /// <summary>
/// 域名持久化 /// 域名持久化
/// </summary> /// </summary>
sealed class PersistenceService sealed partial class PersistenceService
{ {
private static readonly string dataFile = "dnsendpoints.json"; private static readonly string dataFile = "dnsendpoints.json";
private static readonly SemaphoreSlim dataLocker = new(1, 1); private static readonly SemaphoreSlim dataLocker = new(1, 1);
private static readonly JsonSerializerOptions jsonOptions = new()
{
WriteIndented = true,
PropertyNameCaseInsensitive = true,
PropertyNamingPolicy = JsonNamingPolicy.CamelCase
};
private readonly FastGithubConfig fastGithubConfig; private readonly FastGithubConfig fastGithubConfig;
private readonly ILogger<PersistenceService> logger; private readonly ILogger<PersistenceService> logger;
private record EndPointItem(string Host, int Port); private record EndPointItem(string Host, int Port);
[JsonSerializable(typeof(EndPointItem[]))]
[JsonSourceGenerationOptions(
WriteIndented = true,
PropertyNamingPolicy = JsonKnownNamingPolicy.CamelCase)]
private partial class EndPointItemsContext : JsonSerializerContext
{
}
/// <summary> /// <summary>
/// 域名持久化 /// 域名持久化
@ -60,7 +65,7 @@ namespace FastGithub.DomainResolve
dataLocker.Wait(); dataLocker.Wait();
var utf8Json = File.ReadAllBytes(dataFile); var utf8Json = File.ReadAllBytes(dataFile);
var endPointItems = JsonSerializer.Deserialize<EndPointItem[]>(utf8Json, jsonOptions); var endPointItems = JsonSerializer.Deserialize(utf8Json, EndPointItemsContext.Default.EndPointItemArray);
if (endPointItems == null) if (endPointItems == null)
{ {
return Array.Empty<DnsEndPoint>(); return Array.Empty<DnsEndPoint>();
@ -100,7 +105,7 @@ namespace FastGithub.DomainResolve
await dataLocker.WaitAsync(CancellationToken.None); await dataLocker.WaitAsync(CancellationToken.None);
var endPointItems = dnsEndPoints.Select(item => new EndPointItem(item.Host, item.Port)).ToArray(); var endPointItems = dnsEndPoints.Select(item => new EndPointItem(item.Host, item.Port)).ToArray();
var utf8Json = JsonSerializer.SerializeToUtf8Bytes(endPointItems, jsonOptions); var utf8Json = JsonSerializer.SerializeToUtf8Bytes(endPointItems, EndPointItemsContext.Default.EndPointItemArray);
await File.WriteAllBytesAsync(dataFile, utf8Json, cancellationToken); await File.WriteAllBytesAsync(dataFile, utf8Json, cancellationToken);
} }
catch (Exception ex) catch (Exception ex)