zwhui 2 лет назад
Родитель
Сommit
7b300fed69

+ 3 - 2
midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -298,7 +298,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
                 if (button.getCustomId().equals(customId)) {
                     log.info("action customId:{}", customId);
                     button.setStyle(3);
-                    if (!"Vary (Region)".equals(button.getLabel()) && !"Custom Zoom".equals(button.getLabel())) {
+                    if (!"Vary (Region)".equals(button.getLabel()) && !"Custom Zoom".equals(button.getLabel()) && !customId.contains("PicReader")) {
                         modalFlag.set(true);
                     }
                 }
@@ -533,15 +533,16 @@ public class MidjourneyServiceImpl implements MidjourneyService {
                     Integer mode = dbConversation.getMode();
                     if (mode == 1){
                         num =  redisService.incr(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + userId, 1L);
+                        midjourneyUserMapper.update(null, wrapper);
                         wrapper.set(MidjourneyUser::getMjFastNum, num);
                     }
                     if (mode == 2){
                         if (num != null) {
                             num = redisService.incr(RedisService.key.MIDJOURNEY_RELAX_LIMIT.getName() + userId, 1L);
                             wrapper.set(MidjourneyUser::getMjRelaxNum, num);
+                            midjourneyUserMapper.update(null, wrapper);
                         }
                     }
-                    midjourneyUserMapper.update(null, wrapper);
                 }
             }
         }