zoujiajian 2 жил өмнө
parent
commit
40e686fa89

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

@@ -345,4 +345,9 @@ public interface Constant {
 
 	//英文
 	String EN_US = "en_US";
+
+	/**
+	 * 镜像GPT cookie
+	 */
+	String MIRROR_GPT_COOKIE = "nf_ticket_info";
 }

+ 26 - 13
netflix-web/src/main/java/com/cyksj/web/controller/mirror/MirrorController.java

@@ -7,11 +7,10 @@ import com.cyksj.common.exception.BusinessRuntimeException;
 import com.cyksj.common.task.GlobalThreadPoolTaskExecutor;
 import com.cyksj.dto.Result;
 import com.cyksj.enums.GatewayResponse;
+import com.cyksj.mapper.GroupsMapper;
+import com.cyksj.mapper.GroupsRelationMapper;
 import com.cyksj.mapper.renew.RenewUpgradePackageMapper;
-import com.cyksj.model.entity.ChatgptSession;
-import com.cyksj.model.entity.ChatgptUser;
-import com.cyksj.model.entity.MidjourneyUser;
-import com.cyksj.model.entity.RenewUpgradePackage;
+import com.cyksj.model.entity.*;
 import com.cyksj.model.request.gpt.ConversationRequest;
 import com.cyksj.model.response.ConversationLimitResponse;
 import com.cyksj.model.views.ChatGptUserConversationRecordHistoryView;
@@ -75,6 +74,10 @@ public class MirrorController {
 
     private final RenewUpgradePackageMapper renewUpgradePackageMapper;
 
+    private final GroupsRelationMapper relationMapper;
+
+    private final GroupsMapper groupsMapper;
+
 
     /**
      * GPT车票跳转登录
@@ -364,14 +367,24 @@ public class MirrorController {
      * gpt续费广告弹窗
      */
     @GetMapping("/get/gpt/renew/discount")
-    public Result<RenewUpgradePackage> getGptRenewDiscount() {
-        RenewUpgradePackage renewUpgradePackage = renewUpgradePackageMapper.selectOne(Wrappers.lambdaQuery(RenewUpgradePackage.class)
-                .eq(RenewUpgradePackage::getGoodsId, 18)
-                .eq(RenewUpgradePackage::getStatus, 1)
-                .eq(RenewUpgradePackage::getDeleted, 1)
-                .orderByAsc(RenewUpgradePackage::getDiscount)
-                .last("limit 1")
-                .select(RenewUpgradePackage::getId, RenewUpgradePackage::getDiscount));
-        return GatewayResponse.SUCCESS.newBuilder().toResult(renewUpgradePackage);
+    public Result<RenewUpgradePackage> getGptRenewDiscount(Long relationId) {
+        if (relationId != null) {
+            GroupsRelation relation = relationMapper.selectById(relationId);
+            if (relation != null) {
+                GroupsTrips groupsTrips = groupsMapper.selectById(relation.getGroupsId());
+                if (groupsTrips != null) {
+                    RenewUpgradePackage renewUpgradePackage = renewUpgradePackageMapper.selectOne(Wrappers.lambdaQuery(RenewUpgradePackage.class)
+                            .eq(RenewUpgradePackage::getGoodsId, 18)
+                            .eq(RenewUpgradePackage::getStatus, 1)
+                            .eq(RenewUpgradePackage::getDeleted, 1)
+                            .eq(RenewUpgradePackage::getSkuId, groupsTrips.getSkuId())
+                            .orderByAsc(RenewUpgradePackage::getDiscount)
+                            .last("limit 1")
+                            .select(RenewUpgradePackage::getId, RenewUpgradePackage::getDiscount));
+                    return GatewayResponse.SUCCESS.newBuilder().toResult(renewUpgradePackage);
+                }
+            }
+        }
+        return GatewayResponse.SUCCESS.newBuilder().toResult();
     }
 }