tool.cs 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  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. string daily_table = $"tool_parse_logs_{data.create_time:yyyyMMdd}";
  44. new DBContext.Table(connection, daily_table)
  45. .Add("end_point", data.end_point)
  46. .Add("channel", (int)data.channel)
  47. .Add("rawContent", data.rawContent)
  48. .Add("success", data.success)
  49. .Add("message", data.message)
  50. .Add("reason", data.reason)
  51. .Add("content", data.content)
  52. .Add("taoToken", data.taoToken)
  53. .Add("shortLinkurl", data.shortLinkurl)
  54. .Add("deeplink_url", data.deeplink_url)
  55. .Add("elapsedTime", data.elapsedTime)
  56. .Add("ip", data.ip)
  57. .Add("oaid", data.oaid)
  58. .Add("create_time", data.create_time)
  59. .Create(DBContext.InsertType.NORMAL, transaction);
  60. }
  61. total++;
  62. }
  63. transaction.Commit();
  64. }
  65. catch (Exception ex)
  66. {
  67. transaction.Rollback();
  68. new LoggerLibrary("TkLogCore_error", "task_insert_parse_tool_logs")
  69. .Info(ex.Message, ex.StackTrace)
  70. .SaveAsync();
  71. throw;
  72. }
  73. finally
  74. {
  75. connection.Close();
  76. }
  77. return total;
  78. }
  79. public static async Task ParseLogAsync(ToolParseDataDTO response)
  80. {
  81. try
  82. {
  83. var ts = DateTime.Now - response.create_time;
  84. response.elapsedTime = (int)ts.TotalMilliseconds;
  85. _ = RedisKit.RPushAsync(queue_parse_tool_key, response);
  86. if (!TestParseCore.InWhitelist(response.ip, response.oaid))
  87. {
  88. await SaveParseCacheAsync(response.channel.ToString(), 0, "tool",
  89. response.success, response.message, response.reason,
  90. response.deeplink_url);
  91. }
  92. }
  93. catch (Exception ex) { }
  94. }
  95. }
  96. }