116 lines
4.9 KiB
Java
116 lines
4.9 KiB
Java
package com.ycwl.basic.task;
|
|
|
|
import cn.hutool.core.date.DateUtil;
|
|
import com.ycwl.basic.biz.TemplateBiz;
|
|
import com.ycwl.basic.mapper.FaceMapper;
|
|
import com.ycwl.basic.mapper.FaceSampleMapper;
|
|
import com.ycwl.basic.mapper.ScenicMapper;
|
|
import com.ycwl.basic.mapper.TemplateMapper;
|
|
import com.ycwl.basic.model.mobile.scenic.content.ContentPageVO;
|
|
import com.ycwl.basic.model.pc.face.req.FaceReqQuery;
|
|
import com.ycwl.basic.model.pc.face.resp.FaceRespVO;
|
|
import com.ycwl.basic.model.pc.scenic.entity.ScenicConfigEntity;
|
|
import com.ycwl.basic.model.pc.scenic.req.ScenicReqQuery;
|
|
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;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.context.annotation.Profile;
|
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
|
import org.springframework.scheduling.annotation.Scheduled;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import java.util.Calendar;
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
|
|
@Slf4j
|
|
@EnableScheduling
|
|
@Component
|
|
@Profile("prod")
|
|
public class VideoTaskGenerator {
|
|
@Autowired
|
|
private FaceMapper faceMapper;
|
|
@Autowired
|
|
private TaskFaceService taskFaceService;
|
|
@Autowired
|
|
private TemplateBiz templateBiz;
|
|
@Autowired
|
|
private TaskTaskServiceImpl taskTaskService;
|
|
@Autowired
|
|
private TemplateMapper templateMapper;
|
|
@Autowired
|
|
private ScenicMapper scenicMapper;
|
|
@Autowired
|
|
private ScenicRepository scenicRepository;
|
|
@Autowired
|
|
private FaceService faceService;
|
|
|
|
@Scheduled(cron = "0 0 * * * *")
|
|
public void generateVideoTask() {
|
|
List<ScenicRespVO> scenicList = scenicMapper.list(new ScenicReqQuery());
|
|
if (scenicList.isEmpty()) {
|
|
return;
|
|
}
|
|
|
|
Calendar calendar = Calendar.getInstance();
|
|
calendar.setTime(new Date());
|
|
int currentHour = calendar.get(Calendar.HOUR_OF_DAY);
|
|
calendar.clear();
|
|
scenicList.parallelStream().forEach(scenic -> {
|
|
Long scenicId = scenic.getId();
|
|
ScenicConfigEntity scenicConfig = scenicRepository.getScenicConfig(scenicId);
|
|
if (scenicConfig == null) {
|
|
log.info("当前景区{}，无配置信息", scenic.getName());
|
|
return;
|
|
}
|
|
if (Integer.valueOf(1).equals(scenicConfig.getBookRoutine()) || Integer.valueOf(3).equals(scenicConfig.getBookRoutine())) {
|
|
Integer hour = scenicConfig.getForceFinishTime();
|
|
if (hour != currentHour) {
|
|
return;
|
|
}
|
|
// 定时逻辑
|
|
List<ContentPageVO> contentList = templateMapper.listFor(scenicId);
|
|
if (contentList.isEmpty()) {
|
|
return;
|
|
}
|
|
FaceReqQuery query = new FaceReqQuery();
|
|
query.setScenicId(scenicId);
|
|
query.setStartTime(DateUtil.beginOfDay(new Date()));
|
|
query.setEndTime(DateUtil.endOfDay(new Date()));
|
|
List<FaceRespVO> list = faceMapper.list(query);
|
|
list.stream().parallel().forEach(face -> {
|
|
faceService.matchFaceId(face.getId(), false);
|
|
if (Integer.valueOf(3).equals(scenicConfig.getBookRoutine())) {
|
|
// 全部生成
|
|
contentList.forEach(content -> {
|
|
Long templateId = content.getTemplateId();
|
|
boolean canAutoGenerate = templateBiz.determineTemplateCanAutoGenerate(templateId, face.getId(), false);
|
|
if (canAutoGenerate) {
|
|
log.info("task callback: 自动生成");
|
|
taskTaskService.forceCreateTaskByFaceIdAndTempalteId(face.getId(), templateId);
|
|
} else {
|
|
log.info("task callback: 不自动生成");
|
|
}
|
|
});
|
|
} else {
|
|
Long templateId = contentList.get(0).getTemplateId();
|
|
boolean canAutoGenerate = templateBiz.determineTemplateCanAutoGenerate(templateId, face.getId(), false);
|
|
if (canAutoGenerate) {
|
|
log.info("task callback: 自动生成");
|
|
taskTaskService.forceCreateTaskByFaceIdAndTempalteId(face.getId(), templateId);
|
|
} else {
|
|
log.info("task callback: 不自动生成");
|
|
}
|
|
}
|
|
});
|
|
}
|
|
});
|
|
}
|
|
}
|