ソースを参照

fix 收藏不扣次数

zwhui 2 年 前
コミット
f04be5ffae

+ 0 - 1
netflix-service/src/main/java/com/cyksj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -224,7 +224,6 @@ public class MidjourneyServiceImpl implements MidjourneyService {
         if (customId.contains("BOOKMARK")) {
             conversation.setBookmark(true);
             conversationMapper.updateById(conversation);
-            recoverUserLimit(user.getId(), user.getMode(),num);
             return null;
         }
         Map<String, Object> param = MapUtil.builder(new HashMap<String,Object>())

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

@@ -306,9 +306,12 @@ public class MidjourneyController {
     public Result<SubmitResult> action(@RequestBody SubmitActionDTO actionDTO) {
         log.info("任务id:{},执行动作:{}",actionDTO.getTaskId(),actionDTO.getCustomId());
         MidjourneyUser user = getUser();
-        Long num = checkUserLimit(user);
-        if (num != null && num < 0){
-            throw BusinessRuntimeException.getInstance("次数已用完");
+        Long num = 0L;
+        if (!actionDTO.getCustomId().contains("BOOKMARK")) {
+            num = checkUserLimit(user);
+            if (num != null && num < 0) {
+                throw BusinessRuntimeException.getInstance("次数已用完");
+            }
         }
         SubmitResult conversation;
         try {