using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc.Controllers; using Microsoft.AspNetCore.Mvc.Filters; using System; using System.Collections.Generic; using System.Linq; using System.Text; using dodohold.core; using Dataoke; using Google.Protobuf.WellKnownTypes; using System.Xml.Linq; using Mysqlx.Crud; using ZstdSharp.Unsafe; using YunhuiKit; namespace molilian.core { public partial class JdPoolCore { public enum JdAction { all, api, parse, coupon, promotionQuery } private static string _end_point; private static Dictionary _incomeAmt = new(); static JdPoolCore() { _end_point = Environment.GetEnvironmentVariable("EndPoint"); } private static readonly object _lockObj = new(); private static IEnumerable _cached; private static IEnumerable _all_cached; public static async Task GetOneAsync() { var list = await ListAsync(); if (!list.Any()) return null; return list.OrderBy(l => Guid.NewGuid()).FirstOrDefault(); } public static async Task GetSpecialOneAsync(int accountid) { var list = await AllListAsync(); if (!list.Any()) return null; return list.Where(e => e.id == accountid).OrderBy(l => Guid.NewGuid()).FirstOrDefault(); } public static async Task GetOneAsync(int id, JdAction action) { var list = await ListAsync(); if (!list.Any()) return null; var eligibleItems = new List(); foreach (var item in list.Where(e => e.id == id)) { if (await IsNotExceedDailyIncomeLimitAsync(item, action)) { eligibleItems.Add(item); } } return eligibleItems.OrderBy(l => Guid.NewGuid()).FirstOrDefault(); } public static async Task GetOneAsync(JdAction action) { var list = await ListAsync(); if (!list.Any()) return null; var eligibleItems = new List(); foreach (var item in list) { if (await IsNotExceedDailyIncomeLimitAsync(item, action)) { eligibleItems.Add(item); } } return eligibleItems.OrderBy(l => Guid.NewGuid()).FirstOrDefault(); } private static async Task IsNotExceedDailyIncomeLimitAsync(JdPoolDTO item, JdAction action) { if (!string.IsNullOrEmpty(_end_point) && !string.IsNullOrEmpty(item.end_point)) { if (item.end_point != _end_point) return false; } switch (action) { case JdAction.api: if (item.work_mode != JdUnionWorkMode.SiteApi) return false; break; case JdAction.parse: if (!item.enable_parse) return false; break; case JdAction.coupon: if (!item.enable_coupon) return false; break; } // 使用初始化参数创建工作时间表 if (!new WorkSchedule(item.time_range).IsWorkHour()) return false; if (item.daily_calls_limit > 0) { int daily_num = RiskControlCore.GetCalls(TkChannelEnum.jd, item.id, DateTime.Now.ToString("yyyyMMdd")); if (daily_num >= item.daily_calls_limit) return false; } if (item.hourly_calls_limit > 0) { int hourly_num = RiskControlCore.GetCalls(TkChannelEnum.jd, item.id, DateTime.Now.ToString("yyyyMMddHH")); if (hourly_num >= item.hourly_calls_limit) return false; } return true; } public static async Task> AllListAsync(bool force = false) { if (!force && _all_cached != default) return _all_cached; string cache_key = $"cache:all_jd_pool"; var list = await RedisKit.GetAsync>(cache_key); if (force || list == default) { lock (_lockObj) { list = new DBContext.Table("jd_pool").Select(); if (list == null) return default; _ = RedisKit.SetAsync(cache_key, list, 30 * 86400); } } _all_cached = list; return list; } public static async Task> ListAsync(bool force = false) { if (!force && _cached != null) return _cached; string cache_key = $"cache:jd_pool"; var list = await RedisKit.GetAsync>(cache_key); if (force || list == null) { lock (_lockObj) { list = new DBContext.Table("jd_pool") .Where("status=@status", new { status = 1 }) .Select(); if (list == null) return default; foreach (var item in list) { _ = RiskControlCore.SetCallsAsync(TkChannelEnum.jd, item.id, DateTime.Now.ToString("yyyyMMddHH"), item.current_hourly_calls); _ = RiskControlCore.SetCallsAsync(TkChannelEnum.jd, item.id, DateTime.Now.ToString("yyyyMMdd"), item.current_daily_calls); } _ = RedisKit.SetAsync(cache_key, list, 30 * 86400); } } _cached = list; return list; } public static void Refresh() { _all_cached = []; _cached = []; _ = ListAsync(true); _ = AllListAsync(); } public static int Update(JdPoolDTO account) { return new DBContext.Table("jd_pool") .Add("current_hourly_calls", account.current_hourly_calls) .Add("current_daily_calls", account.current_daily_calls) .Add("today_clickNum", account.today_clickNum) .Add("today_cosFee", account.today_cosFee) .Add("today_cosPrice", account.today_cosPrice) .Add("today_finishCosFee", account.today_finishCosFee) .Add("today_finishCosPrice", account.today_finishCosPrice) .Add("today_finishOrderNum", account.today_finishOrderNum) .Add("today_orderNum", account.today_orderNum) .Where("id=@id", new { account.id }) .Update(); } public static int UpdateCookies(string cookies, string user_agent, string h5st) { if (string.IsNullOrEmpty(cookies)) return 0; string pin = cookies.GetContentPart("pin=", ";"); pin = pin.UrlDecode(); string company = pin; int accountId = 0; if (string.IsNullOrEmpty(pin)) return 0; var exist = new DBContext.Table("jd_pool").Get("pin=@pin", new { pin }); if (exist != null) { var status = exist.status; var work_mode = exist.work_mode; accountId = exist.id; if (work_mode == JdUnionWorkMode.Crawler) status = true; var update = new DBContext.Table("jd_pool") .Add("pin", pin) .Add("cookies", cookies) .Add("status", status) .Add("last_time", DateTime.Now) .Add("login_time", DateTime.Now) .Where("id=@id", new { exist.id }); if (!string.IsNullOrEmpty(user_agent)) update.Add("user_agent", user_agent); if (!string.IsNullOrEmpty(h5st)) update.Add("h5st", h5st); update.Update(); if (status) _ = ListAsync(true); } else { var update = new DBContext.Table("jd_pool") .Add("pin", pin) .Add("name", pin) .Add("company", pin) .Add("description", "由cookies上报创建此记录") .Add("cookies", cookies) .Add("create_time", DateTime.Now) .Add("last_time", DateTime.Now) .Add("login_time", DateTime.Now) .Add("status", 0); if (!string.IsNullOrEmpty(user_agent)) update.Add("user_agent", user_agent); if (!string.IsNullOrEmpty(h5st)) update.Add("h5st", h5st); accountId = update.Create(); } NotifyCore.Notify(new NifyMessage { message = $"【京东{accountId}:{company}】cookie 上线", tags = ["green_circle"] }); _ = EndPointCore.NotifyReload(true); //NotifyCore.AnPushNotify("上线", $"【淘宝联盟:{dnk}】cookie 上报更新"); return accountId; } public static async Task DisabledAsync(int accountId, string name, string content) { string cache_key = $"cache:jd_pool:{name}:disabled"; long count = await RedisKit.IncrByAsync(cache_key); _ = RedisKit.ExpireAsync(cache_key, 10); if (count > 1) return; var update = new DBContext.Table("jd_pool").Add("status", 0); if (accountId > 0) { update.Where("id=@accountId", new { accountId }).Update(); } else { update.Where("name=@name", new { name }).Update(); } _ = ListAsync(true); NotifyCore.Notify(new NifyMessage { message = $"【京东联盟{accountId}:{name}】cookie 掉线\n\n{content}", priority = NifyMessagePriority.high, tags = ["red_circle"] }); //NotifyCore.AnPushNotify("掉线", $"【京东{accountId}:{name}】cookie 掉线"); _ = EndPointCore.NotifyReload(true); } internal static async Task AccountExhaustedAsync() { string cache_key = $"cache:tk_pool:account:exhausted"; long count = await RedisKit.IncrByAsync(cache_key); if (count > 1) return; _ = RedisKit.ExpireAsync(cache_key, 3600); NotifyCore.Notify(new NifyMessage { message = $"【京东联盟】没有匹配账号", priority = NifyMessagePriority.high, tags = ["red_circle"] }); NotifyCore.AnPushNotify("没账号", $"【京东联盟】没有匹配账号"); } } }