|
|
@@ -1,29 +1,24 @@
|
|
|
package com.cyksj.web.controller.mirror;
|
|
|
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
-import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import com.cyksj.common.annotation.NoSubmit;
|
|
|
-import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.cyksj.common.exception.BusinessRuntimeException;
|
|
|
import com.cyksj.common.util.IoKit;
|
|
|
import com.cyksj.dto.Result;
|
|
|
import com.cyksj.enums.GatewayResponse;
|
|
|
import com.cyksj.mapper.GroupsRelationMapper;
|
|
|
-import com.cyksj.mapper.MidjourneyPaintingCollectRecordMapper;
|
|
|
import com.cyksj.mapper.MidjourneyPaintingPlazaMapper;
|
|
|
+import com.cyksj.mapper.MidjourneyPaintingCollectRecordMapper;
|
|
|
import com.cyksj.mapper.MidjourneyUserMapper;
|
|
|
-import com.cyksj.model.entity.MidjourneyUser;
|
|
|
-import com.cyksj.model.entity.MidjourneyUserConversation;
|
|
|
-import com.cyksj.model.response.SubmitResult;
|
|
|
import com.cyksj.model.dto.*;
|
|
|
-import com.cyksj.model.entity.GroupsRelation;
|
|
|
import com.cyksj.model.entity.MidjourneyPaintingPlaza;
|
|
|
import com.cyksj.model.entity.MidjourneyUser;
|
|
|
import com.cyksj.model.entity.MidjourneyUserConversation;
|
|
|
import com.cyksj.model.views.MidjourneyPaintingUserView;
|
|
|
import com.cyksj.model.views.MidjourneyUserPaintingDayRecordView;
|
|
|
import com.cyksj.model.views.MidjourneyUserPaintingRecordView;
|
|
|
+import com.cyksj.model.response.SubmitResult;
|
|
|
import com.cyksj.redis.RedisService;
|
|
|
import com.cyksj.service.midjourney.MidjourneyService;
|
|
|
import com.cyksj.service.user.UserBindRelationService;
|
|
|
@@ -65,29 +60,46 @@ public class MidjourneyController {
|
|
|
|
|
|
private final GroupsRelationMapper groupsRelationMapper;
|
|
|
|
|
|
- private final UserBindRelationService userBindRelationService;
|
|
|
-
|
|
|
private final MidjourneyPaintingPlazaMapper midjourneyPaintingPlazaMapper;
|
|
|
|
|
|
- private final MidjourneyPaintingCollectRecordMapper midjourneyPaintingCollectRecordMapper;
|
|
|
-
|
|
|
public MidjourneyUser getUser(){
|
|
|
String userToken = request.getHeader("user-token");
|
|
|
-
|
|
|
- MidjourneyUser midjourneyUser = midjourneyUserMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUser.class)
|
|
|
- .eq(MidjourneyUser::getUserToken, userToken).last("limit 1"));
|
|
|
+ MidjourneyUser midjourneyUser = (MidjourneyUser) redisService.get(RedisService.key.MIDJOURNEY_USER.getName() + userToken);
|
|
|
if (midjourneyUser == null){
|
|
|
- throw new BusinessRuntimeException("账号不存在,请重新登录");
|
|
|
- }
|
|
|
- Long relationId = midjourneyUser.getRelationId();
|
|
|
- GroupsRelation relation = groupsRelationMapper.selectById(relationId);
|
|
|
- midjourneyUser.setAqType(relation.getAqType());
|
|
|
- if(!redisService.hasKey(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId())){
|
|
|
- redisService.set(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId(),midjourneyUser.getMjFastNum());
|
|
|
- }
|
|
|
-
|
|
|
- if(!redisService.hasKey(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId())){
|
|
|
- redisService.set(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId(),midjourneyUser.getMjRelaxNum());
|
|
|
+ midjourneyUser = midjourneyUserMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUser.class)
|
|
|
+ .eq(MidjourneyUser::getUserToken, userToken).last("limit 1"));
|
|
|
+ if (midjourneyUser == null){
|
|
|
+ throw new BusinessRuntimeException("账号不存在,请重新登录");
|
|
|
+ }
|
|
|
+ Long relationId = midjourneyUser.getRelationId();
|
|
|
+ GroupsRelation relation = groupsRelationMapper.selectById(relationId);
|
|
|
+ midjourneyUser.setAqType(relation.getAqType());
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_USER.getName() + userToken, midjourneyUser, RedisService.key.MIDJOURNEY_USER.getTimeout());
|
|
|
+ Object fastNum = redisService.get(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId());
|
|
|
+ if (fastNum == null){
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId(), midjourneyUser.getMjFastNum(), RedisService.key.MIDJOURNEY_FAST_LIMIT.getTimeout());
|
|
|
+ }
|
|
|
+ Object relaxNum = redisService.get(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId());
|
|
|
+ if (relaxNum == null){
|
|
|
+ if (midjourneyUser.getMjRelaxNum() != null){
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId(), midjourneyUser.getMjRelaxNum(), RedisService.key.MIDJOURNEY_RELAX_LIMIT.getTimeout());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }else {
|
|
|
+ Object fastNum = redisService.get(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId());
|
|
|
+ if (fastNum == null){
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId(), midjourneyUser.getMjFastNum(), RedisService.key.MIDJOURNEY_FAST_LIMIT.getTimeout());
|
|
|
+ }else {
|
|
|
+ midjourneyUser.setMjFastNum(Integer.parseInt(fastNum.toString()));
|
|
|
+ }
|
|
|
+ Object relaxNum = redisService.get(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId());
|
|
|
+ if (relaxNum == null){
|
|
|
+ if (midjourneyUser.getMjRelaxNum() != null){
|
|
|
+ redisService.set(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + midjourneyUser.getId(), midjourneyUser.getMjRelaxNum(), RedisService.key.MIDJOURNEY_RELAX_LIMIT.getTimeout());
|
|
|
+ }
|
|
|
+ }else {
|
|
|
+ midjourneyUser.setMjRelaxNum(Integer.parseInt(relaxNum.toString()));
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
return midjourneyUser;
|