fix 慢速限制

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

@@ -271,14 +271,17 @@ public class MidjourneyServiceImpl implements MidjourneyService {
             accountWithMinUsage = getAccountWithMinUsage(instanceId);
             param.put("accountFilter",MapUtil.builder(new HashMap<String,Object>())
                     .put("instanceId",accountWithMinUsage).build());
-            String key = RedisService.key.MIDJOURNEY_USER_TIME.getName() + user.getId();
-            Object count = redisService.get(key);
-            if (count == null) {
-                redisService.set(key, 1L,RedisService.key.MIDJOURNEY_USER_TIME.getTimeout());
-            }else {
-                if (redisService.incr(key, 1L) > 200) {
-                    user.setIsBlack(true);
-                    midjourneyUserMapper.updateById(user);
+            Object customId = param.get("customId");
+            if (customId != null && !customId.toString().contains("upsample") && !action.equals("modal")){
+                String key = RedisService.key.MIDJOURNEY_USER_TIME.getName() + user.getId();
+                Object count = redisService.get(key);
+                if (count == null) {
+                    redisService.set(key, 1L,RedisService.key.MIDJOURNEY_USER_TIME.getTimeout());
+                }else {
+                    if (redisService.incr(key, 1L) > 200) {
+                        user.setIsBlack(true);
+                        midjourneyUserMapper.updateById(user);
+                    }
                 }
             }
         }

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

@@ -76,6 +76,7 @@ public class MidjourneyController {
             throw new BusinessRuntimeException("账号不存在,请重新登录");
         }
         if (midjourneyUser.getIsBlack()){
+            log.info("账号异常,请联系客服 userId:" + midjourneyUser.getId());
             throw new BusinessRuntimeException("账号异常,请联系客服");
         }
         if(!redisService.hasKey(RedisService.key.MIDJOURNEY_FAST_LIMIT.getName() + midjourneyUser.getId())){