EndPointCore.cs 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. using Microsoft.AspNetCore.Http;
  2. using Microsoft.AspNetCore.Mvc.Controllers;
  3. using Microsoft.AspNetCore.Mvc.Filters;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Linq;
  7. using System.Text;
  8. using dodohold.core;
  9. using Spire.Pdf.Exporting.XPS.Schema;
  10. using System.Xml.Linq;
  11. using System.Net;
  12. using TencentCloud.Mrs.V20200910.Models;
  13. using static molilian.core.TkPoolCore;
  14. using Org.BouncyCastle.Tls;
  15. using MySql.Data.MySqlClient;
  16. using System.Data;
  17. namespace molilian.core
  18. {
  19. public partial class EndPointCore
  20. {
  21. private static readonly object _lockObj = new();
  22. private static IEnumerable<EndPointDTO> _cached;
  23. public static IDbConnection GetDbConnection(string connString)
  24. {
  25. var conn = new MySqlConnection(connString);
  26. return conn;
  27. }
  28. public static EndPointDTO GetOne(string node)
  29. {
  30. var list = List();
  31. if (!list.Any()) return null;
  32. var item = list.Where(s => s.name == node).FirstOrDefault();
  33. return item;
  34. }
  35. public static EndPointDTO GetCouponAdmin()
  36. {
  37. var list = List();
  38. if (!list.Any()) return null;
  39. var item = list.Where(s => s.is_coupon_api && !s.is_public_api).FirstOrDefault();
  40. #if DEBUG
  41. item.db_server = "Server=rm-2ze74506m3gfsqe7mco.rwlb.rds.aliyuncs.com; Port=3306; Database=coupon; Uid=coupon; Pwd=67ktWBmw5G4yMs4J;SslMode=None;CharSet=utf8mb4;ConnectionTimeout=60;";
  42. #endif
  43. /*
  44. Server=rm-2zey1jqxnoqy9mcc0zo.rwlb.rds.aliyuncs.com; Port=3306; Database=taoke; Uid=taoke; Pwd=67ktWBmw5G4yMs4J;SslMode=None;CharSet=utf8mb4;ConnectionTimeout=60
  45. Server=rm-2zey1jqxnoqy9mcc0zo.rwlb.rds.aliyuncs.com; Port=3306; Database=taoke; Uid=taoke; Pwd=67ktWBmw5G4yMs4J;SslMode=None;CharSet=utf8mb4;ConnectionTimeout=60
  46. Server=rm-2zey1jqxnoqy9mcc0zo.rwlb.rds.aliyuncs.com; Port=3306; Database=taoke; Uid=taoke; Pwd=67ktWBmw5G4yMs4J;SslMode=None;CharSet=utf8mb4;ConnectionTimeout=60
  47. Server=rm-2ze74506m3gfsqe7mco.rwlb.rds.aliyuncs.com; Port=3306; Database=coupon; Uid=coupon; Pwd=67ktWBmw5G4yMs4J;SslMode=None;CharSet=utf8mb4;ConnectionTimeout=60;
  48. */
  49. return item;
  50. }
  51. public static EndPointDTO GetParseAdmin()
  52. {
  53. var list = List();
  54. if (!list.Any()) return null;
  55. var item = list.Where(s => !s.is_coupon_api && !s.is_public_api).FirstOrDefault();
  56. #if DEBUG
  57. item.db_server = "Server=rm-2zey1jqxnoqy9mcc0zo.rwlb.rds.aliyuncs.com; Port=3306; Database=taoke; Uid=taoke; Pwd=67ktWBmw5G4yMs4J;SslMode=None;CharSet=utf8mb4;ConnectionTimeout=60;";
  58. #endif
  59. return item;
  60. }
  61. public static IEnumerable<EndPointDTO> List(bool force = false)
  62. {
  63. if (!force && _cached != null) return _cached;
  64. string cache_key = $"cache:end_point";
  65. var list = RedisHelper.Get<IEnumerable<EndPointDTO>>(cache_key);
  66. if (force || list == null)
  67. {
  68. lock (_lockObj)
  69. {
  70. list = new DBContext.Table("end_point")
  71. .Where("status=@status", new { status = 1 })
  72. .Select<EndPointDTO>();
  73. if (list == null) return default;
  74. RedisHelper.Set(cache_key, list, 30 * 86400);
  75. }
  76. }
  77. _cached = list;
  78. return list;
  79. }
  80. public static void Refresh()
  81. {
  82. _ = List(true);
  83. }
  84. public static List<T> ProcessEndPointNodes<T>(Func<EndPointDTO, T> nodeAction, bool force = false)
  85. {
  86. var resultList = new List<T>();
  87. var list = List(force);
  88. foreach (var node in list)
  89. {
  90. var result = nodeAction(node);
  91. if (result != null) resultList.Add(result);
  92. }
  93. return resultList;
  94. }
  95. public static async Task ProcessEndPointNodesAsync(Func<EndPointDTO, Task> nodeAction, bool force = false)
  96. {
  97. var list = List(force);
  98. var tasks = new List<Task>();
  99. foreach (var node in list)
  100. {
  101. tasks.Add(nodeAction(node));
  102. }
  103. await Task.WhenAll(tasks);
  104. }
  105. public static async Task NotifyReload(bool onlyAccount = false, CancellationToken cancellationToken = default)
  106. {
  107. await ProcessEndPointNodesAsync(async node =>
  108. {
  109. try
  110. {
  111. string url = $"{node.api_server}Task_70160bd632/reload";
  112. if (onlyAccount) url = $"{node.api_server}Task_70160bd632/reloadAccount";
  113. await new WebClientUtility().RequestAsync(url, "GET", cancellationToken);
  114. }
  115. catch (Exception ex)
  116. {
  117. _ = new LoggerLibrary("NotifyReload", "error")
  118. .Info(node.Convert2Json())
  119. .Info(ex.Message, ex.StackTrace)
  120. .SaveAsync();
  121. NotifyCore.Notify(new NifyMessage
  122. {
  123. message = $"【NotifyReload异常】{node.description}\n{ex.Message}\n{ex.StackTrace}",
  124. priority = NifyMessagePriority.high,
  125. tags = ["red_circle"]
  126. });
  127. }
  128. });
  129. }
  130. }
  131. }