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 static molilian.core.TkPoolCore; namespace molilian.core { public partial class JdPoolCore { public enum JdAction { all, parse, coupon, promotionQuery } private static string _end_point; static JdPoolCore() { _end_point = Environment.GetEnvironmentVariable("EndPoint"); } private static readonly object _lockObj = new(); private static IEnumerable _cached; public static JdPoolDTO? GetOne() { var list = List(); if (!list.Any()) return null; return list.OrderBy(l => Guid.NewGuid()).FirstOrDefault(); } public static JdPoolDTO? GetOne(JdAction action) { var list = List(); if (!list.Any()) return null; return list.Where(e => IsNotExceedDailyIncomeLimit(e, action)).OrderBy(l => Guid.NewGuid()).FirstOrDefault(); } public static JdPoolDTO? GetOne(int id) { var list = List(); if (!list.Any()) return null; return list.Where(e => e.id == id).FirstOrDefault(); } private static bool IsNotExceedDailyIncomeLimit(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.parse: if (!item.enable_parse) return false; break; case JdAction.coupon: if (!item.enable_coupon) return false; break; } return true; //if (item.daily_income_limit == 0) return true; //decimal income_amt = GetIncomeAmt($"{item.id}"); //return income_amt < item.daily_income_limit; } public static IEnumerable List(bool force = false) { if (!force && _cached != null) return _cached; string cache_key = $"cache:jd_pool"; var list = RedisHelper.Get>(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; RedisHelper.Set(cache_key, list, 30 * 86400); } } _cached = list; return list; } public static void Refresh() { _ = List(true); } public static int UpdateCookies(string cookies, string user_agent) { if (string.IsNullOrEmpty(cookies)) return 0; cookies += ";"; 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; new DBContext.Table("jd_pool") .Add("pin", pin) .Add("cookies", cookies) .Add("user_agent", user_agent) .Add("status", status) .Add("last_time", DateTime.Now) .Add("login_time", DateTime.Now) .Where("id=@id", new { exist.id }) .Update(); if (status) _ = List(true); } else { accountId = new DBContext.Table("jd_pool") .Add("pin", pin) .Add("name", pin) .Add("company", pin) .Add("description", "由cookies上报创建此记录") .Add("cookies", cookies) .Add("user_agent", user_agent) .Add("create_time", DateTime.Now) .Add("last_time", DateTime.Now) .Add("login_time", DateTime.Now) .Add("status", 0) .Create(); } NotifyCore.Notify(new NifyMessage { message = $"【京东{accountId}:{company}】cookie 上线", tags = ["green_circle"] }); EndPointCore.NotifyReload(true); //NotifyCore.AnPushNotify("上线", $"【淘宝联盟:{dnk}】cookie 上报更新"); return accountId; } public static void Disabled(int accountId, string name, string content) { string cache_key = $"cache:jd_pool:{name}:disabled"; long count = RedisHelper.IncrBy(cache_key); RedisHelper.Expire(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(); } _ = List(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 void AccountExhausted() { string cache_key = $"cache:tk_pool:account:exhausted"; long count = RedisHelper.IncrBy(cache_key); if (count > 1) return; RedisHelper.Expire(cache_key, 3600); NotifyCore.Notify(new NifyMessage { message = $"【京东联盟】没有匹配账号", priority = NifyMessagePriority.high, tags = ["red_circle"] }); NotifyCore.AnPushNotify("没账号", $"【京东联盟】没有匹配账号"); } } }