ChatGptAccountServiceImpl.java 26 KB

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