zwhui 2 years ago
parent
commit
457e7b9311

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

@@ -50,6 +50,8 @@ public class MidjourneyServiceImpl implements MidjourneyService {
 
     private static final List<String> progress = Arrays.asList("25%","35%", "45%","65%", "75%", "85%", "95%","100%");
 
+    private static final List<String> status = Arrays.asList("MODAL","IN_PROGRESS", "SUCCESS", "FAILURE", "CANCEL");
+
     private static final GlobalThreadPoolTaskExecutor TASK_EXECUTOR = GlobalThreadPoolTaskExecutor.getInstance();
 
 
@@ -196,7 +198,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
 
 
     public void sync(MidjourneyUserConversation conversation) throws IOException {
-        if (progress.contains(conversation.getProgress()) || StringUtils.equals(conversation.getStatus(), "MODAL")) {
+        if (progress.contains(conversation.getProgress()) || status.contains(conversation.getStatus())) {
             log.info("同步任务:{},进度:{}",conversation.getTaskId(),conversation.getProgress());
             MidjourneyUserConversation dbConversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, conversation.getTaskId())
                     .eq(MidjourneyUserConversation::getUserId, conversation.getUserId()).last("limit 1"));