chenbiao 2 роки тому
батько
коміт
8d7a3ebccd

+ 16 - 11
midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -621,7 +621,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
      * 用户生图次数限制
      */
     public void userSubmitLimit(MidjourneyUser user,String action,Object customId) {
-        if (StringUtils.equals("modal", action) || (customId != null && customId.toString().contains("upsample"))) {
+        if (StringUtils.equals("modal", action) || (customId != null)) {
             return;
         }
 
@@ -649,17 +649,22 @@ public class MidjourneyServiceImpl implements MidjourneyService {
         }
 
         // 根据提交次数设置提交间隔
+
         long submitInterval;
-        if (submitCount >= 140) {
-            submitInterval = 100L + (long) (Math.random() * 21); // 140次后随机间隔100-120秒
-        } else if (submitCount >= 100) {
-            submitInterval = 60L; // 100次后60秒
-        } else if (submitCount >= 60) {
-            submitInterval = 40L; // 60次后40秒
-        } else if (submitCount >= 30) {
-            submitInterval = 30L; // 30次后30秒
-        } else {
-            submitInterval = 20L; // 默认20秒
+        if(customId.toString().contains("upsample")){
+            submitInterval = 10L;
+        }else {
+            if (submitCount >= 140) {
+                submitInterval = 100L + (long) (Math.random() * 21); // 140次后随机间隔100-120秒
+            } else if (submitCount >= 100) {
+                submitInterval = 60L; // 100次后60秒
+            } else if (submitCount >= 60) {
+                submitInterval = 40L; // 60次后40秒
+            } else if (submitCount >= 30) {
+                submitInterval = 30L; // 30次后30秒
+            } else {
+                submitInterval = 20L; // 默认20秒
+            }
         }
 
         // 用户只能在指定间隔内提交一次