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