zwhui 2 éve
szülő
commit
1411174fc2

+ 4 - 1
midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -78,7 +78,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
 
     private final MidjourneyUserConversationMapper conversationMapper;
 
-    private static final List<String> progress = List.of("100%");
+    private static final List<String> progress = List.of("35%","65","100%");
 
     private static final List<String> status = List.of("MODAL","CANCEL","FAILURE");
 
@@ -728,6 +728,9 @@ public class MidjourneyServiceImpl implements MidjourneyService {
             MidjourneyUserConversation dbConversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, conversation.getTaskId())
                     .eq(MidjourneyUserConversation::getUserId, userId).orderByDesc(MidjourneyUserConversation::getId).last("limit 1"));
             if (dbConversation != null) {
+                if (conversation.getProgressNum() <= dbConversation.getProgressNum()){
+                    return;
+                }
                 if ("SUCCESS".equals(dbConversation.getStatus()) || "FAILURE".equals(dbConversation.getStatus())) {
                     return;
                 }