package com.ard.sample.wayline.service.impl;
|
|
import com.ard.sample.utils.TimestampUtils;
|
import com.ard.sample.wayline.model.dto.WaylineJobDTO;
|
import com.ard.sample.wayline.model.entity.WaylineJobEntity;
|
import com.ard.sample.wayline.model.enums.WaylineJobStatusEnum;
|
import com.ard.sample.wayline.model.param.CreateJobParam;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
|
import java.time.LocalDateTime;
|
import java.util.List;
|
import java.util.Optional;
|
|
@Slf4j
|
@Component
|
public class WaylineLoopTaskServiceImpl {
|
|
@Autowired
|
private WaylineJobServiceImpl _waylineJobService;
|
@Autowired
|
private TimestampUtils _timestampUtils;
|
|
/**
|
* 定时任务，生成循环任务在有效其内的每天的定时任务
|
* */
|
@Scheduled(cron = "0 0 10 * * ?")
|
public void loopTask() {
|
try {
|
// 执行定时任务逻辑，例如检查并执行已到时间的飞行任务
|
List<WaylineJobEntity> lineList = _waylineJobService.selectList(3L, null, null);
|
for (WaylineJobEntity waylineJobEntity : lineList) {
|
// 获取taskType为3的数据列表，
|
// 判断是否已完成
|
if (waylineJobEntity.getStatus() == WaylineJobStatusEnum.SUCCESS.getVal()) {
|
continue;
|
}
|
// 判断时间是否在有期内,循环任务的逻辑是，开始执行时间+循环次数。
|
Long beginTime = waylineJobEntity.getBeginTime();
|
// 获取循环任务开始的那天的0点0分0秒的时间戳，方便比对时间有效期范围；
|
// 循环任务按每天下发定时任务走，必须要在当次任务执行时间之前执行。
|
Long beginTime_Start = _timestampUtils.getDayStartTimestamp(beginTime);
|
Integer loopSum = waylineJobEntity.getLoopSum();
|
Long endTime = _timestampUtils.plusDaysByMillis(beginTime, loopSum);
|
if (beginTime_Start.compareTo(System.currentTimeMillis()) < 0 && endTime.compareTo(System.currentTimeMillis()) > 0) {
|
// 执行任务逻辑
|
// 如果在有效期内，判断任务是否已存在，如果存在则不生成子任务；新时间组成为：当前时间戳的年月日，主任务时间的时分秒。
|
Long newTime = _timestampUtils.combine(beginTime);
|
List<WaylineJobEntity> jobSum= _waylineJobService.selectList(null, null, waylineJobEntity.getJobId());
|
if(!jobSum.isEmpty() && jobSum.size() >= loopSum)
|
{
|
log.info("循环子任务已经完成，主任务ID：{}。", waylineJobEntity.getJobId());
|
// 循环子任务已经完成，修改主任务状态为已完成，即：status=3.
|
_waylineJobService.updateJob(WaylineJobDTO.builder()
|
.jobId(waylineJobEntity.getJobId())
|
.executeTime(LocalDateTime.now())
|
.status(WaylineJobStatusEnum.SUCCESS.getVal())
|
.completedTime(LocalDateTime.now())
|
.code(0).build());
|
continue;
|
}
|
List<WaylineJobEntity> jobEntities = _waylineJobService.selectList(null, newTime, waylineJobEntity.getJobId());
|
if (!jobEntities.isEmpty()) {
|
log.info("循环子任务已经存在，主任务ID：{}，时间：{}！", waylineJobEntity.getJobId(), _timestampUtils.formatByMillis(newTime));
|
continue;
|
}
|
CreateJobParam param = getCreateJobParam(waylineJobEntity, newTime);
|
Optional<WaylineJobDTO> waylineJobDTO = _waylineJobService.createWaylineJob(param, waylineJobEntity.getUsername(), newTime, newTime);
|
_waylineJobService.publishFlightTaskPrepare(waylineJobDTO);
|
log.info("添加循环子任务成功，任务ID：{}，时间：{}！", waylineJobEntity.getJobId(), _timestampUtils.formatByMillis(newTime));
|
}
|
|
}
|
}
|
catch (Exception e) {
|
e.printStackTrace();
|
log.error("添加循环子任务出现异常，时间：{}！异常原因：{}", _timestampUtils.formatByMillis(System.currentTimeMillis()), e.getMessage());
|
}
|
}
|
|
private CreateJobParam getCreateJobParam(WaylineJobEntity waylineJobEntity, Long newTime) {
|
CreateJobParam param = new CreateJobParam();
|
param.setName(waylineJobEntity.getName() + "-子任务");
|
param.setFileId(waylineJobEntity.getFileId());
|
param.setDockSn(waylineJobEntity.getDockSn());
|
param.setWaylineType(waylineJobEntity.getWaylineType());
|
param.setTaskType(1);
|
param.setExecuteTime(newTime);
|
param.setRthAltitude(waylineJobEntity.getRthAltitude());
|
param.setOutOfControlAction(waylineJobEntity.getOutOfControlAction());
|
param.setParentId(waylineJobEntity.getJobId());
|
param.setLoopSum(null);
|
return param;
|
}
|
}
|
