zoujiajian 2 gadi atpakaļ
vecāks
revīzija
f176e81afb

+ 31 - 16
netflix-service/src/main/java/com/cyksj/service/relation/impl/GroupRelationFrontServiceImpl.java

@@ -34,6 +34,7 @@ import com.cyksj.model.views.*;
 import com.cyksj.redis.RedisService;
 import com.cyksj.service.chatgpt.ChatGptAccountService;
 import com.cyksj.service.chatgpt.ChatGptAuthService;
+import com.cyksj.service.claude.ClaudeService;
 import com.cyksj.service.groups.GroupsFuncService;
 import com.cyksj.service.mail.CuiQiuMailService;
 import com.cyksj.service.mail.MailService;
@@ -142,6 +143,8 @@ public class GroupRelationFrontServiceImpl implements GroupRelationFrontService
 
 	private final RenewUpgradePackageMapper renewUpgradePackageMapper;
 
+	private final ClaudeService claudeService;
+
 	@Override
 	public SearchResult<RenewalView> getMyTicket(long userId, Boolean isLoginPopularize, String customId, String account) {
 		User u = userMapper.selectById(userId);
@@ -281,23 +284,35 @@ public class GroupRelationFrontServiceImpl implements GroupRelationFrontService
 				}
 			}
 			if (ticket.getGoodsId() == 18 && ticket.getIsCar() && ticket.getExpiryTime() != null) {
-				//gpt用户镜像
-				ChatGptUserView userInfo = chatGptAccountService.getUserInfo(userId, ticket.getRelationId());
-				ticket.setGptUserLimitNum(userInfo.getLimitNum());
-				ticket.setGptUserLimitTime(userInfo.getLimitTime());
-				ticket.setUse(userInfo.getUse());
-				Integer count = upgradePackageMapper.selectCount(Wrappers.lambdaQuery(UpgradePackage.class).apply("JSON_CONTAINS(sku_ids,'\"" + ticket.getSkuId() + "\"')")
-						.gt(UpgradePackage::getNum, userInfo.getLimitNum())
-						.eq(UpgradePackage::getStatus, Boolean.TRUE)
-						.eq(UpgradePackage::getGptType, 0)
-						.eq(UpgradePackage::getDeleted, Boolean.TRUE));
-				ticket.setUpgrade(count > 0 ? true : false);
-				ticket.setPackageId(userInfo.getPackageId());
-				if (userInfo.getPackageId() != 0) {
-					UpgradePackage upgradePackage = upgradePackageMapper.selectById(userInfo.getPackageId());
-					ticket.setPackagePrice(upgradePackage.getPrice());
+				try {
+					//gpt用户镜像
+					ChatGptUserView userInfo = chatGptAccountService.getUserInfo(userId, ticket.getRelationId());
+					ticket.setGptUserLimitNum(userInfo.getLimitNum());
+					ticket.setGptUserLimitTime(userInfo.getLimitTime());
+					ticket.setUse(userInfo.getUse());
+					Integer count = upgradePackageMapper.selectCount(Wrappers.lambdaQuery(UpgradePackage.class).apply("JSON_CONTAINS(sku_ids,'\"" + ticket.getSkuId() + "\"')")
+							.gt(UpgradePackage::getNum, userInfo.getLimitNum())
+							.eq(UpgradePackage::getStatus, Boolean.TRUE)
+							.eq(UpgradePackage::getGptType, 0)
+							.eq(UpgradePackage::getDeleted, Boolean.TRUE));
+					ticket.setUpgrade(count > 0 ? true : false);
+					ticket.setPackageId(userInfo.getPackageId());
+					if (userInfo.getPackageId() != 0) {
+						UpgradePackage upgradePackage = upgradePackageMapper.selectById(userInfo.getPackageId());
+						ticket.setPackagePrice(upgradePackage.getPrice());
+					}
+					ticket.setIsRenewPackage(userInfo.getIsRenewPackage());
+				} catch (Exception e) {
+				}
+			}
+
+			if (ticket.getGoodsId() == 75l && ticket.getExpiryTime() != null) {
+				try {
+					ClaudeUser claudeUser = claudeService.getClaudeUser(userId, ticket.getRelationId());
+					ticket.setGptUserLimitNum(claudeUser.getLimitNum());
+					ticket.setGptUserLimitTime(claudeUser.getLimitTime());
+				} catch (Exception e) {
 				}
-				ticket.setIsRenewPackage(userInfo.getIsRenewPackage());
 			}
 			List<GroupsRelationUserView> relationUserViews = groupsRelationMapper.getUserRelationUserViews(ticket.getGroupId(), ticket.getRelationId(), ticket.getUserId(), List.of(GroupsRelation.Status.validity, GroupsRelation.Status.outside));
 			ticket.setRelationViews(relationUserViews);