Examples with ScheduleParam org.apache.dolphinscheduler.api.dto.ScheduleParam used on opensource projects

Example 1 with ScheduleParam

use of org.apache.dolphinscheduler.api.dto.ScheduleParam in project dolphinscheduler by apache.

the class SchedulerService method insertSchedule.

/**
 * save schedule
 *
 * @param loginUser login user
 * @param projectName project name
 * @param processDefineId process definition id
 * @param schedule scheduler
 * @param warningType  warning type
 * @param warningGroupId warning group id
 * @param failureStrategy failure strategy
 * @param processInstancePriority process instance priority
 * @param receivers receivers
 * @param receiversCc receivers cc
 * @param workerGroup worker group
 * @return create result code
 * @throws IOException ioexception
 */
@Transactional(rollbackFor = Exception.class)
public Map<String, Object> insertSchedule(User loginUser, String projectName, Integer processDefineId, String schedule, WarningType warningType, int warningGroupId, FailureStrategy failureStrategy, String receivers, String receiversCc, Priority processInstancePriority, String workerGroup) throws IOException {
    Map<String, Object> result = new HashMap<String, Object>(5);
    Project project = projectMapper.queryByName(projectName);
    // check project auth
    boolean hasProjectAndPerm = projectService.hasProjectAndPerm(loginUser, project, result);
    if (!hasProjectAndPerm) {
        return result;
    }
    // check work flow define release state
    ProcessDefinition processDefinition = processService.findProcessDefineById(processDefineId);
    result = executorService.checkProcessDefinitionValid(processDefinition, processDefineId);
    if (result.get(Constants.STATUS) != Status.SUCCESS) {
        return result;
    }
    Schedule scheduleObj = new Schedule();
    Date now = new Date();
    scheduleObj.setProjectName(projectName);
    scheduleObj.setProcessDefinitionId(processDefinition.getId());
    scheduleObj.setProcessDefinitionName(processDefinition.getName());
    ScheduleParam scheduleParam = JSONUtils.parseObject(schedule, ScheduleParam.class);
    if (DateUtils.differSec(scheduleParam.getStartTime(), scheduleParam.getEndTime()) == 0) {
        logger.warn("The start time must not be the same as the end");
        putMsg(result, Status.SCHEDULE_START_TIME_END_TIME_SAME);
        return result;
    }
    scheduleObj.setStartTime(scheduleParam.getStartTime());
    scheduleObj.setEndTime(scheduleParam.getEndTime());
    if (!org.quartz.CronExpression.isValidExpression(scheduleParam.getCrontab())) {
        logger.error(scheduleParam.getCrontab() + " verify failure");
        putMsg(result, Status.REQUEST_PARAMS_NOT_VALID_ERROR, scheduleParam.getCrontab());
        return result;
    }
    scheduleObj.setCrontab(scheduleParam.getCrontab());
    scheduleObj.setWarningType(warningType);
    scheduleObj.setWarningGroupId(warningGroupId);
    scheduleObj.setFailureStrategy(failureStrategy);
    scheduleObj.setCreateTime(now);
    scheduleObj.setUpdateTime(now);
    scheduleObj.setUserId(loginUser.getId());
    scheduleObj.setUserName(loginUser.getUserName());
    scheduleObj.setReleaseState(ReleaseState.OFFLINE);
    scheduleObj.setProcessInstancePriority(processInstancePriority);
    scheduleObj.setWorkerGroup(workerGroup);
    scheduleMapper.insert(scheduleObj);
    /**
     * updateProcessInstance receivers and cc by process definition id
     */
    processDefinition.setReceivers(receivers);
    processDefinition.setReceiversCc(receiversCc);
    processDefinitionMapper.updateById(processDefinition);
    putMsg(result, Status.SUCCESS);
    result.put("scheduleId", scheduleObj.getId());
    return result;
}

Example 2 with ScheduleParam

use of org.apache.dolphinscheduler.api.dto.ScheduleParam in project dolphinscheduler by apache.

the class SchedulerService method updateSchedule.

/**
 * updateProcessInstance schedule
 *
 * @param loginUser login user
 * @param projectName project name
 * @param id scheduler id
 * @param scheduleExpression scheduler
 * @param warningType warning type
 * @param warningGroupId warning group id
 * @param failureStrategy failure strategy
 * @param workerGroup worker group
 * @param processInstancePriority process instance priority
 * @param receiversCc receiver cc
 * @param receivers receivers
 * @param scheduleStatus schedule status
 * @return update result code
 * @throws IOException ioexception
 */
