tkActivity.cs 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148
  1. using dodohold.core;
  2. using CSRedis;
  3. using System.Data;
  4. namespace molilian.core
  5. {
  6. public partial class TkLogCore
  7. {
  8. static string queue_activity_key = "queue:activity_logs";
  9. public static async Task<int> InsertActivityLogAsync(int limit, YunhuiKit.RedisClient redis)
  10. {
  11. int total = 0;
  12. using var connection = DBContext.GetOpenConnection();
  13. connection.Open();
  14. using var transaction = connection.BeginTransaction();
  15. try
  16. {
  17. for (int i = 0; i < limit; i++)
  18. {
  19. var data = await redis.LPopAsync<TkActivityDTO>(queue_activity_key);
  20. if (data == null) break;
  21. if (TestParseCore.InWhitelist(data.ip, data.oaid))
  22. {
  23. // For test environment, could create a test version if needed
  24. string daily_table = $"tk_activity_logs_test";
  25. save_activity_logs(data, daily_table, connection, transaction);
  26. }
  27. else
  28. {
  29. string daily_table = $"tk_activity_logs_{data.create_time:yyyyMMdd}";
  30. save_activity_logs(data, daily_table, connection, transaction);
  31. }
  32. total++;
  33. }
  34. transaction.Commit();
  35. }
  36. catch (Exception ex)
  37. {
  38. transaction.Rollback();
  39. new LoggerLibrary("TkLogCore_error", "task_insert_activity_logs")
  40. .Info(ex.Message, ex.StackTrace)
  41. .SaveAsync();
  42. throw;
  43. }
  44. finally
  45. {
  46. connection.Close();
  47. }
  48. return total;
  49. }
  50. public static async Task TkActivityLogAsync(TkActivityDTO response)
  51. {
  52. try
  53. {
  54. if (response.create_time == null)
  55. {
  56. response.create_time = DateTime.Now;
  57. }
  58. var ts = DateTime.Now - response.create_time.Value;
  59. response.elapsedTime = (int)ts.TotalMilliseconds;
  60. _ = RedisHelper.RPushAsync(queue_activity_key, response);
  61. if (!TestParseCore.InWhitelist(response.ip, response.oaid))
  62. {
  63. saveActivityCache(response.page_id, response.accountId, response.accountName, response.success, response.message, response.reason);
  64. }
  65. if (response.success) await saveClientRequestTotalAsync(response.channel, response.ip, response.oaid);
  66. var account = await TkPoolCore.ALLGetOneAsync(response.accountId);
  67. bool is_hide = account == null ? false : account.is_hide;
  68. if (!response.success && "nologin".Equals(response.message))
  69. {
  70. switch (response.channel)
  71. {
  72. case TkChannelEnum.tb:
  73. TkPoolCore.Disabled(response.accountId, response.accountName, $"{response.reason}", is_hide);
  74. break;
  75. }
  76. }
  77. }
  78. catch (Exception ex)
  79. {
  80. _ = new LoggerLibrary("tkActivity", "database_error")
  81. .Info(response.Convert2Json())
  82. .Info(ex.Message, ex.StackTrace)
  83. .SaveAsync();
  84. }
  85. }
  86. private static void saveActivityCache(string channel, int accountId, string accountName, bool success, string message, string reason)
  87. {
  88. saveAccountActivityCache("all", success, message, reason);
  89. saveAccountActivityCache($"{channel}", success, message, reason);
  90. if (accountId != 0)
  91. {
  92. saveAccountActivityCache($"{channel}_{accountId}", success, message, reason);
  93. }
  94. }
  95. private static int save_activity_logs(TkActivityDTO data, string tablename, IDbConnection connection, IDbTransaction transaction)
  96. {
  97. return new DBContext.Table(connection, tablename)
  98. .Add("end_point", data.end_point)
  99. .Add("channel", data.channel)
  100. .Add("accountId", data.accountId)
  101. .Add("accountName", data.accountName)
  102. .Add("proxy_node", data.proxy_node)
  103. .Add("page_id", data.page_id)
  104. .Add("item_id", data.item_id)
  105. .Add("success", data.success)
  106. .Add("reason", data.reason)
  107. .Add("message", data.message)
  108. .Add("deeplink_url", data.deeplink_url)
  109. .Add("create_time", data.create_time)
  110. .Add("elapsedTime", data.elapsedTime)
  111. .Add("elapsedTime2", data.elapsedTime2)
  112. .Add("ip", data.ip)
  113. .Add("oaid", data.oaid)
  114. .Add("estimatedCommission", data.estimatedCommission)
  115. .Add("maxCommission", data.maxCommission)
  116. .Add("zkFinalPrice", data.zkFinalPrice)
  117. .Add("reservePrice", data.reservePrice)
  118. .Add("title", data.title)
  119. .Add("picUrl", data.picUrl)
  120. .Add("shortTitle", data.shortTitle)
  121. .Add("priceAfterCoupon", data.priceAfterCoupon)
  122. .Add("commissionRate", data.commissionRate)
  123. .Add("couponAmount", data.couponAmount)
  124. .Create(DBContext.InsertType.NORMAL, transaction);
  125. }
  126. private static void saveAccountActivityCache(string accountName, bool success, string message, string reason)
  127. {
  128. SaveStatsAccountCache("activity_total", accountName, success, message, reason);
  129. }
  130. }
  131. }