diff --git a/src/main/java/com/ycwl/basic/task/VideoTaskGenerator.java b/src/main/java/com/ycwl/basic/task/VideoTaskGenerator.java index bd73c15..f24691f 100644 --- a/src/main/java/com/ycwl/basic/task/VideoTaskGenerator.java +++ b/src/main/java/com/ycwl/basic/task/VideoTaskGenerator.java @@ -15,6 +15,7 @@ import com.ycwl.basic.model.pc.scenic.resp.ScenicRespVO; import com.ycwl.basic.model.task.resp.SearchFaceRespVo; import com.ycwl.basic.repository.ScenicRepository; import com.ycwl.basic.repository.TemplateRepository; +import com.ycwl.basic.service.pc.FaceService; import com.ycwl.basic.service.task.TaskFaceService; import com.ycwl.basic.service.task.impl.TaskTaskServiceImpl; import lombok.extern.slf4j.Slf4j; @@ -47,8 +48,8 @@ public class VideoTaskGenerator { private ScenicMapper scenicMapper; @Autowired private ScenicRepository scenicRepository; - - // TODO: 可配置,现在赶时间暂时写死 + @Autowired + private FaceService faceService; @Scheduled(cron = "0 0 * * * *") public void generateVideoTask() { @@ -84,7 +85,7 @@ public class VideoTaskGenerator { query.setEndTime(DateUtil.endOfDay(new Date())); List list = faceMapper.list(query); list.stream().parallel().forEach(face -> { - taskFaceService.searchFace(face.getId()); + faceService.matchFaceId(face.getId(), false); if (Integer.valueOf(3).equals(scenicConfig.getBookRoutine())) { // 全部生成 contentList.forEach(content -> {