@Transactional(rollbackFor = Exception.class)
public Map<String, Object> updateSchedule(User loginUser, String projectName, Integer id, String scheduleExpression, WarningType warningType, int warningGroupId, FailureStrategy failureStrategy, String receivers, String receiversCc, ReleaseState scheduleStatus, Priority processInstancePriority, String workerGroup) throws IOException {
    Map<String, Object> result = new HashMap<String, Object>(5);
    Project project = projectMapper.queryByName(projectName);
    // check project auth
    boolean hasProjectAndPerm = projectService.hasProjectAndPerm(loginUser, project, result);
    if (!hasProjectAndPerm) {
        return result;
    }
    // check schedule exists
    Schedule schedule = scheduleMapper.selectById(id);
    if (schedule == null) {
        putMsg(result, Status.SCHEDULE_CRON_NOT_EXISTS, id);
        return result;
    }
    ProcessDefinition processDefinition = processService.findProcessDefineById(schedule.getProcessDefinitionId());
    if (processDefinition == null) {
        putMsg(result, Status.PROCESS_DEFINE_NOT_EXIST, schedule.getProcessDefinitionId());
        return result;
    }
    /**
     * scheduling on-line status forbid modification
     */
    if (checkValid(result, schedule.getReleaseState() == ReleaseState.ONLINE, Status.SCHEDULE_CRON_ONLINE_FORBID_UPDATE)) {
        return result;
    }
    Date now = new Date();
    // updateProcessInstance param
    if (StringUtils.isNotEmpty(scheduleExpression)) {
        ScheduleParam scheduleParam = JSONUtils.parseObject(scheduleExpression, ScheduleParam.class);
        if (DateUtils.differSec(scheduleParam.getStartTime(), scheduleParam.getEndTime()) == 0) {
            logger.warn("The start time must not be the same as the end");
            putMsg(result, Status.SCHEDULE_START_TIME_END_TIME_SAME);
            return result;
        }
        schedule.setStartTime(scheduleParam.getStartTime());
        schedule.setEndTime(scheduleParam.getEndTime());
        if (!org.quartz.CronExpression.isValidExpression(scheduleParam.getCrontab())) {
            putMsg(result, Status.SCHEDULE_CRON_CHECK_FAILED, scheduleParam.getCrontab());
            return result;
        }
        schedule.setCrontab(scheduleParam.getCrontab());
    }
    if (warningType != null) {
        schedule.setWarningType(warningType);
    }
    schedule.setWarningGroupId(warningGroupId);
    if (failureStrategy != null) {
        schedule.setFailureStrategy(failureStrategy);
    }
    if (scheduleStatus != null) {
        schedule.setReleaseState(scheduleStatus);
    }
    schedule.setWorkerGroup(workerGroup);
    schedule.setUpdateTime(now);
    schedule.setProcessInstancePriority(processInstancePriority);
    scheduleMapper.updateById(schedule);
    /**
     * updateProcessInstance recipients and cc by process definition ID
     */
    processDefinition.setReceivers(receivers);
    processDefinition.setReceiversCc(receiversCc);
    processDefinitionMapper.updateById(processDefinition);
    putMsg(result, Status.SUCCESS);
    return result;
}

Example 3 with ScheduleParam

use of org.apache.dolphinscheduler.api.dto.ScheduleParam in project dolphinscheduler by apache.

the class SchedulerService method previewSchedule.

/**
 * preview schedule
 *
 * @param loginUser login user
 * @param projectName project name
 * @param schedule schedule expression
 * @return the next five fire time
 */
public Map<String, Object> previewSchedule(User loginUser, String projectName, String schedule) {
    Map<String, Object> result = new HashMap<>(5);
    CronExpression cronExpression;
    ScheduleParam scheduleParam = JSONUtils.parseObject(schedule, ScheduleParam.class);
    Date now = new Date();
    Date startTime = now.after(scheduleParam.getStartTime()) ? now : scheduleParam.getStartTime();
    Date endTime = scheduleParam.getEndTime();
    try {
        cronExpression = CronUtils.parse2CronExpression(scheduleParam.getCrontab());
    } catch (ParseException e) {
        logger.error(e.getMessage(), e);
        putMsg(result, Status.PARSE_TO_CRON_EXPRESSION_ERROR);
        return result;
    }
    List<Date> selfFireDateList = CronUtils.getSelfFireDateList(startTime, endTime, cronExpression, Constants.PREVIEW_SCHEDULE_EXECUTE_COUNT);
    result.put(Constants.DATA_LIST, selfFireDateList.stream().map(t -> DateUtils.dateToString(t)));
    putMsg(result, Status.SUCCESS);
    return result;
}