From 21811bad262b3f774ca7c12a68e0bf64a8321501 Mon Sep 17 00:00:00 2001 From: baoshiwei <baoshiwei@shlanbao.cn> Date: 星期四, 29 五月 2025 10:08:15 +0800 Subject: [PATCH] feat(eims): 支持多文件导入点检计划 --- eims/ruoyi-modules/lb-eims/src/main/java/org/dromara/eims/service/impl/EimsInspectPlanServiceImpl.java | 89 ++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 85 insertions(+), 4 deletions(-) diff --git a/eims/ruoyi-modules/lb-eims/src/main/java/org/dromara/eims/service/impl/EimsInspectPlanServiceImpl.java b/eims/ruoyi-modules/lb-eims/src/main/java/org/dromara/eims/service/impl/EimsInspectPlanServiceImpl.java index 865f6de..ab7a0a9 100644 --- a/eims/ruoyi-modules/lb-eims/src/main/java/org/dromara/eims/service/impl/EimsInspectPlanServiceImpl.java +++ b/eims/ruoyi-modules/lb-eims/src/main/java/org/dromara/eims/service/impl/EimsInspectPlanServiceImpl.java @@ -1,6 +1,9 @@ package org.dromara.eims.service.impl; +import com.alibaba.excel.EasyExcel; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import org.dromara.common.core.exception.ServiceException; +import org.dromara.common.core.utils.DateUtils; import org.dromara.common.core.utils.MapstructUtils; import org.dromara.common.core.utils.StringUtils; import org.dromara.common.mybatis.core.page.TableDataInfo; @@ -9,7 +12,13 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import lombok.RequiredArgsConstructor; +import org.dromara.eims.domain.EimsEqu; +import org.dromara.eims.domain.vo.EimsEquVo; import org.dromara.eims.domain.vo.EimsMaintPlanVo; +import org.dromara.eims.domain.vo.InspectCheckItemVo; +import org.dromara.eims.listener.EasyExcelCellListener; +import org.dromara.eims.listener.InspectCheckItemImportListener; +import org.dromara.eims.mapper.EimsEquMapper; import org.dromara.system.domain.SysDept; import org.dromara.system.domain.vo.SysDeptVo; import org.dromara.system.mapper.SysDeptMapper; @@ -19,11 +28,10 @@ import org.dromara.eims.domain.EimsInspectPlan; import org.dromara.eims.mapper.EimsInspectPlanMapper; import org.dromara.eims.service.IEimsInspectPlanService; +import org.springframework.web.multipart.MultipartFile; -import java.util.ArrayList; -import java.util.List; -import java.util.Map; -import java.util.Collection; +import java.io.IOException; +import java.util.*; /** * 鐐规璁″垝Service涓氬姟灞傚鐞� @@ -37,6 +45,7 @@ private final EimsInspectPlanMapper baseMapper; private final SysDeptMapper sysDeptMapper; + private final EimsEquMapper equMapper; /** @@ -95,6 +104,8 @@ lqw.eq(bo.getInspUser() != null, EimsInspectPlan::getInspUser, bo.getInspUser()); lqw.eq(bo.getInspDept() != null, EimsInspectPlan::getInspDept, bo.getInspDept()); lqw.eq(StringUtils.isNotBlank(bo.getStatus()), EimsInspectPlan::getStatus, bo.getStatus()); + // 鎸夊垱寤烘椂闂村�掑簭 + lqw.orderByDesc(EimsInspectPlan::getCreateTime); return lqw; } private QueryWrapper<EimsInspectPlan> buildWrapper(EimsInspectPlanBo bo) { @@ -196,5 +207,75 @@ return baseMapper.deleteByIds(ids) > 0; } + @Override + public String importData(MultipartFile file, boolean updateSupport) { + try { + // 鐐规椤圭洰鍒楄〃锛堝亣璁惧瓨鍦ㄥ搴旂殑鐐规椤筕O锛� + InspectCheckItemImportListener checkItemImportListener = new InspectCheckItemImportListener(updateSupport); + EasyExcel.read(file.getInputStream(), InspectCheckItemVo.class, checkItemImportListener) + .headRowNumber(3).sheet().doRead(); + List<InspectCheckItemVo> successList = checkItemImportListener.getSuccessList(); + + // 璇诲彇鍥哄畾璧勪骇缂栧彿锛堝亣璁句綅缃笉鍚岋級 + EasyExcelCellListener assetNoListener = new EasyExcelCellListener(2, 23); + EasyExcel.read(file.getInputStream(), assetNoListener).headRowNumber(0).sheet().doReadSync(); + String assetNo = Optional.ofNullable(assetNoListener.getCellValue()) + .map(value -> { + int colonIndex = Math.max(value.indexOf(":"), value.indexOf("锛�")); // 鍚堝苟鍐掑彿澶勭悊 + return colonIndex != -1 ? value.substring(colonIndex + 1) : value; + }) + .map(String::trim) + .orElse(""); + + if (assetNo.isEmpty()) { + return file.getOriginalFilename() + " 瀵煎叆澶辫触锛屾棤娉曡鍙栧浐瀹氳祫浜х紪鍙�"; + } + + // 鏌ヨ璁惧淇℃伅 + QueryWrapper<EimsEqu> query = new QueryWrapper<>(); + query.eq("asset_no", assetNo); + EimsEquVo equVo = equMapper.selectVoOne(query); + if (equVo == null) { + return file.getOriginalFilename() + " 璁惧鏈壘鍒帮紝璇峰厛鍦ㄨ澶囧彴甯愪腑娣诲姞"; + } + + int successNum = 0; + int failureNum = 0; + + for (InspectCheckItemVo itemVo : successList) { + if ("璁惧鐘舵�佸崱鐘舵��".equals(itemVo.getItemName()) || itemVo.getItemName().contains("杩愯鏃堕棿")) { + break; + } + + // 濡傛灉鍏佽瑕嗙洊锛屽垯鍒犻櫎宸插瓨鍦ㄧ殑璁板綍 + if (updateSupport) { + LambdaQueryWrapper<EimsInspectPlan> deleteWrapper = Wrappers.lambdaQuery(); + deleteWrapper.eq(EimsInspectPlan::getEquId, equVo.getEquId()) + .eq(EimsInspectPlan::getInspName, itemVo.getItemName()) + .eq(EimsInspectPlan::getStatus, "0"); + baseMapper.delete(deleteWrapper); + } + + // 鏂板鐐规璁″垝 + EimsInspectPlanBo bo = new EimsInspectPlanBo(); + bo.setEquId(equVo.getEquId()); + bo.setInspName(itemVo.getItemName()); + bo.setStatus("0"); + bo.setInspType("1"); + bo.setInspRule("0"); + + if (!insertByBo(bo)) { + failureNum++; + } else { + successNum++; + } + } + + return file.getOriginalFilename() + " 瀵煎叆鎴愬姛锛�" + successNum + "鏉★紝瀵煎叆澶辫触锛�" + failureNum + "鏉�"; + } catch (Exception e) { + return file.getOriginalFilename() + " 瀵煎叆寮傚父锛�" + e.getMessage(); + } + } + } -- Gitblit v1.9.3