mj详情添加车票类型

netflix-dao/src/main/java/com/cyksj/model/entity/MidjourneyUser.java

@@ -5,7 +5,6 @@ package com.cyksj.model.entity;
  * @date 2024/4/16 17:01
  */
 
-import com.baomidou.mybatisplus.annotation.TableField;
 import com.ejlchina.searcher.bean.SearchBean;
 import lombok.Data;
 
@@ -54,4 +53,9 @@ public class MidjourneyUser extends BaseEntity{
      * 当前模式 1:fast 2:relax
      */
     private Integer mode;
+
+    /**
+     * 车票类型 2试用
+     */
+    private Integer aqType;
 }

netflix-web/src/main/java/com/cyksj/web/controller/mirror/MidjourneyController.java

@@ -12,10 +12,7 @@ import com.cyksj.mapper.MidjourneyPaintingCollectRecordMapper;
 import com.cyksj.mapper.MidjourneyPaintingPlazaMapper;
 import com.cyksj.mapper.MidjourneyUserMapper;
 import com.cyksj.model.dto.*;
-import com.cyksj.model.entity.MidjourneyPaintingCollectRecord;
-import com.cyksj.model.entity.MidjourneyPaintingPlaza;
-import com.cyksj.model.entity.MidjourneyUser;
-import com.cyksj.model.entity.MidjourneyUserConversation;
+import com.cyksj.model.entity.*;
 import com.cyksj.model.views.MidjourneyPaintingUserView;
 import com.cyksj.model.views.MidjourneyUserPaintingDayRecordView;
 import com.cyksj.model.views.MidjourneyUserPaintingRecordView;
@@ -73,6 +70,9 @@ public class MidjourneyController {
             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){