| | |
| | | |
| | | import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; |
| | | import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.qc.template.vo.MesQcTemplateImportExcelVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.qc.template.vo.MesQcTemplateImportRespVO; |
| | | import cn.hutool.core.util.StrUtil; |
| | | import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; |
| | | import java.util.ArrayList; |
| | | import java.util.LinkedHashMap; |
| | | import java.util.concurrent.atomic.AtomicInteger; |
| | | |
| | | /** |
| | | * MES 质检方案 Service 实现类 |
| | |
| | | return templateMapper.selectByIds(ids); |
| | | } |
| | | |
| | | |
| | | @Override |
| | | public MesQcTemplateImportRespVO importTemplateList(List<MesQcTemplateImportExcelVO> importList, boolean updateSupport) { |
| | | if (CollUtil.isEmpty(importList)) { |
| | | throw exception(QC_TEMPLATE_IMPORT_LIST_IS_EMPTY); |
| | | } |
| | | MesQcTemplateImportRespVO respVO = MesQcTemplateImportRespVO.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; |
| | | } |
| | | if (StrUtil.isBlank(importItem.getTypes())) { |
| | | respVO.getFailureCodes().put(key, "检测种类不能为空"); |
| | | return; |
| | | } |
| | | List<Integer> types; |
| | | try { |
| | | types = StrUtil.split(importItem.getTypes(), ',').stream() |
| | | .map(s -> Integer.valueOf(s.trim())).collect(java.util.stream.Collectors.toList()); |
| | | } catch (NumberFormatException e) { |
| | | respVO.getFailureCodes().put(key, "检测种类格式不正确"); |
| | | return; |
| | | } |
| | | Integer status = ObjUtil.defaultIfNull(importItem.getStatus(), CommonStatusEnum.ENABLE.getStatus()); |
| | | MesQcTemplateDO exist = templateMapper.selectByCode(importItem.getCode()); |
| | | if (exist == null) { |
| | | MesQcTemplateDO template = BeanUtils.toBean(importItem, MesQcTemplateDO.class); |
| | | template.setId(null); |
| | | template.setTypes(types); |
| | | template.setStatus(status); |
| | | templateMapper.insert(template); |
| | | respVO.getCreateCodes().add(importItem.getCode()); |
| | | } else if (updateSupport) { |
| | | MesQcTemplateDO updateObj = BeanUtils.toBean(importItem, MesQcTemplateDO.class); |
| | | updateObj.setId(exist.getId()); |
| | | updateObj.setCode(exist.getCode()); |
| | | updateObj.setTypes(types); |
| | | updateObj.setStatus(status); |
| | | templateMapper.updateById(updateObj); |
| | | respVO.getUpdateCodes().add(importItem.getCode()); |
| | | } else { |
| | | respVO.getFailureCodes().put(key, "方案编号已存在"); |
| | | } |
| | | }); |
| | | return respVO; |
| | | } |
| | | |
| | | } |
| | | |