TASK:新增状态ALMOST_DONE

src/main/java/com/aida/lanecarford/common/enums/StatusEnum.java

@@ -24,7 +24,10 @@ public enum StatusEnum {
     RUNNING(3),
 
     @Schema(description = "重试中")
-    RETRYING(4);
+    RETRYING(4),
+
+    @Schema(description = "生成即将结束")
+    ALMOST_DONE(5);
 
     private int code;
 

src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java

@@ -174,6 +174,10 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
                     ((OutfitResultVO) outfitResult).setCreateTimeStamp(System.currentTimeMillis());
                     ((OutfitResultVO) outfitResult).setPath(null);
                     break;
+                case "almost_done":
+                    // 此时是没有更新path的
+                    status = StatusEnum.ALMOST_DONE.name();
+                    break;
                 case /*"failed",*/ "retry_failed":
                     status = StatusEnum.FAILED.name();
                     break;
@@ -236,7 +240,8 @@ public class StyleServiceImpl extends ServiceImpl<StyleMapper, Style> implements
             // 判断这条记录的状态是否为成功或者失败，否，判断这条记录的创建时间是否超过3分钟，否，继续往后，是，设置为失败并更新数据库
             if (outfit instanceof OutfitResultVO) {
                 if ((((OutfitResultVO) outfit).getStatus().equals(StatusEnum.PENDING.name())
-                        || ((OutfitResultVO) outfit).getStatus().equals(StatusEnum.RUNNING.name()))
+                        || ((OutfitResultVO) outfit).getStatus().equals(StatusEnum.RUNNING.name())
+                        || ((OutfitResultVO) outfit).getStatus().equals(StatusEnum.ALMOST_DONE.name()))
                         && isExpired(((OutfitResultVO) outfit).getCreateTimeStamp())) {
                     // 设置状态为失败
                     ((OutfitResultVO) outfit).setStatus(StatusEnum.FAILED.name());