| | |
| | | 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; |
| | |
| | | import org.springframework.web.bind.annotation.*; |
| | | |
| | | import java.util.*; |
| | | import java.util.stream.Collectors; |
| | | |
| | | import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; |
| | | import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet; |
| | |
| | | |
| | | @Resource |
| | | private MesProWorkOrderProcessService workOrderProcessService; |
| | | @Resource |
| | | private MesProWorkOrderBomService workOrderBomService; |
| | | @Resource |
| | | private MesProProcessService processService; |
| | | @Resource |
| | |
| | | processIds.addAll(nextProcessIds); |
| | | Map<Long, MesProProcessDO> processMap = CollUtil.isNotEmpty(processIds) |
| | | ? processService.getProcessMap(processIds) : Collections.emptyMap(); |
| | | // 获取产出物料 |
| | | Set<Long> outputItemIds = convertSet(list, MesProWorkOrderProcessDO::getOutputItemId); |
| | | outputItemIds.removeAll(Collections.singleton(null)); |
| | | Map<Long, MesMdItemDO> itemMap = CollUtil.isNotEmpty(outputItemIds) |
| | | ? itemService.getItemMap(outputItemIds) : Collections.emptyMap(); |
| | | // 获取产出物料(解析逗号分隔的字符串) |
| | | Set<Long> allOutputItemIds = new HashSet<>(); |
| | | Map<Long, List<Long>> voOutputItemIdsMap = new HashMap<>(); |
| | | for (int i = 0; i < list.size(); i++) { |
| | | String outputItemIdStr = list.get(i).getOutputItemId(); |
| | | if (outputItemIdStr != null && !outputItemIdStr.isBlank()) { |
| | | List<Long> ids = Arrays.stream(outputItemIdStr.split(",")) |
| | | .map(String::trim).filter(s -> !s.isEmpty()) |
| | | .map(Long::valueOf).collect(Collectors.toList()); |
| | | voOutputItemIdsMap.put((long) i, ids); |
| | | allOutputItemIds.addAll(ids); |
| | | } |
| | | } |
| | | Map<Long, MesMdItemDO> itemMap = CollUtil.isNotEmpty(allOutputItemIds) |
| | | ? itemService.getItemMap(allOutputItemIds) : 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); |
| | | for (int i = 0; i < result.size(); i++) { |
| | | MesProWorkOrderProcessRespVO vo = result.get(i); |
| | | MapUtils.findAndThen(processMap, vo.getProcessId(), process -> { |
| | | vo.setProcessCode(process.getCode()).setProcessName(process.getName()); |
| | | }); |
| | | MapUtils.findAndThen(processMap, vo.getNextProcessId(), process -> { |
| | | vo.setNextProcessName(process.getName()); |
| | | }); |
| | | MapUtils.findAndThen(itemMap, vo.getOutputItemId(), item -> { |
| | | vo.setOutputItemCode(item.getCode()).setOutputItemName(item.getName()); |
| | | }); |
| | | }); |
| | | // 产出物料(支持多选) |
| | | List<Long> ids = voOutputItemIdsMap.getOrDefault((long) i, Collections.emptyList()); |
| | | if (!ids.isEmpty()) { |
| | | vo.setOutputItemIds(ids); |
| | | List<String> names = ids.stream() |
| | | .map(id -> { |
| | | MesMdItemDO item = itemMap.get(id); |
| | | return item != null ? item.getName() : String.valueOf(id); |
| | | }).collect(Collectors.toList()); |
| | | vo.setOutputItemNames(String.join(", ", names)); |
| | | // 向后兼容:第一个产出物料 |
| | | Long firstId = ids.get(0); |
| | | vo.setOutputItemId(firstId); |
| | | MesMdItemDO firstItem = itemMap.get(firstId); |
| | | if (firstItem != null) { |
| | | vo.setOutputItemCode(firstItem.getCode()); |
| | | vo.setOutputItemName(firstItem.getName()); |
| | | } |
| | | } |
| | | } |
| | | // 设置每个工序各自的投料BOM |
| | | result.forEach(vo -> vo.setBomItems(bomItemsByProcessId.getOrDefault(vo.getProcessId(), Collections.emptyList()))); |
| | | return result; |
| | | } |
| | | |
| | | } |