Merge branch 'go.afdian'

netflix-common/src/main/java/com/cyksj/common/constant/Constant.java

@@ -813,6 +813,11 @@ public interface Constant {
      */
     Long YEAR_AF_DIAN_CHANNEL_ID = 8L;
 
+    /**
+     * go.afdian
+     */
+    Long GO_AF_DIAN_CHANNEL_ID = 9L;
+
 	/**
 	 * Adobe 全家桶
 	 */

netflix-dao/src/main/java/com/cyksj/model/response/recharge/GoAfDianTaskResp.java

@@ -0,0 +1,15 @@
+package com.cyksj.model.response.recharge;
+
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class GoAfDianTaskResp {
+    private Integer code;
+
+    private String message;
+
+    private Object data;
+}

netflix-service/src/main/java/com/cyksj/server/recharge/impl/GptProxyRechargeServiceImpl.java

@@ -23,10 +23,7 @@ import com.cyksj.server.recharge.dto.TaskResult;
 import com.cyksj.server.recharge.dto.TaskSubmitResult;
 import com.cyksj.server.recharge.dto.TokenParseResult;
 import com.cyksj.service.chatgpt.GptUserRechargeRecordService;
-import com.cyksj.service.recharge.AfDianRechargeService;
-import com.cyksj.service.recharge.ChongAfDianRechargeService;
-import com.cyksj.service.recharge.GptRechargeCardKeyChannelService;
-import com.cyksj.service.recharge.YearAfDianRechargeService;
+import com.cyksj.service.recharge.*;
 import com.cyksj.service.user.UserBindRelationService;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -70,6 +67,8 @@ public class GptProxyRechargeServiceImpl implements GptProxyRechargeService {
 
     private final YearAfDianRechargeService yearAfDianRechargeService;
 
+    private final GoAfDianRechargeService goAfDianRechargeService;
+
 	private static final GlobalThreadPoolTaskExecutor TASK_EXECUTOR = GlobalThreadPoolTaskExecutor.getInstance();
 
 	private HttpRequest createBaseRequest(String url) {
@@ -367,6 +366,10 @@ public class GptProxyRechargeServiceImpl implements GptProxyRechargeService {
 			if (channelId == Constant.YEAR_AF_DIAN_CHANNEL_ID) {
 				return yearAfDianRechargeService.confirmRecharge(relation, gptCardKey, null);
 			}
+            //go.afdian
+            if (channelId == Constant.GO_AF_DIAN_CHANNEL_ID) {
+                return goAfDianRechargeService.confirmRecharge(relation, gptCardKey, null);
+            }
 			//校验卡密是否可用
 			CardKeyValidationResult cardKeyValidationResult = validateCardKey(gptCardKey);
 			if (!cardKeyValidationResult.getAvailable()) {

netflix-service/src/main/java/com/cyksj/service/mange/gpt/impl/GptRechargeServiceImpl.java

@@ -25,10 +25,7 @@ import com.cyksj.server.recharge.dto.TaskSubmitResult;
 import com.cyksj.server.recharge.dto.TokenParseResult;
 import com.cyksj.service.chatgpt.GptUserRechargeRecordService;
 import com.cyksj.service.mange.gpt.GptRechargeService;
-import com.cyksj.service.recharge.AfDianRechargeService;
-import com.cyksj.service.recharge.ChongAfDianRechargeService;
-import com.cyksj.service.recharge.GptRechargeCardKeyChannelService;
-import com.cyksj.service.recharge.YearAfDianRechargeService;
+import com.cyksj.service.recharge.*;
 import com.cyksj.service.user.UserBindRelationService;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -80,6 +77,8 @@ public class GptRechargeServiceImpl implements GptRechargeService {
 
     private final YearAfDianRechargeService yearAfDianRechargeService;
 
+    private final GoAfDianRechargeService goAfDianRechargeService;
+
 	private HttpRequest createBaseRequest(String url) {
 		return HttpRequest.get(url)
 				.header("accept", "application/json, text/plain, */*")
@@ -247,6 +246,12 @@ public class GptRechargeServiceImpl implements GptRechargeService {
                 relation.setCardKey(gptCardKey);
                 return yearAfDianRechargeService.confirmRecharge(relation, gptCardKey, cmsUser);
             }
+            if (gptRechargeCardKey.getChannelId() == Constant.GO_AF_DIAN_CHANNEL_ID) {
+                req.setCmsUser(cmsUser);
+                relation.setCardKey(gptCardKey);
+                return goAfDianRechargeService.confirmRecharge(relation, gptCardKey, cmsUser);
+            }
+
 			//校验卡密是否可用
 			CardKeyValidationResult cardKeyValidationResult = validateCardKey(gptCardKey);
 			if (!cardKeyValidationResult.getAvailable()) {

netflix-service/src/main/java/com/cyksj/service/recharge/GoAfDianRechargeService.java

@@ -0,0 +1,8 @@
+package com.cyksj.service.recharge;
+
+import com.cyksj.model.entity.CmsUser;
+import com.cyksj.model.entity.GroupsRelation;
+
+public interface GoAfDianRechargeService {
+    String confirmRecharge(GroupsRelation relation, String cardKey, CmsUser user);
+}

netflix-service/src/main/java/com/cyksj/service/recharge/impl/GoAfDianRechargeServiceImpl.java

@@ -0,0 +1,197 @@
+package com.cyksj.service.recharge.impl;
+
+import cn.hutool.core.date.DateTime;
+import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.RandomUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import cn.hutool.json.JSONObject;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.cyksj.common.exception.BusinessRuntimeException;
+import com.cyksj.common.task.GlobalThreadPoolTaskExecutor;
+import com.cyksj.common.util.Jsons;
+import com.cyksj.mapper.*;
+import com.cyksj.model.entity.*;
+import com.cyksj.model.response.recharge.GoAfDianTaskResp;
+import com.cyksj.redis.RedisService;
+import com.cyksj.service.chatgpt.GptUserRechargeRecordService;
+import com.cyksj.service.recharge.GoAfDianRechargeService;
+import com.cyksj.service.recharge.GptRechargeCardKeyChannelService;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+import java.util.Date;
+
+@Service
+@RequiredArgsConstructor
+@Slf4j
+public class GoAfDianRechargeServiceImpl implements GoAfDianRechargeService {
+
+    private final GptRechargeCardKeyChannelService gptRechargeCardKeyChannelService;
+
+    private final GroupsRelationMapper relationMapper;
+
+    private final GroupsMapper groupsMapper;
+
+    private final GoodsDonSkuMapper skuMapper;
+
+    private final RedisService redisService;
+
+    private final OrderDonMapper orderDonMapper;
+
+    private final GptRechargeCardKeyMapper gptRechargeCardKeyMapper;
+
+    private final GptUserRechargeRecordService gptUserRechargeRecordService;
+
+    private static final GlobalThreadPoolTaskExecutor TASK_EXECUTOR = GlobalThreadPoolTaskExecutor.getInstance();
+
+    @Override
+    public String confirmRecharge(GroupsRelation relation, String gptCardKey, CmsUser cmsUser) {
+        Long relationId = relation.getId();
+        String accessToken = relation.getGptToken();
+        String rechargeAccount = relation.getAccount();
+        //提交任务返回状态
+        Boolean success;
+        try {
+            success = submitTask(gptCardKey, accessToken);
+        } catch (Exception e) {
+            setRechargeRelationStatusAutoError(relation);
+            if (cmsUser == null) {
+                throw BusinessRuntimeException.getInstance("充值失败，请联系客服开通");
+            }
+            throw BusinessRuntimeException.getInstance("充值失败，{0}", e.getMessage());
+        }
+        String operator = null;
+        if (cmsUser != null) {
+            operator = cmsUser.getNickname();
+        }
+        String password = relation.getPassword();
+        String taskId = "yhrgzc:" + RandomUtil.randomNumbers(20);
+        relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
+                //记录代充账号
+                .set(GroupsRelation::getAccount, rechargeAccount)
+                //记录操作人员
+                .set(GroupsRelation::getOperator, operator)
+                //密码
+                .set(StrUtil.isNotBlank(password), GroupsRelation::getPassword, password)
+                .set(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharging)
+                .set(GroupsRelation::getCardKey, gptCardKey)
+                .set(GroupsRelation::getGptToken, accessToken)
+                .set(GroupsRelation::getGptTaskId, taskId)
+                .eq(GroupsRelation::getId, relationId));
+        relation.setRechargeStatus(GroupsRelation.RechargeStatus.recharging);
+        relation.setCardKey(gptCardKey);
+        resetGroupsRelationRechargeStatus(relation, success, rechargeAccount);
+        return taskId;
+    }
+
+    private Boolean submitTask(String cardKey, String accessToken) {
+        log.info("开始提交year.afdian任务 - 卡密: {}", cardKey);
+        try {
+            String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
+            String url = BASE_URL + "/charge";
+
+            JSONObject requestBody = new JSONObject();
+            requestBody.putOpt("cdk", cardKey);
+            requestBody.putOpt("account", accessToken);
+
+            HttpResponse response = HttpRequest.post(url)
+                    .setConnectionTimeout(10000)
+                    .body(requestBody.toString())
+                    .execute();
+            String body = response.body();
+            log.info("go.afdian任务提交响应: {}", body);
+
+            GoAfDianTaskResp goAfDianTaskResp = Jsons.parseObject(body, GoAfDianTaskResp.class);
+            if ("充值成功".equals(goAfDianTaskResp.getMessage())) {
+                return true;
+            }
+            //失败
+            throw BusinessRuntimeException.getInstance("卡密：" + cardKey + goAfDianTaskResp.getMessage());
+        } catch (Exception e) {
+            log.error("提交任务失败: {}", e.getMessage(), e);
+            throw new RuntimeException("提交任务失败: " + e.getMessage(), e);
+        }
+    }
+
+    /**
+     * 设置代充自动充值异常
+     */
+    public void setRechargeRelationStatusAutoError(GroupsRelation relation) {
+        TASK_EXECUTOR.execute(()->{
+            relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
+                    .set(GroupsRelation::getAccount, relation.getAccount())
+                    .set(GroupsRelation::getGptToken, relation.getGptToken())
+                    .set(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharge_error)
+                    .eq(GroupsRelation::getId, relation.getId()));
+        });
+    }
+
+    /**
+     * 更新代充状态
+     */
+    private void resetGroupsRelationRechargeStatus(GroupsRelation relation, Boolean taskStatus, String result) {
+        GroupsRelation.RechargeStatus rechargeStatus;
+        if (taskStatus) {
+            rechargeStatus = GroupsRelation.RechargeStatus.complete;
+        } else {
+            rechargeStatus = GroupsRelation.RechargeStatus.recharge_error;
+        }
+        String taskId = relation.getGptTaskId();
+        if (relation != null && relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharging) {
+            Integer rechargeRemainNum = relation.getRechargeRemainNum();
+            String cardKey = relation.getCardKey();
+            GptRechargeCardKey gptRechargeCardKey = gptRechargeCardKeyMapper.selectOne(Wrappers.lambdaQuery(GptRechargeCardKey.class).eq(GptRechargeCardKey::getCardKey, cardKey).last("limit 1"));
+            if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
+                if (gptRechargeCardKey != null) {
+                    gptRechargeCardKeyMapper.update(null, Wrappers.lambdaUpdate(GptRechargeCardKey.class).set(GptRechargeCardKey::getStatus, Boolean.FALSE).set(GptRechargeCardKey::getOrderId, null).set(GptRechargeCardKey::getOrderNo, null).eq(GptRechargeCardKey::getId, gptRechargeCardKey.getId()).eq(GptRechargeCardKey::getStatus, Boolean.TRUE));
+                }
+            }
+            if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
+                if (!redisService.setNx(RedisService.key.GPT_RECHARGE_RECOVER_KEY.getName() + taskId, taskId, RedisService.key.GPT_RECHARGE_RECOVER_KEY.getTimeout())) {
+                    return;
+                }
+            }
+            Date startTime = relation.getStartTime();
+            DateTime expiryTime = null;
+            if (rechargeStatus == GroupsRelation.RechargeStatus.complete && startTime == null) {
+                GroupsTrips groupsTrips = groupsMapper.selectById(relation.getGroupsId());
+                if (groupsTrips != null) {
+                    startTime = DateTime.now();
+                    GoodsDonSku sku = skuMapper.selectById(groupsTrips.getSkuId());
+                    if (sku.getDays() != null && sku.getDays() > 0) {
+                        expiryTime = DateUtil.offsetDay(startTime, sku.getDays());
+                    } else {
+                        expiryTime = DateUtil.offsetMonth(startTime, sku.getMonths());
+                    }
+                }
+            }
+            Integer rechargeNum = 1;
+            String rechargeErrorInfo = null;
+            if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
+                rechargeErrorInfo = result;
+            }
+            relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class).set(rechargeStatus == GroupsRelation.RechargeStatus.recharge_error, GroupsRelation::getRechargeRemainNum, rechargeRemainNum + rechargeNum).set(GroupsRelation::getRechargeStatus, rechargeStatus).set(expiryTime != null, GroupsRelation::getStartTime, startTime).set(expiryTime != null, GroupsRelation::getExpiryTime, expiryTime).set(StrUtil.isNotEmpty(rechargeErrorInfo), GroupsRelation::getRechargeErrorInfo, rechargeErrorInfo).eq(GroupsRelation::getId, relation.getId()).le(GroupsRelation::getRechargeRemainNum, relation.getRechargeNum()).eq(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharging));
+            //记录代充记录
+            if (rechargeStatus == GroupsRelation.RechargeStatus.complete) {
+                String remark = "自动代充";
+                if (StrUtil.isNotBlank(relation.getOperator())) {
+                    remark = "后台自动代充";
+                }
+                gptUserRechargeRecordService.recordGptUserRechargeNum(relation.getUserId(), relation.getId(), relation.getAccount(), -rechargeNum, relation.getOperator(), remark);
+                //修改对应订单状态
+                if (gptRechargeCardKey != null) {
+                    Long orderId = gptRechargeCardKey.getOrderId();
+                    OrderDon orderDon = orderDonMapper.selectById(orderId);
+                    if (orderDon != null && orderDon.getStatus() == OrderDon.Status.hasPayment) {
+                        orderDon.setStatus(OrderDon.Status.complete);
+                        orderDonMapper.updateById(orderDon);
+                    }
+
+                }
+            }
+        }
+    }
+}