mom-pro2-after.git

			@@ -2,17 +2,23 @@

			import cn.hutool.core.collection.CollUtil;
			import cn.hutool.core.util.ObjUtil;
			import cn.hutool.core.util.StrUtil;
			import cn.iocoder.yudao.framework.common.pojo.PageResult;
			import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
			import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
			import cn.iocoder.yudao.module.mes.controller.admin.md.workstation.vo.MesMdWorkstationImportExcelVO;
			import cn.iocoder.yudao.module.mes.controller.admin.md.workstation.vo.MesMdWorkstationImportRespVO;
			import cn.iocoder.yudao.module.mes.controller.admin.md.workstation.vo.MesMdWorkstationPageReqVO;
			import cn.iocoder.yudao.module.mes.controller.admin.md.workstation.vo.MesMdWorkstationSaveReqVO;
			import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkshopDO;
			import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkstationDO;
			import cn.iocoder.yudao.module.mes.dal.dataobject.pro.process.MesProProcessDO;
			import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseAreaDO;
			import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseDO;
			import cn.iocoder.yudao.module.mes.dal.dataobject.wm.warehouse.MesWmWarehouseLocationDO;
			import cn.iocoder.yudao.module.mes.dal.mysql.md.workstation.MesMdWorkshopMapper;
			import cn.iocoder.yudao.module.mes.dal.mysql.md.workstation.MesMdWorkstationMapper;
			import cn.iocoder.yudao.module.mes.dal.mysql.pro.process.MesProProcessMapper;
			import cn.iocoder.yudao.module.mes.enums.wm.BarcodeBizTypeEnum;
			import cn.iocoder.yudao.module.mes.service.wm.barcode.MesWmBarcodeService;
			import cn.iocoder.yudao.module.mes.service.wm.warehouse.MesWmWarehouseAreaService;
			@@ -25,9 +31,12 @@
			import org.springframework.transaction.annotation.Transactional;
			import org.springframework.validation.annotation.Validated;

			import java.util.ArrayList;
			import java.util.Collection;
			import java.util.Collections;
			import java.util.LinkedHashMap;
			import java.util.List;
			import java.util.concurrent.atomic.AtomicInteger;

			import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
			import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;
			@@ -53,6 +62,10 @@
			private MesMdWorkstationWorkerService workstationWorkerService;
			@Resource
			private MesMdWorkshopService workshopService;
			@Resource
			private MesMdWorkshopMapper workshopMapper;
			@Resource
			private MesProProcessMapper processMapper;
			@Resource
			private MesWmWarehouseService warehouseService;
			@Resource
			@@ -271,4 +284,79 @@
			return workstationMapper.selectByIds(ids);
			}

			@Override
			@Transactional(rollbackFor = Exception.class)
			public MesMdWorkstationImportRespVO importWorkstationList(List<MesMdWorkstationImportExcelVO> importList,
			boolean updateSupport) {
			// 1. 校验非空
			if (CollUtil.isEmpty(importList)) {
			throw exception(MD_WORKSTATION_IMPORT_LIST_IS_EMPTY);
			}

			// 2. 逐行处理
			MesMdWorkstationImportRespVO respVO = MesMdWorkstationImportRespVO.builder()
			.createCodes(new ArrayList<>()).updateCodes(new ArrayList<>())
			.failureCodes(new LinkedHashMap<>()).build();
			AtomicInteger index = new AtomicInteger(1);
			importList.forEach(importItem -> {
			int currentIndex = index.getAndIncrement();
			String key = StrUtil.blankToDefault(importItem.getCode(), "第 " + currentIndex + " 行");
			if (StrUtil.isBlank(importItem.getCode())) {
			respVO.getFailureCodes().put(key, "工作站编码不能为空");
			return;
			}
			if (StrUtil.isBlank(importItem.getName())) {
			respVO.getFailureCodes().put(key, "工作站名称不能为空");
			return;
			}
			// 2.1 解析车间
			if (StrUtil.isBlank(importItem.getWorkshopCode())) {
			respVO.getFailureCodes().put(key, "所在车间编码不能为空");
			return;
			}
			MesMdWorkshopDO workshop = workshopMapper.selectByCode(importItem.getWorkshopCode());
			if (workshop == null) {
			respVO.getFailureCodes().put(key, "所在车间编码[" + importItem.getWorkshopCode() + "]不存在");
			return;
			}
			// 2.2 解析工序
			if (StrUtil.isBlank(importItem.getProcessCode())) {
			respVO.getFailureCodes().put(key, "工序编码不能为空");
			return;
			}
			MesProProcessDO process = processMapper.selectByCode(importItem.getProcessCode());
			if (process == null) {
			respVO.getFailureCodes().put(key, "工序编码[" + importItem.getProcessCode() + "]不存在");
			return;
			}
			// 2.3 判断：创建 or 更新
			MesMdWorkstationDO existWorkstation = workstationMapper.selectByCode(importItem.getCode());
			if (existWorkstation == null) {
			// 2.3.1 新增（复用 create 的仓库层级处理与条码生成逻辑）
			MesMdWorkstationSaveReqVO reqVO = BeanUtils.toBean(importItem, MesMdWorkstationSaveReqVO.class);
			reqVO.setWorkshopId(workshop.getId());
			reqVO.setProcessId(process.getId());
			reqVO.setStatus(ObjUtil.defaultIfNull(importItem.getStatus(), CommonStatusEnum.ENABLE.getStatus()));
			handleWarehouseHierarchy(reqVO);
			MesMdWorkstationDO workstation = BeanUtils.toBean(reqVO, MesMdWorkstationDO.class);
			workstationMapper.insert(workstation);
			// 自动生成条码
			barcodeService.autoGenerateBarcode(BarcodeBizTypeEnum.WORKSTATION.getValue(),
			workstation.getId(), workstation.getCode(), workstation.getName());
			respVO.getCreateCodes().add(importItem.getCode());
			} else if (updateSupport) {
			// 2.3.2 更新（保留原有线边库层级）
			MesMdWorkstationDO updateObj = BeanUtils.toBean(importItem, MesMdWorkstationDO.class);
			updateObj.setId(existWorkstation.getId());
			updateObj.setWorkshopId(workshop.getId());
			updateObj.setProcessId(process.getId());
			workstationMapper.updateById(updateObj);
			respVO.getUpdateCodes().add(importItem.getCode());
			} else {
			respVO.getFailureCodes().put(key, "工作站编码已存在");
			}
			});
			return respVO;
			}

			}