diff --git a/src/main/java/com/aida/lanecarford/common/enums/StatusEnum.java b/src/main/java/com/aida/lanecarford/common/enums/StatusEnum.java index f5efc29..0f29090 100644 --- a/src/main/java/com/aida/lanecarford/common/enums/StatusEnum.java +++ b/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; diff --git a/src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java b/src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java index 8fd798b..23cc2ed 100644 --- a/src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java +++ b/src/main/java/com/aida/lanecarford/service/impl/StyleServiceImpl.java @@ -174,6 +174,10 @@ public class StyleServiceImpl extends ServiceImpl 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 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());