tool.cs 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. using dodohold.core;
  2. using CSRedis;
  3. using System.Data;
  4. using YunhuiKit;
  5. namespace molilian.core
  6. {
  7. public partial class TkLogCore
  8. {
  9. static string queue_parse_tool_key = "queue:parse_logs:tool";
  10. public static async Task<int> InsertToolLogAsync(int limit, YunhuiKit.RedisClient redis)
  11. {
  12. int total = 0;
  13. using var connection = DBContext.GetOpenConnection();
  14. connection.Open();
  15. using var transaction = connection.BeginTransaction();
  16. try
  17. {
  18. for (int i = 0; i < limit; i++)
  19. {
  20. var data = await redis.LPopAsync<ToolParseDataDTO>(queue_parse_tool_key);
  21. if (data == null) break;
  22. if (TestParseCore.InWhitelist(data.ip, data.oaid))
  23. {
  24. new DBContext.Table(connection, "tool_parse_logs_test")
  25. .Add("end_point", data.end_point)
  26. .Add("channel", (int)data.channel)
  27. .Add("rawContent", data.rawContent)
  28. .Add("success", data.success)
  29. .Add("message", data.message)
  30. .Add("reason", data.reason)
  31. .Add("content", data.content)
  32. .Add("taoToken", data.taoToken)
  33. .Add("shortLinkurl", data.shortLinkurl)
  34. .Add("deeplink_url", data.deeplink_url)
  35. .Add("elapsedTime", data.elapsedTime)
  36. .Add("ip", data.ip)
  37. .Add("oaid", data.oaid)
  38. .Add("create_time", data.create_time)
  39. .Create(DBContext.InsertType.NORMAL, transaction);
  40. }
  41. else
  42. {
  43. new DBContext.Table(connection, "tool_parse_logs")
  44. .Add("end_point", data.end_point)
  45. .Add("channel", (int)data.channel)
  46. .Add("rawContent", data.rawContent)
  47. .Add("success", data.success)
  48. .Add("message", data.message)
  49. .Add("reason", data.reason)
  50. .Add("content", data.content)
  51. .Add("taoToken", data.taoToken)
  52. .Add("shortLinkurl", data.shortLinkurl)
  53. .Add("deeplink_url", data.deeplink_url)
  54. .Add("elapsedTime", data.elapsedTime)
  55. .Add("ip", data.ip)
  56. .Add("oaid", data.oaid)
  57. .Add("create_time", data.create_time)
  58. .Create(DBContext.InsertType.NORMAL, transaction);
  59. }
  60. total++;
  61. }
  62. transaction.Commit();
  63. }
  64. catch (Exception ex)
  65. {
  66. transaction.Rollback();
  67. new LoggerLibrary("TkLogCore_error", "task_insert_parse_tool_logs")
  68. .Info(ex.Message, ex.StackTrace)
  69. .SaveAsync();
  70. throw;
  71. }
  72. finally
  73. {
  74. connection.Close();
  75. }
  76. return total;
  77. }
  78. public static int task_insert_parse_tool_logs(int limit, CSRedisClient redis)
  79. {
  80. int total = 0;
  81. using var connection = DBContext.GetOpenConnection();
  82. connection.Open();
  83. using var transaction = connection.BeginTransaction();
  84. try
  85. {
  86. for (int i = 0; i < limit; i++)
  87. {
  88. var data = redis.LPop<ToolParseDataDTO>(queue_parse_tool_key);
  89. if (data == null) break;
  90. if (TestParseCore.InWhitelist(data.ip, data.oaid))
  91. {
  92. new DBContext.Table(connection, "tool_parse_logs_test")
  93. .Add("end_point", data.end_point)
  94. .Add("channel", (int)data.channel)
  95. .Add("rawContent", data.rawContent)
  96. .Add("success", data.success)
  97. .Add("message", data.message)
  98. .Add("reason", data.reason)
  99. .Add("content", data.content)
  100. .Add("taoToken", data.taoToken)
  101. .Add("shortLinkurl", data.shortLinkurl)
  102. .Add("deeplink_url", data.deeplink_url)
  103. .Add("elapsedTime", data.elapsedTime)
  104. .Add("ip", data.ip)
  105. .Add("oaid", data.oaid)
  106. .Add("create_time", data.create_time)
  107. .Create(DBContext.InsertType.NORMAL, transaction);
  108. }
  109. else
  110. {
  111. new DBContext.Table(connection, "tool_parse_logs")
  112. .Add("end_point", data.end_point)
  113. .Add("channel", (int)data.channel)
  114. .Add("rawContent", data.rawContent)
  115. .Add("success", data.success)
  116. .Add("message", data.message)
  117. .Add("reason", data.reason)
  118. .Add("content", data.content)
  119. .Add("taoToken", data.taoToken)
  120. .Add("shortLinkurl", data.shortLinkurl)
  121. .Add("deeplink_url", data.deeplink_url)
  122. .Add("elapsedTime", data.elapsedTime)
  123. .Add("ip", data.ip)
  124. .Add("oaid", data.oaid)
  125. .Add("create_time", data.create_time)
  126. .Create(DBContext.InsertType.NORMAL, transaction);
  127. }
  128. total++;
  129. }
  130. transaction.Commit();
  131. }
  132. catch (Exception ex)
  133. {
  134. transaction.Rollback();
  135. new LoggerLibrary("TkLogCore_error", "task_insert_parse_tool_logs")
  136. .Info(ex.Message, ex.StackTrace)
  137. .SaveAsync();
  138. throw;
  139. }
  140. finally
  141. {
  142. connection.Close();
  143. }
  144. return total;
  145. }
  146. public static async Task ParseLogAsync(ToolParseDataDTO response)
  147. {
  148. try
  149. {
  150. var ts = DateTime.Now - response.create_time;
  151. response.elapsedTime = (int)ts.TotalMilliseconds;
  152. _ = RedisKit.RPushAsync(queue_parse_tool_key, response);
  153. if (!TestParseCore.InWhitelist(response.ip, response.oaid))
  154. {
  155. await SaveParseCacheAsync(response.channel.ToString(), 0, "tool",
  156. response.success, response.message, response.reason,
  157. response.deeplink_url);
  158. }
  159. }
  160. catch (Exception ex) { }
  161. }
  162. }
  163. }