2
0

PersistenceService.cs 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. using FastGithub.Configuration;
  2. using Microsoft.Extensions.Logging;
  3. using System;
  4. using System.Collections.Generic;
  5. using System.IO;
  6. using System.Linq;
  7. using System.Net;
  8. using System.Text.Json;
  9. using System.Threading;
  10. using System.Threading.Tasks;
  11. namespace FastGithub.DomainResolve
  12. {
  13. /// <summary>
  14. /// 域名持久化
  15. /// </summary>
  16. sealed class PersistenceService
  17. {
  18. private static readonly string dataFile = "dnsendpoints.json";
  19. private static readonly SemaphoreSlim dataLocker = new(1, 1);
  20. private static readonly JsonSerializerOptions jsonOptions = new()
  21. {
  22. WriteIndented = true,
  23. PropertyNameCaseInsensitive = true,
  24. PropertyNamingPolicy = JsonNamingPolicy.CamelCase
  25. };
  26. private readonly FastGithubConfig fastGithubConfig;
  27. private readonly ILogger<PersistenceService> logger;
  28. private record EndPointItem(string Host, int Port);
  29. /// <summary>
  30. /// 域名持久化
  31. /// </summary>
  32. /// <param name="fastGithubConfig"></param>
  33. /// <param name="logger"></param>
  34. public PersistenceService(
  35. FastGithubConfig fastGithubConfig,
  36. ILogger<PersistenceService> logger)
  37. {
  38. this.fastGithubConfig = fastGithubConfig;
  39. this.logger = logger;
  40. }
  41. /// <summary>
  42. /// 读取保存的节点
  43. /// </summary>
  44. /// <returns></returns>
  45. public IList<DnsEndPoint> ReadDnsEndPoints()
  46. {
  47. if (File.Exists(dataFile) == false)
  48. {
  49. return Array.Empty<DnsEndPoint>();
  50. }
  51. try
  52. {
  53. dataLocker.Wait();
  54. var utf8Json = File.ReadAllBytes(dataFile);
  55. var endPointItems = JsonSerializer.Deserialize<EndPointItem[]>(utf8Json, jsonOptions);
  56. if (endPointItems == null)
  57. {
  58. return Array.Empty<DnsEndPoint>();
  59. }
  60. var dnsEndPoints = new List<DnsEndPoint>();
  61. foreach (var item in endPointItems)
  62. {
  63. if (this.fastGithubConfig.IsMatch(item.Host) == true)
  64. {
  65. dnsEndPoints.Add(new DnsEndPoint(item.Host, item.Port));
  66. }
  67. }
  68. return dnsEndPoints;
  69. }
  70. catch (Exception ex)
  71. {
  72. this.logger.LogWarning(ex.Message, "读取dns记录异常");
  73. return Array.Empty<DnsEndPoint>();
  74. }
  75. finally
  76. {
  77. dataLocker.Release();
  78. }
  79. }
  80. /// <summary>
  81. /// 保存节点到文件
  82. /// </summary>
  83. /// <param name="dnsEndPoints"></param>
  84. /// <param name="cancellationToken"></param>
  85. /// <returns></returns>
  86. public async Task WriteDnsEndPointsAsync(IEnumerable<DnsEndPoint> dnsEndPoints, CancellationToken cancellationToken)
  87. {
  88. try
  89. {
  90. await dataLocker.WaitAsync(CancellationToken.None);
  91. var endPointItems = dnsEndPoints.Select(item => new EndPointItem(item.Host, item.Port)).ToArray();
  92. var utf8Json = JsonSerializer.SerializeToUtf8Bytes(endPointItems, jsonOptions);
  93. await File.WriteAllBytesAsync(dataFile, utf8Json, cancellationToken);
  94. }
  95. catch (Exception ex)
  96. {
  97. this.logger.LogWarning(ex.Message, "保存dns记录异常");
  98. }
  99. finally
  100. {
  101. dataLocker.Release();
  102. }
  103. }
  104. }
  105. }