GptProxyRechargeServiceImpl.java 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542
  1. package com.cyksj.server.recharge.impl;
  2. import cn.hutool.core.date.DateTime;
  3. import cn.hutool.core.date.DateUtil;
  4. import cn.hutool.core.util.StrUtil;
  5. import cn.hutool.http.HttpRequest;
  6. import cn.hutool.http.HttpResponse;
  7. import cn.hutool.json.JSONObject;
  8. import cn.hutool.json.JSONUtil;
  9. import com.baomidou.mybatisplus.core.toolkit.Wrappers;
  10. import com.cyksj.common.constant.Constant;
  11. import com.cyksj.common.exception.BusinessRuntimeException;
  12. import com.cyksj.common.task.GlobalThreadPoolTaskExecutor;
  13. import com.cyksj.common.util.Jsons;
  14. import com.cyksj.mapper.*;
  15. import com.cyksj.model.dto.GptCardTokenDto;
  16. import com.cyksj.model.entity.*;
  17. import com.cyksj.model.request.GroupsRelationRechargeReq;
  18. import com.cyksj.redis.RedisService;
  19. import com.cyksj.server.recharge.GptProxyRechargeService;
  20. import com.cyksj.server.recharge.dto.CardKeyValidationResult;
  21. import com.cyksj.server.recharge.dto.TaskResult;
  22. import com.cyksj.server.recharge.dto.TaskSubmitResult;
  23. import com.cyksj.server.recharge.dto.TokenParseResult;
  24. import com.cyksj.service.chatgpt.GptUserRechargeRecordService;
  25. import com.cyksj.service.recharge.*;
  26. import com.cyksj.service.user.UserBindRelationService;
  27. import lombok.RequiredArgsConstructor;
  28. import lombok.extern.slf4j.Slf4j;
  29. import org.springframework.stereotype.Service;
  30. import org.springframework.transaction.annotation.Transactional;
  31. import java.util.Date;
  32. import java.util.List;
  33. import java.util.Optional;
  34. /**
  35. * @author chan
  36. * @date 2025/7/30 18:55
  37. */
  38. @Slf4j
  39. @Service
  40. @RequiredArgsConstructor
  41. public class GptProxyRechargeServiceImpl implements GptProxyRechargeService {
  42. private final GptRechargeCardKeyMapper gptRechargeCardKeyMapper;
  43. private final GroupsRelationMapper relationMapper;
  44. private final UserBindRelationService userBindRelationService;
  45. private final GroupsMapper groupsMapper;
  46. private final GoodsDonSkuMapper skuMapper;
  47. private final OrderDonMapper orderDonMapper;
  48. private final GptUserRechargeRecordService gptUserRechargeRecordService;
  49. private final RedisService redisService;
  50. private final GptRechargeCardKeyChannelService gptRechargeCardKeyChannelService;
  51. private final AfDianRechargeService afDianRechargeService;
  52. private final ChongAfDianRechargeService chongAfDianRechargeService;
  53. private final YearAfDianRechargeService yearAfDianRechargeService;
  54. private final GoAfDianRechargeService goAfDianRechargeService;
  55. private final HiBoyRechargeService hiBoyRechargeService;
  56. private final PlusAfDianRechargeService plusAfDianRechargeService;
  57. private final SancRechargeService sancRechargeService;
  58. private final CdkRechargeService cdkRechargeService;
  59. private final GuofenzaixianRechargeService guofenzaixianRechargeService;
  60. private final GptUserRechargeNumRecordMapper gptUserRechargeNumRecordMapper;
  61. private static final GlobalThreadPoolTaskExecutor TASK_EXECUTOR = GlobalThreadPoolTaskExecutor.getInstance();
  62. private HttpRequest createBaseRequest(String url) {
  63. return HttpRequest.get(url)
  64. .header("accept", "application/json, text/plain, */*")
  65. .header("accept-language", "zh-CN,zh;q=0.9,en;q=0.8")
  66. .header("origin", "https://www.ow520.com")
  67. .header("referer", "https://www.ow520.com/")
  68. .header("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36");
  69. }
  70. public CardKeyValidationResult validateCardKey(String cardKey) {
  71. log.info("开始验证卡密: {}", cardKey);
  72. try {
  73. String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
  74. HttpResponse response = createBaseRequest(BASE_URL + "/card-keys/" + cardKey)
  75. .execute();
  76. String body = response.body();
  77. log.info("卡密验证响应: {}", body);
  78. JSONObject jsonResponse = JSONUtil.parseObj(body);
  79. CardKeyValidationResult result = new CardKeyValidationResult(
  80. jsonResponse.getBool("available")
  81. );
  82. //{"available":false,"error":"卡密已被使用"}
  83. log.info("卡密验证结果: {}", result);
  84. return result;
  85. } catch (Exception e) {
  86. log.error("验证卡密失败: {}", e.getMessage(), e);
  87. throw new RuntimeException("验证卡密失败: " + e.getMessage(), e);
  88. }
  89. }
  90. public TokenParseResult parseToken(String accessToken) {
  91. log.info("开始验证凭证: {}", accessToken.substring(0, Math.min(20, accessToken.length())) + "...");
  92. try {
  93. String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(null);
  94. String url = BASE_URL + "/parse-token";
  95. JSONObject requestBody = new JSONObject();
  96. requestBody.put("access_token", accessToken);
  97. HttpResponse response = HttpRequest.post(url)
  98. .header("accept", "application/json, text/plain, */*")
  99. .header("accept-language", "zh-CN,zh;q=0.9,en;q=0.8")
  100. .header("content-type", "application/json")
  101. .header("origin", "https://www.ow520.com")
  102. .header("referer", "https://www.ow520.com/")
  103. .header("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36")
  104. .body(requestBody.toString())
  105. .execute();
  106. String body = response.body();
  107. log.info("凭证验证响应: {}", body);
  108. JSONObject jsonResponse = JSONUtil.parseObj(body);
  109. TokenParseResult result = new TokenParseResult(
  110. jsonResponse.getStr("message"),
  111. jsonResponse.getBool("success")
  112. );
  113. log.info("凭证验证结果: {}", result);
  114. return result;
  115. } catch (Exception e) {
  116. log.error("验证凭证失败: {}", e.getMessage(), e);
  117. throw new RuntimeException("验证凭证失败: " + e.getMessage(), e);
  118. }
  119. }
  120. public TaskSubmitResult submitTask(String cardKey, String accessToken, String idp) {
  121. log.info("开始提交任务 - 卡密: {}, idp: {}", cardKey, idp);
  122. try {
  123. String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
  124. String url = BASE_URL + "/tasks";
  125. JSONObject requestBody = new JSONObject();
  126. requestBody.put("card_key", cardKey);
  127. requestBody.put("access_token", accessToken);
  128. requestBody.put("idp", idp);
  129. HttpResponse response = HttpRequest.post(url)
  130. .setConnectionTimeout(60000)
  131. .header("accept", "application/json, text/plain, */*")
  132. .header("accept-language", "zh-CN,zh;q=0.9,en;q=0.8")
  133. .header("content-type", "application/json")
  134. .header("origin", "https://www.ow520.com")
  135. .header("referer", "https://www.ow520.com/")
  136. .header("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36")
  137. .body(requestBody.toString())
  138. .execute();
  139. String body = response.body();
  140. log.info("任务提交响应: {}", body);
  141. JSONObject jsonResponse = JSONUtil.parseObj(body);
  142. TaskSubmitResult result = new TaskSubmitResult(
  143. jsonResponse.getStr("task_id"),
  144. jsonResponse.getBool("success")
  145. );
  146. log.info("任务提交结果: {}", result);
  147. return result;
  148. } catch (Exception e) {
  149. log.error("提交任务失败: {}", e.getMessage(), e);
  150. throw new RuntimeException("提交任务失败: " + e.getMessage(), e);
  151. }
  152. }
  153. public TaskResult getTaskResult(String taskId) {
  154. //特殊处理year活动代充 走通流程
  155. if (StrUtil.isNotBlank(taskId) && taskId.contains("yhrgzc:")) {
  156. GroupsRelation relation = relationMapper.selectOne(Wrappers.lambdaQuery(GroupsRelation.class)
  157. .eq(GroupsRelation::getGptTaskId, taskId)
  158. .last("limit 1"));
  159. if (relation == null) {
  160. return null;
  161. }
  162. TaskResult taskResult = new TaskResult();
  163. taskResult.setStatus("recharging");
  164. if (relation.getRechargeStatus() == GroupsRelation.RechargeStatus.complete) {
  165. taskResult.setStatus("completed");
  166. }
  167. if (relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharge_error) {
  168. taskResult.setStatus("failed");
  169. }
  170. return taskResult;
  171. }
  172. log.info("开始查询任务结果: {}", taskId);
  173. try {
  174. GroupsRelation relation = relationMapper.selectOne(Wrappers.lambdaQuery(GroupsRelation.class)
  175. .eq(GroupsRelation::getGptTaskId, taskId)
  176. .last("limit 1"));
  177. if (relation == null) {
  178. return null;
  179. }
  180. String cardKey = relation.getCardKey();
  181. GptRechargeCardKey gptRechargeCardKey = gptRechargeCardKeyMapper.selectOne(Wrappers.lambdaQuery(GptRechargeCardKey.class).eq(GptRechargeCardKey::getCardKey, cardKey).last("limit 1"));
  182. if (gptRechargeCardKey.getChannelId() == Constant.AF_DIAN_CHANNEL_ID) {
  183. return afDianRechargeService.getTaskResult(relation);
  184. }
  185. if (gptRechargeCardKey.getChannelId()==Constant.CHONG_AF_DIAN_CHANNEL_ID){
  186. return chongAfDianRechargeService.getTaskResult(relation);
  187. }
  188. if (gptRechargeCardKey.getChannelId() == Constant.SC_CHANNEL_ID) {
  189. return sancRechargeService.getTaskResult(relation);
  190. }
  191. String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
  192. String url = BASE_URL + "/tasks/" + taskId;
  193. HttpResponse response = createBaseRequest(url)
  194. .execute();
  195. String body = response.body();
  196. log.info("任务查询响应: {}", body);
  197. JSONObject jsonResponse = JSONUtil.parseObj(body);
  198. TaskResult result = new TaskResult(
  199. jsonResponse.getStr("status"),
  200. jsonResponse.getStr("result"),
  201. jsonResponse.getStr("error")
  202. );
  203. log.info("任务查询结果: {}", result);
  204. String status = result.getStatus();
  205. String resultResult = Optional.ofNullable(result.getError()).orElse(result.getResult());
  206. //更新代充状态
  207. resetGroupsRelationRechargeStatus(relation, status, resultResult);
  208. return result;
  209. } catch (Exception e) {
  210. log.error("获取任务结果失败: {}", e.getMessage(), e);
  211. throw new RuntimeException("获取任务结果失败: " + e.getMessage(), e);
  212. }
  213. }
  214. @Override
  215. @Transactional(rollbackFor = Throwable.class)
  216. public String confirmRecharge(GroupsRelationRechargeReq req) {
  217. Long relationId = req.getRelationId();
  218. Long userId = req.getUserId();
  219. List<Long> userIdList = userBindRelationService.getRelationUserIdList(userId, null);
  220. GroupsRelation relation = relationMapper.selectOne(Wrappers.lambdaQuery(GroupsRelation.class)
  221. .in(GroupsRelation::getUserId, userIdList)
  222. .eq(GroupsRelation::getId, relationId)
  223. .eq(GroupsRelation::getStatus, GroupsRelation.Status.validity));
  224. if (relation == null) {
  225. throw BusinessRuntimeException.getInstance("车票不存在");
  226. }
  227. if (relation.getRechargeStatus() != null && relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharge_error) {
  228. throw BusinessRuntimeException.getInstance("充值失败,清联系客服");
  229. }
  230. String userAccount = relation.getAccount();
  231. GroupsTrips groupsTrips = groupsMapper.selectById(relation.getGroupsId());
  232. GoodsDonSku sku = skuMapper.selectById(groupsTrips.getSkuId());
  233. // //GPT 代充
  234. // if (sku.getGoodsId() != Constant.GPT_RECHARGE_GOODS_ID) {
  235. // throw BusinessRuntimeException.getInstance("车票类型错误");
  236. // }
  237. Integer rechargeNum = relation.getRechargeNum();
  238. Integer rechargeRemainNum = relation.getRechargeRemainNum();
  239. if (rechargeRemainNum == null || rechargeRemainNum <= 0) {
  240. throw BusinessRuntimeException.getInstance("剩余可代充次数为0");
  241. }
  242. //近一个小时成功
  243. Integer successHour = gptUserRechargeNumRecordMapper.selectCount(Wrappers.lambdaQuery(GptUserRechargeNumRecord.class)
  244. .eq(GptUserRechargeNumRecord::getRelationId, relationId)
  245. .in(GptUserRechargeNumRecord::getUserId, userIdList)
  246. .gt(GptUserRechargeNumRecord::getCreatedTime, DateUtil.offsetHour(DateTime.now(), -1)));
  247. if (successHour > 0) {
  248. throw BusinessRuntimeException.getInstance("近一个小时只可充值一次");
  249. }
  250. if (relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharging) {
  251. throw BusinessRuntimeException.getInstance("确认充值中");
  252. }
  253. if (relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharge_error) {
  254. throw BusinessRuntimeException.getInstance("充值失败,请联系客服开通");
  255. }
  256. String key = RedisService.key.GPT_RECHARGE_KEY.getName() + relationId;
  257. boolean b = redisService.setNx(key, relationId, RedisService.key.GPT_RECHARGE_KEY.getTimeout());
  258. if (!b) {
  259. throw BusinessRuntimeException.getInstance("确认充值中");
  260. }
  261. Integer subRechargeNum = 1;
  262. Integer specificChannel = null;
  263. //活动gpt代充类型
  264. if (sku.getRechargeType() == 2) {
  265. specificChannel = 8;
  266. subRechargeNum = 12;
  267. }
  268. //指定了代充渠道 直接用这个渠道 其他不能用
  269. if (sku.getRechargeChannelId() != null) {
  270. specificChannel = sku.getRechargeChannelId();
  271. }
  272. try {
  273. int updateRemainNum = relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
  274. .set(GroupsRelation::getRechargeRemainNum, rechargeRemainNum - subRechargeNum)
  275. .eq(GroupsRelation::getId, relationId)
  276. .eq(GroupsRelation::getUserId, relation.getUserId())
  277. .eq(GroupsRelation::getRechargeRemainNum, rechargeRemainNum));
  278. if (updateRemainNum == 0) {
  279. throw BusinessRuntimeException.getInstance("代充异常,请重试");
  280. }
  281. OrderDon orderDon = orderDonMapper.selectOne(Wrappers.lambdaQuery(OrderDon.class)
  282. .in(OrderDon::getUserId, userIdList)
  283. .eq(OrderDon::getRelationId, relationId)
  284. .notIn(OrderDon::getStatus, Constant.noOrderAllStatus)
  285. .orderByDesc(OrderDon::getId)
  286. .last("limit 1"));
  287. if (orderDon == null) {
  288. throw BusinessRuntimeException.getInstance("代充订单不存在或已退款");
  289. }
  290. //入库token json
  291. String saveTokenJson = req.getAccessToken();
  292. String accessToken;
  293. try {
  294. GptCardTokenDto gptCardTokenDto = Jsons.parseObject(saveTokenJson, GptCardTokenDto.class);
  295. accessToken = gptCardTokenDto.getAccessToken();
  296. } catch (Exception e) {
  297. accessToken = saveTokenJson;
  298. }
  299. //验证用户token是否正确
  300. TokenParseResult tokenParseResult = parseToken(accessToken);
  301. if (!tokenParseResult.getSuccess()) {
  302. throw BusinessRuntimeException.getInstance("您输入的凭证有误,请重新输入");
  303. }
  304. String rechargeAccount = tokenParseResult.getMessage();
  305. //月付校验前后账号是否一致
  306. if (sku.getMonths() == 1 && !StrUtil.equals(rechargeAccount, userAccount)) {
  307. throw BusinessRuntimeException.getInstance("请填入绑定账号的token");
  308. }
  309. //记录代充账号 token
  310. relation.setGptToken(saveTokenJson);
  311. relation.setAccount(rechargeAccount);
  312. //获取GPT代充key
  313. //代充卡密类型
  314. Integer gptRechargeCardKeyType = 0;
  315. //若是季、年付 且首月代充
  316. if (sku.getMonths() > 1 && rechargeNum == rechargeRemainNum) {
  317. gptRechargeCardKeyType = 1;
  318. }
  319. GptRechargeCardKey gptRechargeCardKey = gptRechargeCardKeyMapper.selectOne(Wrappers.lambdaQuery(GptRechargeCardKey.class)
  320. .eq(GptRechargeCardKey::getType, gptRechargeCardKeyType)
  321. .eq(GptRechargeCardKey::getStatus, Boolean.FALSE)
  322. .eq(specificChannel != null, GptRechargeCardKey::getChannelId, specificChannel)
  323. .notIn(specificChannel == null, GptRechargeCardKey::getChannelId, List.of(Constant.YEAR_AF_DIAN_CHANNEL_ID, Constant.HIBOY_PRO_ONE_CHANNEL_ID, Constant.HIBOY_PRO_TWO_CHANNEL_ID))
  324. .last("order by channel_id desc, rand() limit 1"));
  325. //季、年付卡密不存在 从月付代充卡密中获取
  326. if (gptRechargeCardKey == null && gptRechargeCardKeyType == 1) {
  327. gptRechargeCardKey = gptRechargeCardKeyMapper.selectOne(Wrappers.lambdaQuery(GptRechargeCardKey.class)
  328. .eq(GptRechargeCardKey::getType, 0)
  329. .eq(GptRechargeCardKey::getStatus, Boolean.FALSE)
  330. .eq(specificChannel != null, GptRechargeCardKey::getChannelId, specificChannel)
  331. .notIn(specificChannel == null, GptRechargeCardKey::getChannelId, List.of(Constant.YEAR_AF_DIAN_CHANNEL_ID, Constant.HIBOY_PRO_ONE_CHANNEL_ID, Constant.HIBOY_PRO_TWO_CHANNEL_ID))
  332. .last("order by channel_id desc, rand() limit 1"));
  333. }
  334. if (gptRechargeCardKey == null) {
  335. setRechargeRelationStatusAutoError(relation);
  336. throw BusinessRuntimeException.getInstance("充值失败,请联系客服开通");
  337. }
  338. int update = gptRechargeCardKeyMapper.update(null, Wrappers.lambdaUpdate(GptRechargeCardKey.class)
  339. .set(GptRechargeCardKey::getStatus, Boolean.TRUE)
  340. .set(GptRechargeCardKey::getOrderId, orderDon.getId())
  341. .set(GptRechargeCardKey::getUpdateTime, DateTime.now())
  342. .set(GptRechargeCardKey::getOrderNo, orderDon.getOrderNo())
  343. .eq(GptRechargeCardKey::getId, gptRechargeCardKey.getId())
  344. .eq(GptRechargeCardKey::getStatus, Boolean.FALSE));
  345. if (update == 0) {
  346. setRechargeRelationStatusAutoError(relation);
  347. throw BusinessRuntimeException.getInstance("充值失败,请联系客服开通");
  348. }
  349. String gptCardKey = gptRechargeCardKey.getCardKey();
  350. Long channelId = gptRechargeCardKey.getChannelId();
  351. //AF dian
  352. if (channelId == Constant.AF_DIAN_CHANNEL_ID) {
  353. return afDianRechargeService.confirmRecharge(relation, gptCardKey, null);
  354. }
  355. //chong.afdian
  356. if (channelId == Constant.CHONG_AF_DIAN_CHANNEL_ID) {
  357. return chongAfDianRechargeService.confirmRecharge(relation, gptCardKey, null);
  358. }
  359. //year.afdian
  360. if (channelId == Constant.YEAR_AF_DIAN_CHANNEL_ID) {
  361. return yearAfDianRechargeService.confirmRecharge(relation, gptCardKey, null);
  362. }
  363. //go.afdian
  364. if (channelId == Constant.GO_AF_DIAN_CHANNEL_ID) {
  365. return goAfDianRechargeService.confirmRecharge(relation, gptCardKey, null);
  366. }
  367. //hiboy
  368. if (channelId == Constant.HIBOY_CHANNEL_ID || channelId == Constant.HIBOY_PRO_ONE_CHANNEL_ID || channelId == Constant.HIBOY_PRO_TWO_CHANNEL_ID) {
  369. return hiBoyRechargeService.confirmRecharge(relation, gptCardKey, null);
  370. }
  371. //plus.afdian
  372. if (channelId == Constant.PLUS_AFDIAN_CHANNEL_ID) {
  373. return plusAfDianRechargeService.confirmRecharge(relation, gptCardKey, null);
  374. }
  375. //三川
  376. if (channelId == Constant.SC_CHANNEL_ID) {
  377. return sancRechargeService.confirmRecharge(relation, gptCardKey, null);
  378. }
  379. //cdk
  380. if (channelId == Constant.CDK_CHANNEL_ID) {
  381. return cdkRechargeService.confirmRecharge(relation, gptCardKey, null);
  382. }
  383. //guofenzai
  384. if (channelId == Constant.GUOFENZAIXIAN_CHANNEL_ID) {
  385. return guofenzaixianRechargeService.confirmRecharge(relation, gptCardKey, null);
  386. }
  387. //校验卡密是否可用
  388. CardKeyValidationResult cardKeyValidationResult = validateCardKey(gptCardKey);
  389. if (!cardKeyValidationResult.getAvailable()) {
  390. setRechargeRelationStatusAutoError(relation);
  391. throw BusinessRuntimeException.getInstance("充值失败,请联系客服开通");
  392. }
  393. //提交任务
  394. TaskSubmitResult result = submitTask(gptCardKey, accessToken, "auth0");
  395. if (!result.getSuccess()) {
  396. setRechargeRelationStatusAutoError(relation);
  397. throw BusinessRuntimeException.getInstance("充值失败,请联系客服开通");
  398. }
  399. String taskId = result.getTaskId();
  400. relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
  401. //记录代充账号
  402. .set(GroupsRelation::getAccount, rechargeAccount)
  403. .set(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharging)
  404. .set(GroupsRelation::getCardKey, gptCardKey)
  405. .set(GroupsRelation::getGptToken, saveTokenJson)
  406. .set(GroupsRelation::getGptTaskId, taskId)
  407. .set(GroupsRelation::getOperator, null)
  408. .eq(GroupsRelation::getId, relationId));
  409. return taskId;
  410. } finally {
  411. if (redisService.hasKey(key)) {
  412. redisService.del(key);
  413. }
  414. }
  415. }
  416. /**
  417. * 设置代充自动充值异常
  418. */
  419. public void setRechargeRelationStatusAutoError(GroupsRelation relation) {
  420. TASK_EXECUTOR.execute(()->{
  421. relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
  422. .set(GroupsRelation::getAccount, relation.getAccount())
  423. .set(GroupsRelation::getGptToken, relation.getGptToken())
  424. .set(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharge_error)
  425. .eq(GroupsRelation::getId, relation.getId()));
  426. });
  427. }
  428. /**
  429. * 更新代充状态
  430. */
  431. private void resetGroupsRelationRechargeStatus(GroupsRelation relation, String status, String result) {
  432. if ("completed".equals(status) || "failed".equals(status) || "unknown".equals(status)) {
  433. String taskId = relation.getGptTaskId();
  434. if (relation != null && relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharging) {
  435. Integer rechargeRemainNum = relation.getRechargeRemainNum();
  436. GroupsRelation.RechargeStatus rechargeStatus = "completed".equals(status) ? GroupsRelation.RechargeStatus.complete : GroupsRelation.RechargeStatus.recharge_error;
  437. String cardKey = relation.getCardKey();
  438. GptRechargeCardKey gptRechargeCardKey = gptRechargeCardKeyMapper.selectOne(Wrappers.lambdaQuery(GptRechargeCardKey.class)
  439. .eq(GptRechargeCardKey::getCardKey, cardKey)
  440. .last("limit 1"));
  441. if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
  442. if (gptRechargeCardKey != null) {
  443. gptRechargeCardKeyMapper.update(null, Wrappers.lambdaUpdate(GptRechargeCardKey.class)
  444. .set(GptRechargeCardKey::getStatus, Boolean.FALSE)
  445. .set(GptRechargeCardKey::getOrderId, null)
  446. .set(GptRechargeCardKey::getOrderNo, null)
  447. .eq(GptRechargeCardKey::getId, gptRechargeCardKey.getId())
  448. .eq(GptRechargeCardKey::getStatus, Boolean.TRUE));
  449. }
  450. }
  451. if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
  452. if (!redisService.setNx(RedisService.key.GPT_RECHARGE_RECOVER_KEY.getName() + taskId, taskId, RedisService.key.GPT_RECHARGE_RECOVER_KEY.getTimeout())) {
  453. return;
  454. }
  455. }
  456. Date startTime = relation.getStartTime();
  457. DateTime expiryTime = null;
  458. if (rechargeStatus == GroupsRelation.RechargeStatus.complete && startTime == null) {
  459. GroupsTrips groupsTrips = groupsMapper.selectById(relation.getGroupsId());
  460. if (groupsTrips != null) {
  461. startTime = DateTime.now();
  462. GoodsDonSku sku = skuMapper.selectById(groupsTrips.getSkuId());
  463. if (sku.getDays() != null && sku.getDays() > 0) {
  464. expiryTime = DateUtil.offsetDay(startTime, sku.getDays());
  465. } else {
  466. expiryTime = DateUtil.offsetMonth(startTime, sku.getMonths());
  467. }
  468. }
  469. }
  470. String rechargeErrorInfo = null;
  471. if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
  472. rechargeErrorInfo = result;
  473. }
  474. relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
  475. .set(rechargeStatus == GroupsRelation.RechargeStatus.recharge_error, GroupsRelation::getRechargeRemainNum, rechargeRemainNum + 1)
  476. .set(GroupsRelation::getRechargeStatus, rechargeStatus)
  477. .set(expiryTime != null, GroupsRelation::getStartTime, startTime)
  478. .set(expiryTime != null, GroupsRelation::getExpiryTime, expiryTime)
  479. .set(StrUtil.isNotEmpty(rechargeErrorInfo), GroupsRelation::getRechargeErrorInfo, rechargeErrorInfo)
  480. .eq(GroupsRelation::getId, relation.getId())
  481. .le(GroupsRelation::getRechargeRemainNum, relation.getRechargeNum())
  482. .eq(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharging));
  483. //记录代充记录
  484. if (rechargeStatus == GroupsRelation.RechargeStatus.complete) {
  485. String remark = "自动代充";
  486. if (StrUtil.isNotBlank(relation.getOperator())) {
  487. remark = "后台自动代充";
  488. }
  489. gptUserRechargeRecordService.recordGptUserRechargeNum(relation.getCardKey(), relation.getUserId(), relation.getId(), relation.getAccount(), -1, relation.getOperator(), remark);
  490. //修改对应订单状态
  491. if (gptRechargeCardKey != null) {
  492. Long orderId = gptRechargeCardKey.getOrderId();
  493. OrderDon orderDon = orderDonMapper.selectById(orderId);
  494. if (orderDon != null && orderDon.getStatus() == OrderDon.Status.hasPayment) {
  495. orderDon.setStatus(OrderDon.Status.complete);
  496. orderDonMapper.updateById(orderDon);
  497. }
  498. }
  499. }
  500. }
  501. }
  502. }
  503. }