ChatGptAccountServiceImpl.java 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630
  1. package com.cyksj.service.chatgpt.impl;
  2. import cn.hutool.core.date.DateTime;
  3. import cn.hutool.core.date.DateUnit;
  4. import cn.hutool.core.date.DateUtil;
  5. import cn.hutool.core.lang.UUID;
  6. import cn.hutool.core.util.RandomUtil;
  7. import cn.hutool.core.util.StrUtil;
  8. import cn.hutool.http.HttpRequest;
  9. import cn.hutool.http.HttpResponse;
  10. import cn.hutool.http.HttpUtil;
  11. import cn.hutool.http.Method;
  12. import cn.hutool.json.JSONObject;
  13. import cn.hutool.json.JSONUtil;
  14. import com.baomidou.mybatisplus.core.toolkit.Wrappers;
  15. import com.cyksj.common.constant.Constant;
  16. import com.cyksj.common.exception.BusinessRuntimeException;
  17. import com.cyksj.common.util.StringUtil;
  18. import com.cyksj.mapper.*;
  19. import com.cyksj.mapper.station.CollegeStudentStationUseTicketRecordMapper;
  20. import com.cyksj.model.entity.*;
  21. import com.cyksj.model.views.ChatGptUserView;
  22. import com.cyksj.model.views.ChatgptCarInfoView;
  23. import com.cyksj.redis.RedisService;
  24. import com.cyksj.service.chatgpt.ChatGptAccountService;
  25. import com.cyksj.service.sys.SysConfigService;
  26. import com.cyksj.service.user.UserBindRelationService;
  27. import com.ejlchina.searcher.BeanSearcher;
  28. import com.ejlchina.searcher.SearchResult;
  29. import com.ejlchina.searcher.util.MapUtils;
  30. import lombok.RequiredArgsConstructor;
  31. import lombok.extern.slf4j.Slf4j;
  32. import org.apache.commons.lang3.StringUtils;
  33. import org.springframework.beans.factory.annotation.Value;
  34. import org.springframework.dao.DuplicateKeyException;
  35. import org.springframework.stereotype.Service;
  36. import javax.imageio.stream.FileImageOutputStream;
  37. import javax.servlet.http.HttpServletRequest;
  38. import java.math.BigDecimal;
  39. import java.math.RoundingMode;
  40. import java.nio.file.Files;
  41. import java.nio.file.Path;
  42. import java.time.LocalDateTime;
  43. import java.util.*;
  44. /**
  45. * @author chan
  46. * @date 2024/3/19 11:01
  47. */
  48. @Service
  49. @Slf4j
  50. @RequiredArgsConstructor
  51. public class ChatGptAccountServiceImpl implements ChatGptAccountService {
  52. @Value("${chatgpt.domain}")
  53. private String GPT_DOMAIN;
  54. @Value("${chatgpt.car.domain}")
  55. private String CAR_GPT_DOMAIN;
  56. private static final String GPT_PROXY = "https://chat-chan-87jztgkf257d.xyhelper.net";
  57. private final RedisService redisService;
  58. private final ChatgptUserMapper chatgptUserMapper;
  59. private final ChatgptSessionMapper chatgptSessionMapper;
  60. private final GroupsRelationMapper groupsRelationMapper;
  61. private final GroupsMapper groupsMapper;
  62. private final GoodsDonSkuMapper skuMapper;
  63. private final UserMapper userMapper;
  64. private final UserBindRelationService userBindRelationService;
  65. private final AccountMapper accountMapper;
  66. private final BeanSearcher beanSearcher;
  67. private final ChatgptUserTokenPrepareMapper chatgptUserTokenPrepareMapper;
  68. private final CollegeStudentStationUseTicketRecordMapper collegeStudentStationUseTicketRecordMapper;
  69. private final UpgradePackageMapper upgradePackageMapper;
  70. private final OrderDonMapper orderDonMapper;
  71. private final SysConfigService sysConfigService;
  72. private final HttpServletRequest request;
  73. @Override
  74. public String getGptSession(String account, String password) {
  75. String officialSession = "";
  76. try {
  77. Account at = new Account();
  78. at.setAccount(account);
  79. at.setPassword(password);
  80. JSONObject loginResult = getLoginResult(at);
  81. String str = loginResult.getStr("detail");
  82. if (StringUtils.isNotBlank(str)) {
  83. throw BusinessRuntimeException.getInstance("登录获取token错误 error:" + str);
  84. }
  85. officialSession = loginResult.toString();
  86. } catch (Exception e) {
  87. throw BusinessRuntimeException.getInstance("登录获取token错误 error:" + e.getMessage());
  88. }
  89. return officialSession;
  90. }
  91. /**
  92. * 获取登录结果
  93. *
  94. * @param account
  95. * @return
  96. * @throws Exception
  97. */
  98. private JSONObject getLoginResult(Account account) throws Exception {
  99. HttpRequest request = new HttpRequest(GPT_PROXY + "/applelogin");
  100. request.form("username", account.getAccount());
  101. request.form("password", account.getPassword());
  102. request.header("Content-Type", "application/x-www-form-urlencoded");
  103. HttpResponse execute = request.method(Method.POST).execute();
  104. String body = execute.body();
  105. log.info("登录返回token res:{}", body);
  106. return new JSONObject(body);
  107. }
  108. /**
  109. * 获取登录url
  110. *
  111. * @param userId
  112. * @param relationId
  113. * @return
  114. */
  115. @Override
  116. public String getLoginUrl(Long userId, Long relationId) {
  117. GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
  118. GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
  119. GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
  120. log.info("用户:{},所在车次:{},座位id:{},在{}获取跳转GPT镜像的登录url", userId, groupsTrips.getId(), relationId, DateTime.now());
  121. if (goodsDonSku != null && goodsDonSku.getIsMirror()) {
  122. ChatgptUser chatgptUser = getChatgptUser(userId, relationId, groupsRelation, groupsTrips, goodsDonSku);
  123. if (214610L == relationId) {
  124. return "https://cdn.galaxydvd.com/login_token?access_token=" + chatgptUser.getUserToken();
  125. }
  126. return GPT_DOMAIN + "/login_token?access_token=" + chatgptUser.getUserToken();
  127. } else {
  128. throw BusinessRuntimeException.getInstance("服务器出了点问题");
  129. }
  130. }
  131. /**
  132. * 获取车队登录url
  133. *
  134. * @param userId
  135. * @param carId
  136. * @return
  137. */
  138. @Override
  139. public String getCarLoginUrl(String domain, Long userId, Long relationId, String carId) {
  140. GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
  141. GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
  142. GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
  143. if (goodsDonSku != null && goodsDonSku.getIsMirror() && goodsDonSku.getIsCar()) {
  144. ChatgptUser chatgptUser = getChatgptCarUser(userId, relationId, groupsRelation, goodsDonSku);
  145. String DOMAIN = getGptDomain(domain, goodsDonSku);
  146. log.info("domain:{},用户:{},所在车次:{},座位id:{},在{}获取跳转GPT镜像的登录url", DOMAIN, userId, groupsTrips.getId(), relationId, DateTime.now());
  147. return DOMAIN + "/auth/logintoken?carid=" + carId + "&usertoken=" + chatgptUser.getUserToken();
  148. } else {
  149. throw BusinessRuntimeException.getInstance("服务器出了点问题");
  150. }
  151. }
  152. /**
  153. * 根据userToken获取车队登录url
  154. */
  155. @Override
  156. public String getCarLoginUrlWithToken(String domain, String userToken, String carId) {
  157. ChatgptUser chatgptUser = getCarChatGptUserWithToken(userToken);
  158. if (chatgptUser != null) {
  159. ChatgptUserTokenPrepare chatgptUserTokenPrepare = chatgptUserTokenPrepareMapper.selectOne(Wrappers.lambdaQuery(ChatgptUserTokenPrepare.class)
  160. .eq(ChatgptUserTokenPrepare::getUserToken, userToken)
  161. .last("limit 1"));
  162. Long skuId = chatgptUserTokenPrepare.getSkuId();
  163. GoodsDonSku goodsDonSku = skuMapper.selectById(skuId);
  164. String DOMAIN = getGptDomain(domain, goodsDonSku);
  165. log.info("domain:{},渠道用户:{},在{}获取跳转GPT镜像的登录url", DOMAIN, userToken, DateTime.now());
  166. return DOMAIN + "/auth/logintoken?carid=" + carId + "&usertoken=" + chatgptUser.getUserToken();
  167. } else {
  168. throw BusinessRuntimeException.getInstance("服务出了点问题");
  169. }
  170. }
  171. /**
  172. * 获取车位信息
  173. *
  174. * @param userId
  175. * @param relationId
  176. * @return
  177. */
  178. private GroupsRelation getGroupsRelation(Long userId, Long relationId) {
  179. List<Long> userIdList = userBindRelationService.getRelationUserIdList(userId, null);
  180. GroupsRelation groupsRelation = groupsRelationMapper.selectOne(Wrappers.lambdaQuery(GroupsRelation.class).in(GroupsRelation::getUserId, userIdList).eq(GroupsRelation::getId, relationId));
  181. if (groupsRelation == null) {
  182. throw BusinessRuntimeException.getInstance("车票不存在");
  183. }
  184. return groupsRelation;
  185. }
  186. /**
  187. * 获取车队信息
  188. *
  189. * @param groupsRelation
  190. * @return
  191. */
  192. private GroupsTrips getGroupsTrips(GroupsRelation groupsRelation) {
  193. GroupsTrips groupsTrips = groupsMapper.selectById(groupsRelation.getGroupsId());
  194. if (groupsTrips == null) {
  195. throw BusinessRuntimeException.getInstance("车队异常");
  196. }
  197. return groupsTrips;
  198. }
  199. private ChatgptUser getChatgptUser(Long userId, Long relationId, GroupsRelation groupsRelation, GroupsTrips groupsTrips, GoodsDonSku goodsDonSku) {
  200. ChatgptUser chatgptUser = chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class).eq(ChatgptUser::getRelationId, relationId));
  201. User user = userMapper.selectById(userId);
  202. if (chatgptUser == null) {
  203. chatgptUser = createChatgptUser(groupsRelation, groupsTrips, user, goodsDonSku);
  204. } else {
  205. updateChatgptUser(groupsRelation, user, chatgptUser);
  206. }
  207. return chatgptUser;
  208. }
  209. private ChatgptUser getChatgptCarUser(Long userId, Long relationId, GroupsRelation groupsRelation, GoodsDonSku goodsDonSku) {
  210. ChatgptUser chatgptUser = chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class).eq(ChatgptUser::getRelationId, relationId));
  211. User user = userMapper.selectById(userId);
  212. if (chatgptUser == null) {
  213. chatgptUser = createChatgptUser(groupsRelation, null, user, goodsDonSku);
  214. } else {
  215. updateChatgptUser(groupsRelation, user, chatgptUser);
  216. }
  217. return chatgptUser;
  218. }
  219. /**
  220. * 创建chatgptUser
  221. */
  222. private synchronized ChatgptUser createChatgptUser(GroupsRelation groupsRelation, GroupsTrips groupsTrips, User user, GoodsDonSku goodsDonSku) {
  223. ChatgptUser chatgptUser = new ChatgptUser();
  224. if (goodsDonSku.getIsCar()) {
  225. chatgptUser.setIsCar(true);
  226. chatgptUser.setLimitNum(goodsDonSku.getGptLimitNum());
  227. chatgptUser.setLimitTime(goodsDonSku.getGptLimitTime());
  228. }
  229. try {
  230. chatgptUser.setExpireTime(groupsRelation.getExpiryTime());
  231. chatgptUser.setIsPlus(1);
  232. chatgptUser.setName(user.getNickname());
  233. chatgptUser.setImg(getWxImg(user.getHeadimgurl(), user));
  234. chatgptUser.setRelationId(groupsRelation.getId());
  235. chatgptUser.setUserToken(UUID.randomUUID().toString());
  236. chatgptUserMapper.insert(chatgptUser);
  237. //是否是大学生限时车队
  238. Date expiryTime = groupsRelation.getExpiryTime();
  239. Date startTime = groupsRelation.getStartTime();
  240. //大学生站 1天免费试用
  241. if (DateUtil.between(startTime, expiryTime, DateUnit.DAY) <= 1) {
  242. CollegeStudentStationUseTicketRecord useTicketRecord = new CollegeStudentStationUseTicketRecord();
  243. useTicketRecord.setUserId(user.getId());
  244. useTicketRecord.setUserToken(chatgptUser.getUserToken());
  245. collegeStudentStationUseTicketRecordMapper.insert(useTicketRecord);
  246. }
  247. } catch (DuplicateKeyException e) {
  248. log.error("重复插入镜像用户:{}token,errmsg:{}", user.getNickname(), StringUtil.getErrorText(e));
  249. }
  250. return chatgptUser;
  251. }
  252. /**
  253. * 创建chatgptUser 无需车队
  254. */
  255. public ChatgptUser generateChatGptUserUnderPrepare(ChatgptUserTokenPrepare prepare) {
  256. ChatgptUser chatgptUser = new ChatgptUser();
  257. GoodsDonSku goodsDonSku = skuMapper.selectById(prepare.getSkuId());
  258. if (goodsDonSku == null) {
  259. throw BusinessRuntimeException.getInstance("车队规格不存在,请联系客服..");
  260. }
  261. if (goodsDonSku.getIsCar()) {
  262. chatgptUser.setIsCar(true);
  263. chatgptUser.setLimitNum(goodsDonSku.getGptLimitNum());
  264. chatgptUser.setLimitTime(goodsDonSku.getGptLimitTime());
  265. }
  266. try {
  267. chatgptUser.setExpireTime(DateUtil.offsetMonth(DateTime.now(), goodsDonSku.getMonths()));
  268. chatgptUser.setIsPlus(1);
  269. chatgptUser.setName(prepare.getName());
  270. chatgptUser.setImg(prepare.getImg());
  271. chatgptUser.setUserToken(prepare.getUserToken());
  272. chatgptUserMapper.insert(chatgptUser);
  273. prepare.setStatus(true);
  274. chatgptUserTokenPrepareMapper.updateById(prepare);
  275. log.info("用户激活userToken:{}成功", prepare.getUserToken());
  276. } catch (DuplicateKeyException e) {
  277. log.error("重复插入镜像车队用户:{}token,errmsg:{}" + chatgptUser.getId(), StringUtil.getErrorText(e));
  278. }
  279. return chatgptUser;
  280. }
  281. private void updateChatgptUser(GroupsRelation groupsRelation, User user, ChatgptUser chatgptUser) {
  282. chatgptUser.setName(user.getNickname());
  283. chatgptUser.setImg(getWxImg(user.getHeadimgurl(), user));
  284. chatgptUser.setExpireTime(groupsRelation.getExpiryTime());
  285. chatgptUserMapper.updateById(chatgptUser);
  286. }
  287. /**
  288. * 更换wx头像至oss
  289. */
  290. private String getWxImg(String headimgurl, User user) {
  291. if (headimgurl.contains("thirdwx.qlogo.cn")) {
  292. try {
  293. byte[] body = HttpUtil.downloadBytes(headimgurl);
  294. Path tempFile = Files.createTempFile("wxheadimg-" + user.getId(), ".jpeg");
  295. try (FileImageOutputStream imageOutput = new FileImageOutputStream(tempFile.toFile())) {
  296. imageOutput.write(body, 0, body.length);
  297. }
  298. Map<String, Object> paramMap = new HashMap<>();
  299. paramMap.put("file", tempFile.toFile());
  300. JSONObject result = JSONUtil.parseObj(HttpUtil.post("https://files.liuliangbang.vip/pic/ups", paramMap));
  301. return result.getJSONObject("value").getJSONArray("saved").getJSONObject(0).getJSONObject("info").getStr("cdnUrl");
  302. } catch (Exception ex) {
  303. log.error("上传微信头像错误!msg:{}", StringUtil.getErrorText(ex));
  304. return "./avatars.png";
  305. }
  306. } else {
  307. return user.getHeadimgurl();
  308. }
  309. }
  310. @Override
  311. public ChatgptUser findByUserTokenAndExpireTimeAfter(String userToken, LocalDateTime now) {
  312. return chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class).eq(ChatgptUser::getUserToken, userToken).gt(ChatgptUser::getExpireTime, now));
  313. }
  314. /**
  315. * 获取指定用户ID在滑动窗口内的提问次数
  316. *
  317. * @param userToken 用户token
  318. * @return 滑动窗口内的请求次数
  319. */
  320. @Override
  321. public Long getConversationCount(String userToken, Long limitTime) {
  322. try {
  323. String res = HttpUtil.get("http://154.198.213.52:9611/getConversationCount?userToken=" + userToken);
  324. return new JSONObject(res).getLong("count");
  325. }catch (Exception e){
  326. return 0L;
  327. }
  328. }
  329. public JSONObject getConversationCount(String userToken) {
  330. try {
  331. String res = HttpUtil.get("http://154.198.213.52:9611/getConversationCount?userToken=" + userToken);
  332. return new JSONObject(res);
  333. }catch (Exception e){
  334. return new JSONObject("{\"count\": 0,\"o1_mini\": 0,\"o1_p\": 0}");
  335. }
  336. }
  337. @Override
  338. public ChatgptSession checkSession(String carId) {
  339. ChatgptSession chatgptSession = chatgptSessionMapper.selectOne(Wrappers.lambdaQuery(ChatgptSession.class).eq(ChatgptSession::getCarId, carId).eq(ChatgptSession::getIsCar, true).last(" limit 1"));
  340. if (chatgptSession == null) {
  341. throw BusinessRuntimeException.getInstance("车队不存在");
  342. }
  343. return chatgptSession;
  344. }
  345. private Long carOaiLimit(String carId, Boolean isTeam) {
  346. if(isTeam && redisService.hasKey("chatgpt:team:clears_in:" + carId)){
  347. return redisService.getExpire("chatgpt:team:clears_in:" + carId);
  348. }
  349. if (redisService.hasKey("chatgpt:clears_in:" + carId)) {
  350. return redisService.getExpire("chatgpt:clears_in:" + carId);
  351. }
  352. return 0L;
  353. }
  354. @Override
  355. public Boolean checkCarAccount(long userId, Long relationId) {
  356. GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
  357. GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
  358. GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
  359. if (goodsDonSku != null && goodsDonSku.getGoodsId() == 18 && goodsDonSku.getIsMirror() && goodsDonSku.getIsCar()) {
  360. ChatgptUser chatgptUser = getChatgptCarUser(userId, relationId, groupsRelation, goodsDonSku);
  361. return true;
  362. } else {
  363. return false;
  364. }
  365. }
  366. @Override
  367. public Boolean checkCarAccountWithToken(String userToken) {
  368. ChatgptUser chatgptUser = getCarChatGptUserWithToken(userToken);
  369. return chatgptUser != null;
  370. }
  371. @Override
  372. public ChatgptUser getCarChatGptUser(long userId, Long relationId) {
  373. GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
  374. GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
  375. GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
  376. if (goodsDonSku != null && goodsDonSku.getIsMirror() && goodsDonSku.getIsCar()) {
  377. return getChatgptCarUser(userId, relationId, groupsRelation, goodsDonSku);
  378. }
  379. return null;
  380. }
  381. @Override
  382. public ChatgptUser getCarChatGptUserWithToken(String userToken) {
  383. ChatgptUser chatgptUser = chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class).eq(ChatgptUser::getUserToken, userToken).last(" limit 1"));
  384. if (chatgptUser == null) {
  385. //是否是用户token预备账号
  386. ChatgptUserTokenPrepare chatgptUserTokenPrepare = chatgptUserTokenPrepareMapper.selectOne(Wrappers.lambdaQuery(ChatgptUserTokenPrepare.class)
  387. .eq(ChatgptUserTokenPrepare::getUserToken, userToken).last("limit 1"));
  388. if (chatgptUserTokenPrepare != null) {
  389. chatgptUser = generateChatGptUserUnderPrepare(chatgptUserTokenPrepare);
  390. }
  391. }
  392. if (chatgptUser != null) {
  393. if(!chatgptUser.getIsCar()){
  394. throw BusinessRuntimeException.getInstance("您非车队用户,请购买车队套餐");
  395. }
  396. if (chatgptUser.getRelationId() != null) {
  397. throw BusinessRuntimeException.getInstance("非渠道用户,请通过官网登录!");
  398. }
  399. return chatgptUser;
  400. }
  401. return null;
  402. }
  403. /**
  404. * 获取车队分页信息
  405. */
  406. @Override
  407. public SearchResult<ChatgptCarInfoView> getCarInfoPage(String userToken, Integer limit) {
  408. SearchResult<ChatgptCarInfoView> chatgptCarInfoViews = beanSearcher.search(ChatgptCarInfoView.class, MapUtils.flatBuilder(request.getParameterMap())
  409. .orderBy(ChatgptCarInfoView::getSort).asc()
  410. .orderBy(ChatgptCarInfoView::getTeamClearsIn).asc()
  411. .orderBy(ChatgptCarInfoView::getClearsIn).asc().orderBy(ChatgptCarInfoView::getScore).asc().build());
  412. if (chatgptCarInfoViews != null) {
  413. chatgptCarInfoViews.getDataList().forEach(this::buildChatgptCarInfoView);
  414. }
  415. return chatgptCarInfoViews;
  416. }
  417. /**
  418. * 获取车队信息
  419. */
  420. @Override
  421. public List<ChatgptCarInfoView> getCarInfoList(String userToken, Integer limit, Boolean isPlus) {
  422. List<ChatgptCarInfoView> lowestScoreFleets = getLowestScoreFleets(limit, isPlus);
  423. return lowestScoreFleets;
  424. }
  425. // 获取得分最低的N个车队的方法
  426. public List<ChatgptCarInfoView> getLowestScoreFleets(int count, Boolean isPlus) {
  427. List<ChatgptCarInfoView> chatgptCarInfoViews = beanSearcher.searchList(ChatgptCarInfoView.class, MapUtils.builder().field(ChatgptCarInfoView::getIsPlus, isPlus).orderBy(ChatgptCarInfoView::getScore).asc().limit(0, count).build());
  428. chatgptCarInfoViews.forEach(this::buildChatgptCarInfoView);
  429. return chatgptCarInfoViews;
  430. }
  431. // 构建ChatgptCarInfoView对象
  432. private ChatgptCarInfoView buildChatgptCarInfoView(ChatgptCarInfoView view) {
  433. // 在这里根据carId获取相关信息并填充到ChatgptCarInfoView对象中
  434. view.setStatus(view.getScore() >= 100 ? "繁忙" : "空闲"); // 或“繁忙”
  435. if(view.getCarName().contains("T")){
  436. view.setType("Team");
  437. Long team = view.getTeamClearsIn();
  438. if (team != 0L) {
  439. view.setTeamExpTime(new DateTime(System.currentTimeMillis() + team * 1000));
  440. }
  441. Long aLong = view.getClearsIn();
  442. if (aLong != 0L) {
  443. view.setExpTime(new DateTime(System.currentTimeMillis() + aLong * 1000));
  444. }
  445. if (view.getExpTime() != null && view.getTeamExpTime() != null) {
  446. view.setStatus("全部停运");
  447. }else if(view.getExpTime() != null || view.getTeamExpTime() != null) {
  448. view.setStatus("部分停运");
  449. }
  450. }else {
  451. view.setType("Plus");
  452. Long aLong = view.getClearsIn();
  453. if (aLong != 0L) {
  454. view.setStatus("停运");
  455. view.setExpTime(new DateTime(System.currentTimeMillis() + aLong * 1000));
  456. }
  457. }
  458. view.setScore(Math.min(BigDecimal.valueOf(view.getScore()).multiply(BigDecimal.valueOf(0.5)).setScale(0, RoundingMode.DOWN).doubleValue(), 200));
  459. return view;
  460. }
  461. @Override
  462. public ChatGptUserView getUserInfo(long userId, Long relationId) {
  463. if (checkCarAccount(userId, relationId)) {
  464. GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
  465. GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
  466. GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
  467. ChatgptUser chatgptUser = chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class).eq(ChatgptUser::getRelationId, relationId));
  468. Integer count = upgradePackageMapper.selectCount(Wrappers.lambdaQuery(UpgradePackage.class).apply("JSON_CONTAINS(sku_ids,'\"" + groupsTrips.getSkuId() + "\"')")
  469. .gt(UpgradePackage::getNum, chatgptUser.getLimitNum())
  470. .eq(UpgradePackage::getStatus, Boolean.TRUE)
  471. .eq(UpgradePackage::getGptType, 0)
  472. .eq(UpgradePackage::getDeleted, Boolean.TRUE));
  473. Boolean upgrade = count > 0 ? true : false;
  474. OrderDon orderDon = orderDonMapper.selectOne(Wrappers.lambdaQuery(OrderDon.class)
  475. .eq(OrderDon::getRelationId, relationId)
  476. .eq(OrderDon::getUserId, groupsRelation.getUserId())
  477. .notIn(OrderDon::getStatus, Constant.noOrderAllStatus)
  478. .orderByDesc(OrderDon::getId)
  479. .last("limit 1"));
  480. Long orderId = null;
  481. if (orderDon != null) {
  482. orderId = orderDon.getId();
  483. }
  484. BigDecimal packagePrice = null;
  485. if (chatgptUser.getPackageId() != 0) {
  486. UpgradePackage upgradePackage = upgradePackageMapper.selectById(chatgptUser.getPackageId());
  487. packagePrice = upgradePackage.getPrice();
  488. }
  489. JSONObject conversationCount = getConversationCount(chatgptUser.getUserToken());
  490. ChatGptUserView view = ChatGptUserView.builder()
  491. .skuName(goodsDonSku.getSubTitle())
  492. .specVal(goodsDonSku.getSpecVal())
  493. .expireTime(chatgptUser.getExpireTime())
  494. .limitNum(chatgptUser.getLimitNum())
  495. .limitTime(chatgptUser.getLimitTime())
  496. .upgrade(upgrade)
  497. .orderId(orderId)
  498. .goodsId(goodsDonSku.getGoodsId())
  499. .skuId(goodsDonSku.getId())
  500. .months(goodsDonSku.getMonths())
  501. .use(Math.min(chatgptUser.getLimitNum(), Optional.ofNullable(conversationCount.getLong("count")).orElse(0L)))
  502. .packageId(chatgptUser.getPackageId())
  503. .packagePrice(packagePrice)
  504. .isRenewPackage(chatgptUser.getRenewOrderId() != null ? true : false)
  505. .renewSkuId(chatgptUser.getRenewSkuId())
  506. .renewOrderId(chatgptUser.getRenewOrderId())
  507. .build();
  508. if (chatgptUser.getLimitNum() > 20) {
  509. view.setO1PUse(Optional.ofNullable(conversationCount.getLong("o1_p")).orElse(0L));
  510. view.setO1MiniUse(Optional.ofNullable(conversationCount.getLong("o1_mini")).orElse(0L));
  511. }
  512. return view;
  513. } else {
  514. throw BusinessRuntimeException.getInstance("您还未购买该车票");
  515. }
  516. }
  517. @Override
  518. public ChatGptUserView getUserInfoWithToken(String userToken) {
  519. if (checkCarAccountWithToken(userToken)) {
  520. ChatgptUser chatgptUser = getCarChatGptUserWithToken(userToken);
  521. return ChatGptUserView.builder()
  522. .skuName("")
  523. .expireTime(chatgptUser.getExpireTime())
  524. .limitNum(chatgptUser.getLimitNum())
  525. .limitTime(chatgptUser.getLimitTime())
  526. .use(getConversationCount(chatgptUser.getUserToken(), chatgptUser.getLimitTime()))
  527. .build();
  528. } else {
  529. throw BusinessRuntimeException.getInstance("您还未购买该车票");
  530. }
  531. }
  532. public String getGptDomain(String domain, GoodsDonSku goodsDonSku) {
  533. String DOMAIN = domain;
  534. if (StrUtil.isEmpty(DOMAIN)) {
  535. DOMAIN = "https://chat.galaxydvd.com";
  536. try {
  537. SysConfig sysConfig = sysConfigService.getOne(Wrappers.lambdaQuery(SysConfig.class).eq(SysConfig::getSysKey, goodsDonSku != null && goodsDonSku.getId() == 444 ? "zhihu_gpt_domain" : "mirror_gpt_domain"));
  538. if (sysConfig != null) {
  539. String sysValue = sysConfig.getSysValue();
  540. if (JSONUtil.isJsonArray(sysValue)) {
  541. List<String> domainList = JSONUtil.parseArray(sysValue).toList(String.class);
  542. DOMAIN = domainList.get(RandomUtil.randomInt(domainList.size()));
  543. }
  544. }
  545. } catch (Exception e) {
  546. log.error("随机抽取域名错误. msg:{}", StringUtil.getErrorText(e));
  547. }
  548. }
  549. return DOMAIN;
  550. }
  551. }