Examples with ReturnTaskDisplayVO com.dtstack.taier.develop.vo.schedule.ReturnTaskDisplayVO used on opensource projects

Example 1 with ReturnTaskDisplayVO

use of com.dtstack.taier.develop.vo.schedule.ReturnTaskDisplayVO in project Taier by DTStack.

the class TaskTaskService method displayAllFlowSubTasks.

/**
 * 查询所有工作流节点
 *
 * @param taskShade 工作流任务
 * @return
 */
private ReturnTaskDisplayVO displayAllFlowSubTasks(ScheduleTaskShade taskShade) {
    ReturnTaskDisplayVO vo = new ReturnTaskDisplayVO();
    // 头节点
    TaskNodeVO root = new TaskNodeVO();
    setNode(taskShade, root);
    vo.setRootTaskNode(root);
    // 查询出所有工作流任务
    List<ScheduleTaskShade> taskShadeList = taskService.findAllFlowTasks(taskShade.getTaskId());
    if (CollectionUtils.isEmpty(taskShadeList)) {
        return vo;
    }
    // 查询出这些工作流的关系数据
    Map<Long, ScheduleTaskShade> taskMaps = taskShadeList.stream().collect(Collectors.toMap(ScheduleTaskShade::getTaskId, g -> (g)));
    Set<Long> taskSet = Sets.newHashSet(taskMaps.keySet());
    taskSet.add(taskShade.getTaskId());
    List<ScheduleTaskTaskShade> taskTaskShades = this.lambdaQuery().in(ScheduleTaskTaskShade::getTaskId, taskSet).eq(ScheduleTaskTaskShade::getIsDeleted, Deleted.NORMAL.getStatus()).list();
    if (CollectionUtils.isEmpty(taskTaskShades)) {
        return vo;
    }
    // 递归工作流任务的关系
    Map<Long, List<ScheduleTaskTaskShade>> taskTaskMap = taskTaskShades.stream().collect(Collectors.groupingBy(ScheduleTaskTaskShade::getParentTaskId));
    root.setChildNode(findChildNode(root, taskMaps, taskTaskMap));
    return vo;
}

Example 2 with ReturnTaskDisplayVO

use of com.dtstack.taier.develop.vo.schedule.ReturnTaskDisplayVO in project Taier by DTStack.

the class TaskTaskService method displayOffSpring.

/**
 * 展开任务上下游
 * @return 上下游规则
 */
public ReturnTaskDisplayVO displayOffSpring(QueryTaskDisplayDTO dto) {
    // 查询的最长层级不能超过 max.jobJob.level
    dto.setLevel(JobUtils.checkLevel(dto.getLevel(), context.getMaxLevel()));
    // 查询任务
    ScheduleTaskShade taskShade = taskService.lambdaQuery().eq(ScheduleTaskShade::getTaskId, dto.getTaskId()).eq(ScheduleTaskShade::getIsDeleted, Deleted.NORMAL.getStatus()).one();
    if (taskShade == null) {
        return null;
    }
    // 创建节点
    TaskNodeVO rootNode = new TaskNodeVO();
    setNode(taskShade, rootNode);
    Tenant tenant = tenantService.getTenantById(taskShade.getTenantId());
    if (null != tenant) {
        rootNode.setTenantName(tenant.getTenantName());
        rootNode.setTenantId(tenant.getId());
    }
    String userName = userService.getUserName(taskShade.getCreateUserId());
    rootNode.setOperatorId(taskShade.getCreateUserId());
    rootNode.setOperatorName(userName);
    if (DisplayDirect.CHILD.getType().equals(dto.getDirectType())) {
        rootNode.setChildNode(displayLevelNode(taskShade, dto.getLevel(), dto.getDirectType()));
    } else {
        rootNode.setParentNode(displayLevelNode(taskShade, dto.getLevel(), dto.getDirectType()));
    }
    ReturnTaskDisplayVO vo = new ReturnTaskDisplayVO();
    vo.setDirectType(dto.getDirectType());
    vo.setRootTaskNode(rootNode);
    return vo;
}