| | |
| | | import cn.iocoder.yudao.framework.common.pojo.PageResult; |
| | | import cn.iocoder.yudao.framework.common.util.collection.MapUtils; |
| | | import cn.iocoder.yudao.framework.common.util.object.BeanUtils; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.route.vo.process.MesProProcessBomItemVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.process.MesProWorkOrderProcessPageReqVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.process.MesProWorkOrderProcessRespVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.workorder.vo.process.MesProWorkOrderProcessSaveReqVO; |
| | | import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemDO; |
| | | import cn.iocoder.yudao.module.mes.dal.dataobject.pro.process.MesProProcessDO; |
| | | import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workorder.MesProWorkOrderBomDO; |
| | | import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workorder.MesProWorkOrderProcessDO; |
| | | import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemService; |
| | | import cn.iocoder.yudao.module.mes.service.pro.process.MesProProcessService; |
| | | import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderBomService; |
| | | import cn.iocoder.yudao.module.mes.service.pro.workorder.MesProWorkOrderProcessService; |
| | | import io.swagger.v3.oas.annotations.Operation; |
| | | import io.swagger.v3.oas.annotations.Parameter; |
| | |
| | | |
| | | @Resource |
| | | private MesProWorkOrderProcessService workOrderProcessService; |
| | | @Resource |
| | | private MesProWorkOrderBomService workOrderBomService; |
| | | @Resource |
| | | private MesProProcessService processService; |
| | | @Resource |
| | |
| | | Map<Long, MesMdItemDO> itemMap = CollUtil.isNotEmpty(outputItemIds) |
| | | ? itemService.getItemMap(outputItemIds) : Collections.emptyMap(); |
| | | |
| | | // 2. 拼接 VO |
| | | return BeanUtils.toBean(list, MesProWorkOrderProcessRespVO.class, vo -> { |
| | | // 2. 查询投料BOM(按工序分组) |
| | | Set<Long> workOrderIds = convertSet(list, MesProWorkOrderProcessDO::getWorkOrderId); |
| | | workOrderIds.removeAll(Collections.singleton(null)); |
| | | Map<Long, List<MesProProcessBomItemVO>> bomItemsByProcessId = new HashMap<>(); |
| | | Set<Long> allBomItemIds = new HashSet<>(); |
| | | for (Long woId : workOrderIds) { |
| | | List<MesProWorkOrderBomDO> bomList = workOrderBomService.getWorkOrderBomListByWorkOrderId(woId); |
| | | if (CollUtil.isNotEmpty(bomList)) { |
| | | allBomItemIds.addAll(convertSet(bomList, MesProWorkOrderBomDO::getItemId)); |
| | | for (MesProWorkOrderBomDO bom : bomList) { |
| | | bomItemsByProcessId.computeIfAbsent(bom.getProcessId(), k -> new ArrayList<>()) |
| | | .add(BeanUtils.toBean(bom, MesProProcessBomItemVO.class)); |
| | | } |
| | | } |
| | | } |
| | | // 填充投料物料编码和名称 |
| | | Map<Long, MesMdItemDO> bomItemMap = CollUtil.isNotEmpty(allBomItemIds) |
| | | ? itemService.getItemMap(allBomItemIds) : Collections.emptyMap(); |
| | | bomItemsByProcessId.values().forEach(vos -> vos.forEach(vo -> { |
| | | MapUtils.findAndThen(bomItemMap, vo.getItemId(), item -> { |
| | | vo.setItemCode(item.getCode()).setItemName(item.getName()); |
| | | }); |
| | | })); |
| | | |
| | | // 3. 拼接 VO |
| | | List<MesProWorkOrderProcessRespVO> result = BeanUtils.toBean(list, MesProWorkOrderProcessRespVO.class, vo -> { |
| | | MapUtils.findAndThen(processMap, vo.getProcessId(), process -> { |
| | | vo.setProcessCode(process.getCode()).setProcessName(process.getName()); |
| | | }); |
| | |
| | | vo.setOutputItemCode(item.getCode()).setOutputItemName(item.getName()); |
| | | }); |
| | | }); |
| | | // 设置每个工序各自的投料BOM |
| | | result.forEach(vo -> vo.setBomItems(bomItemsByProcessId.getOrDefault(vo.getProcessId(), Collections.emptyList()))); |
| | | return result; |
| | | } |
| | | |
| | | } |