Bläddra i källkod

修改图生视频任务状态通知判断

cmy 4 veckor sedan
förälder
incheckning
f695858394

+ 8 - 2
src/main/java/com/backendsys/modules/ai/volcengine/service/impl/VolcengineVideoTaskServiceImpl.java

@@ -33,6 +33,7 @@ import javax.annotation.PostConstruct;
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
 
@@ -308,8 +309,13 @@ public class VolcengineVideoTaskServiceImpl extends ServiceImpl<VolcengineVideoT
         VolcengineVideoTaskDetail dataBean = volcengineVideoTaskDetailService.getOne(Wrappers.<VolcengineVideoTaskDetail>lambdaQuery()
                 .eq(VolcengineVideoTaskDetail::getVolcengine_task_id, notify.getId()));
         if (dataBean == null) {
-            log.error("图生视频通知异常,找不到任务明细,通知内容:" + notify);
-            throw new CustException("图生视频通知异常,找不到任务明细");
+            // 任务状态排队/运行中,可能速度快于平台数据库保存,可忽略,其他状态视为异常
+            if(!Arrays.asList("queued","running").contains(notify.getStatus())){
+                log.error("图生视频通知异常,找不到任务明细,通知内容:" + notify);
+                throw new CustException("图生视频通知异常,找不到任务明细");
+            } else {
+                return;
+            }
         }
 
         if("succeeded".equals(dataBean.getStatus()) || "failed".equals(dataBean.getStatus())) {