diff --git a/src/main/java/com/ycwl/basic/service/pc/impl/FaceServiceImpl.java b/src/main/java/com/ycwl/basic/service/pc/impl/FaceServiceImpl.java index 1a3bad93..20dbfe1c 100644 --- a/src/main/java/com/ycwl/basic/service/pc/impl/FaceServiceImpl.java +++ b/src/main/java/com/ycwl/basic/service/pc/impl/FaceServiceImpl.java @@ -20,6 +20,7 @@ import com.ycwl.basic.mapper.VideoMapper; import com.ycwl.basic.mapper.OrderMapper; import com.ycwl.basic.model.mobile.face.FaceRecognizeResp; import com.ycwl.basic.model.mobile.face.FaceStatusResp; +import com.ycwl.basic.model.mobile.goods.VideoTaskStatusVO; import com.ycwl.basic.model.mobile.order.IsBuyRespVO; import com.ycwl.basic.model.mobile.scenic.content.ContentPageVO; import com.ycwl.basic.model.mobile.statistic.req.StatisticsRecordAddReq; @@ -45,6 +46,7 @@ import com.ycwl.basic.repository.FaceRepository; import com.ycwl.basic.repository.ScenicRepository; import com.ycwl.basic.repository.VideoRepository; import com.ycwl.basic.repository.VideoTaskRepository; +import com.ycwl.basic.service.mobile.GoodsService; import com.ycwl.basic.service.pc.FaceService; import com.ycwl.basic.service.pc.ScenicService; import com.ycwl.basic.service.task.TaskFaceService; @@ -123,6 +125,8 @@ public class FaceServiceImpl implements FaceService { private RedisTemplate redisTemplate; @Autowired private FaceSampleMapper faceSampleMapper; + @Autowired + private GoodsService goodsService; @Override public ApiResponse> pageQuery(FaceReqQuery faceReqQuery) { @@ -876,13 +880,18 @@ public class FaceServiceImpl implements FaceService { statusResp.setStep2Status(false); statusResp.setDisplayText("Hey,快去智能机位打卡吧"); } - List videoEntities = videoMapper.listRelationByFace(face.getMemberId(), faceId); - if (videoEntities != null && !videoEntities.isEmpty()) { - statusResp.setStep3Status(true); - statusResp.setDisplayText("帧途AI已为您渲染"+ videoEntities.size() +"个vlog"); + VideoTaskStatusVO taskStatusByFaceId = goodsService.getTaskStatusByFaceId(faceId); + if (Integer.valueOf(1).equals(taskStatusByFaceId.getStatus())) { + if (taskStatusByFaceId.getCount() > 0) { + statusResp.setStep3Status(true); + statusResp.setDisplayText("帧途AI已为您渲染"+ taskStatusByFaceId.getCount() +"个vlog"); + } else { + statusResp.setStep3Status(false); + statusResp.setDisplayText("帧途AI将会为您渲染vlog,请稍候"); + } } else { statusResp.setStep3Status(false); - statusResp.setDisplayText("帧途AI将会为您渲染vlog,请稍后"); + statusResp.setDisplayText("帧途AI正在为您渲染vlog,请稍候"); } return statusResp; }