云
17 小时以前 ac2e2b5af3726ee64dc37c3ebf26989a0346f22a
yudao-module-mes/src/main/java/cn/iocoder/yudao/module/mes/controller/admin/pro/task/MesProTaskController.java
@@ -3,6 +3,7 @@
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
@@ -46,7 +47,9 @@
import org.springframework.web.bind.annotation.*;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.route.MesProRouteProcessDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workorder.MesProWorkOrderProcessDO;
import cn.iocoder.yudao.module.mes.service.pro.route.MesProRouteProcessService;
import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderProcessService;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.feedback.MesProFeedbackDO;
import cn.iocoder.yudao.module.mes.enums.pro.MesProFeedbackStatusEnum;
import cn.iocoder.yudao.module.mes.service.pro.feedback.MesProFeedbackService;
@@ -65,12 +68,31 @@
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.*;
import static cn.iocoder.yudao.framework.common.util.collection.MapUtils.findAndThen;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
@Tag(name = "管理后台 - MES 生产任务")
@RestController
@RequestMapping("/mes/pro/task")
@Validated
public class MesProTaskController {
    @Resource
    private cn.iocoder.yudao.module.mes.service.pro.task.schedule.MesProBatchScheduleService batchScheduleService;
    @PostMapping("/auto-schedule/preview")
    @Operation(summary = "批量一键排产预览")
    @PreAuthorize("@ss.hasPermission('mes:pro-task:create')")
    public CommonResult<cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProBatchSchedulePreviewRespVO> previewBatchSchedule(
            @Valid @RequestBody cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProBatchScheduleReqVO request) {
        return success(batchScheduleService.preview(request));
    }
    @PostMapping("/auto-schedule")
    @Operation(summary = "确认批量一键排产")
    @PreAuthorize("@ss.hasPermission('mes:pro-task:create')")
    public CommonResult<java.util.Map<String, Object>> executeBatchSchedule(
            @Valid @RequestBody cn.iocoder.yudao.module.mes.controller.admin.pro.task.vo.MesProBatchScheduleReqVO request) {
        return success(java.util.Map.of("taskIds", batchScheduleService.execute(request)));
    }
    @Resource
    private MesProTaskService taskService;
@@ -88,6 +110,8 @@
    private MesMdUnitMeasureService unitMeasureService;
    @Resource
    private MesProRouteProcessService routeProcessService;
    @Resource
    private MesProWorkOrderProcessService workOrderProcessService;
    @Resource
    private MesProFeedbackService feedbackService;
    @Resource
@@ -116,7 +140,6 @@
            @Parameter(name = "workstationId", description = "工作站编号", required = true),
            @Parameter(name = "quantity", description = "排产数量", required = true)
    })
    @PreAuthorize("@ss.hasPermission('mes:pro-task:query')")
    public CommonResult<java.util.Map<String, Object>> calculateDuration(
            @RequestParam("workstationId") Long workstationId,
            @RequestParam("quantity") BigDecimal quantity) {
@@ -169,7 +192,6 @@
    @GetMapping("/get")
    @Operation(summary = "获得生产任务")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('mes:pro-task:query')")
    public CommonResult<MesProTaskRespVO> getTask(@RequestParam("id") Long id) {
        MesProTaskDO task = taskService.getTask(id);
        if (task == null) {
@@ -180,7 +202,6 @@
    @GetMapping("/page")
    @Operation(summary = "获得生产任务分页")
    @PreAuthorize("@ss.hasPermission('mes:pro-task:query')")
    public CommonResult<PageResult<MesProTaskRespVO>> getTaskPage(@Valid MesProTaskPageReqVO pageReqVO) {
        PageResult<MesProTaskDO> pageResult = taskService.getTaskPage(pageReqVO);
        return success(new PageResult<>(buildTaskRespVOList(pageResult.getList()), pageResult.getTotal()));
@@ -188,8 +209,10 @@
    @GetMapping("/pending-feedback-page")
    @Operation(summary = "获得待报工台账分页", description = "已排产且仍有待报工数量的任务(状态=进行中)")
    @PreAuthorize("@ss.hasPermission('mes:pro-task:query')")
    public CommonResult<PageResult<MesProTaskRespVO>> getPendingFeedbackPage(@Valid MesProTaskPageReqVO pageReqVO) {
        if (Boolean.TRUE.equals(pageReqVO.getOnlyMine())) {
            pageReqVO.setAssigneeUserId(getLoginUserId());
        }
        PageResult<MesProTaskDO> pageResult = taskService.getPendingFeedbackPage(pageReqVO);
        List<MesProTaskRespVO> respVOList = buildTaskRespVOList(pageResult.getList());
        
@@ -222,9 +245,16 @@
        return success(new PageResult<>(respVOList, pageResult.getTotal()));
    }
    @GetMapping("/pending-count-by-workstation")
    @Operation(summary = "按工作站获得待报工任务数", description = "工作台选择工作站卡片展示,口径同待报工台账")
    public CommonResult<Map<Long, Long>> getPendingFeedbackCountByWorkstation(
            @RequestParam(value = "onlyMine", required = false) Boolean onlyMine) {
        Long assigneeUserId = Boolean.TRUE.equals(onlyMine) ? getLoginUserId() : null;
        return success(taskService.getPendingFeedbackCountByWorkstation(assigneeUserId));
    }
    @GetMapping("/gantt-list")
    @Operation(summary = "获得甘特图任务列表", description = "后端组装工单=project + 任务=task 列表")
    @PreAuthorize("@ss.hasPermission('mes:pro-task:query')")
    public CommonResult<List<GanttDataRespVO>> listGanttTaskList(@Valid MesProWorkOrderPageReqVO reqVO) {
        // 1.1 查询匹配的工单(不分页)
        reqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
@@ -324,7 +354,25 @@
                convertSet(itemMap.values(), MesMdItemDO::getUnitMeasureId));
        Map<Long, CrmCustomerRespDTO> customerMap = customerApi.getCustomerMap(
                convertSet(list, MesProTaskDO::getClientId));
        // 工序的 checkFlag + backflushFlag:批量查询后构建 routeId -> processId -> flag 的双层 Map
        Map<Long, AdminUserRespDTO> assigneeUserMap = adminUserApi.getUserMap(
                list.stream().flatMap(task -> parseUserIds(task.getAssigneeUserIds()).stream())
                        .collect(java.util.stream.Collectors.toSet()));
        // 工序的 checkFlag + backflushFlag:工单工序快照优先,工艺路线兜底
        Set<Long> workOrderIds = convertSet(list, MesProTaskDO::getWorkOrderId);
        Map<Long, Map<Long, Boolean>> workOrderProcessCheckFlagMap = new HashMap<>();
        Map<Long, Map<Long, Boolean>> workOrderProcessBackflushFlagMap = new HashMap<>();
        if (CollUtil.isNotEmpty(workOrderIds)) {
            List<MesProWorkOrderProcessDO> allWorkOrderProcesses =
                    workOrderProcessService.getWorkOrderProcessListByWorkOrderIds(workOrderIds);
            for (MesProWorkOrderProcessDO wp : allWorkOrderProcesses) {
                workOrderProcessCheckFlagMap
                        .computeIfAbsent(wp.getWorkOrderId(), k -> new HashMap<>())
                        .put(wp.getProcessId(), Boolean.TRUE.equals(wp.getCheckFlag()));
                workOrderProcessBackflushFlagMap
                        .computeIfAbsent(wp.getWorkOrderId(), k -> new HashMap<>())
                        .put(wp.getProcessId(), Boolean.TRUE.equals(wp.getBackflushFlag()));
            }
        }
        Set<Long> routeIds = convertSet(list, MesProTaskDO::getRouteId);
        Map<Long, Map<Long, Boolean>> routeProcessCheckFlagMap = new HashMap<>();
        Map<Long, Map<Long, Boolean>> routeProcessBackflushFlagMap = new HashMap<>();
@@ -362,6 +410,14 @@
            BigDecimal inTransit = inTransitMap.getOrDefault(task.getId(), BigDecimal.ZERO);
            vo.setInTransitQuantity(inTransit);
            vo.setReportableQuantity(quantity.subtract(produced).subtract(inTransit).max(BigDecimal.ZERO));
            List<Long> assigneeUserIds = parseUserIds(task.getAssigneeUserIds());
            vo.setAssigneeUserIds(assigneeUserIds);
            if (CollUtil.isNotEmpty(assigneeUserIds)) {
                vo.setAssigneeUserNames(assigneeUserIds.stream()
                        .map(assigneeUserMap::get).filter(Objects::nonNull)
                        .map(AdminUserRespDTO::getNickname)
                        .collect(java.util.stream.Collectors.joining("、")));
            }
            findAndThen(workOrderMap, task.getWorkOrderId(), wo ->
                    vo.setWorkOrderCode(wo.getCode()).setWorkOrderName(wo.getName()).setRequestDate(wo.getRequestDate()));
            findAndThen(workstationMap, task.getWorkstationId(), ws ->
@@ -375,15 +431,51 @@
            });
            findAndThen(customerMap, task.getClientId(), customer ->
                    vo.setClientName(customer.getName()));
            findAndThen(routeProcessCheckFlagMap, task.getRouteId(), processCheckMap ->
                    findAndThen(processCheckMap, task.getProcessId(), vo::setCheckFlag));
            findAndThen(routeProcessBackflushFlagMap, task.getRouteId(), processBackflushMap ->
                    findAndThen(processBackflushMap, task.getProcessId(), vo::setBackflushFlag));
            Boolean checkFlag = resolveProcessFlag(task, workOrderProcessCheckFlagMap, routeProcessCheckFlagMap);
            if (checkFlag != null) vo.setCheckFlag(checkFlag);
            Boolean backflushFlag = resolveProcessFlag(task, workOrderProcessBackflushFlagMap, routeProcessBackflushFlagMap);
            if (backflushFlag != null) vo.setBackflushFlag(backflushFlag);
            return vo;
        });
    }
    /**
     * 解析任务工序的标识:工单工序快照优先,工艺路线兜底
     */
    private Boolean resolveProcessFlag(MesProTaskDO task,
                                       Map<Long, Map<Long, Boolean>> workOrderProcessMap,
                                       Map<Long, Map<Long, Boolean>> routeProcessMap) {
        Map<Long, Boolean> workOrderFlags = workOrderProcessMap.get(task.getWorkOrderId());
        Boolean flag = workOrderFlags != null ? workOrderFlags.get(task.getProcessId()) : null;
        if (flag != null) {
            return flag;
        }
        Map<Long, Boolean> routeFlags = routeProcessMap.get(task.getRouteId());
        return routeFlags != null ? routeFlags.get(task.getProcessId()) : null;
    }
    /**
     * 解析逗号分隔的用户编号串
     */
    private static List<Long> parseUserIds(String csv) {
        if (StrUtil.isBlank(csv)) {
            return Collections.emptyList();
        }
        List<Long> userIds = new ArrayList<>();
        for (String part : csv.split(",")) {
            String trimmed = part.trim();
            if (trimmed.isEmpty()) {
                continue;
            }
            try {
                userIds.add(Long.valueOf(trimmed));
            } catch (NumberFormatException ignored) {
            }
        }
        return userIds;
    }
    /**
     * 拼接甘特图显示文本,格式:"[产品名][数量][单位]"
     */
    private String buildGanttText(MesMdItemDO item, BigDecimal quantity,