工具管理-工具分类特征参数:刀片参数、通用工具参数导入

cuilei

2025-05-15 3eba8b62a62ecb56f8d286d5a93ba2768d54e3f3

 lxzn-module-tms/src/main/java/org/jeecg/modules/tms/controller/ParaCommonToolController.java

@@ -1,5 +1,8 @@
package org.jeecg.modules.tms.controller;

import java.io.File;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
@@ -9,17 +12,27 @@
import java.net.URLDecoder;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.tms.entity.ParaCommonTool;
import org.jeecg.modules.tms.service.IParaCommonToolService;

import cn.hutool.core.collection.CollectionUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.util.PmsUtil;
import org.jeecg.modules.tms.entity.BaseTools;
import org.jeecg.modules.tms.entity.ParaCommonTool;
import org.jeecg.modules.tms.entity.ToolsClassify;
import org.jeecg.modules.tms.service.IBaseToolsService;
import org.jeecg.modules.tms.service.IParaCommonToolService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;

import org.jeecg.modules.tms.service.IToolsClassifyService;
import org.jeecg.modules.tms.utils.ExcelUtils;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
@@ -49,6 +62,10 @@
public class ParaCommonToolController extends JeecgController<ParaCommonTool, IParaCommonToolService> {
   @Autowired
   private IParaCommonToolService paraCommonToolService;
   @Autowired
   private IToolsClassifyService toolsClassifyService;
   @Autowired
   private IBaseToolsService baseToolsService;
   
   /**
    * 分页列表查询
@@ -171,7 +188,132 @@
    //@RequiresPermissions("tms_para_common_tool:importExcel")
    @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
    public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
        return super.importExcel(request, response, ParaCommonTool.class);
      MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
      Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
      // 错误信息
      List<String> errorMessage = new ArrayList<>();
      int successLines = 0, errorLines = 0, dataLines = 0;
      List<ParaCommonTool> paraCommonToolImportList = new ArrayList<>();
      try {
         for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
            // 获取上传文件对象
            MultipartFile file = entity.getValue();
            InputStream inputStream = file.getInputStream();
            Workbook workbook = new XSSFWorkbook(inputStream);
            Sheet sheet = workbook.getSheetAt(0);
            for (Row row : sheet) {
               ParaCommonTool paraCommonTool = new ParaCommonTool();
               int rowNum = row.getRowNum();
               int currentRowNum  = rowNum + 1;
               //表头行和空行略过
               if (rowNum == 0 || ExcelUtils.isRowEmpty(row)) {
                  continue;
               }
               dataLines++;
               Cell cell1 = row.getCell(1);
               if (cell1 == null || cell1.getCellType() == CellType.BLANK) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,一级分类名称为空! ");
                  errorLines++;
                  continue;
               }
               Cell cell2 = row.getCell(2);
               if (cell2 == null || cell2.getCellType() == CellType.BLANK) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,二级分类名称为空! ");
                  errorLines++;
                  continue;
               }
               Cell cell3 = row.getCell(3);
               if (cell3 == null || cell3.getCellType() == CellType.BLANK) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,三级分类名称为空! ");
                  errorLines++;
                  continue;
               }
               Cell cell4 = row.getCell(4);
               if (cell4 == null || cell4.getCellType() == CellType.BLANK) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,四级分类名称为空! ");
                  errorLines++;
                  continue;
               }
               Cell cell9 = row.getCell(9);
               if (cell9 == null || cell9.getCellType() == CellType.BLANK) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,型号/图号为空! ");
                  errorLines++;
                  continue;
               }
               //根据四级分类名称和型号/图号查询工具信息
               List<ToolsClassify> toolClassifyList = toolsClassifyService.list(new LambdaQueryWrapper<ToolsClassify>()
                     .eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell4))
                     .eq(ToolsClassify::getStatus, CommonConstant.STATUS_1));
               if (CollectionUtil.isEmpty(toolClassifyList)) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,系统不存在该四级分类名称! ");
                  errorLines++;
                  continue;
               } else {
                  ToolsClassify toolsClassify = toolClassifyList.get(0);
                  List<BaseTools> baseToolsList = baseToolsService.list(new LambdaQueryWrapper<BaseTools>()
                        .eq(BaseTools::getClassifyId, toolsClassify.getId())
                        .eq(BaseTools::getToolModel, ExcelUtils.getCellValueAsString(cell9)));
                  if (CollectionUtil.isEmpty(baseToolsList)) {
                     errorMessage.add("第" + currentRowNum + "行，数据导入失败,系统不存在对应的工具信息! ");
                     errorLines++;
                     continue;
                  } else {
                     BaseTools baseTools = baseToolsList.get(0);
                     List<ParaCommonTool> paraCommonToolList = paraCommonToolService.list(new LambdaQueryWrapper<ParaCommonTool>()
                           .eq(ParaCommonTool::getToolCode, baseTools.getId())
                           .eq(ParaCommonTool::getToolModel, ExcelUtils.getCellValueAsString(cell9)));
                     if (CollectionUtil.isEmpty(paraCommonToolList)) {
                        paraCommonTool.setClassifyId(toolsClassify.getId());
                        paraCommonTool.setToolCode(baseTools.getId());
                        paraCommonTool.setToolModel(ExcelUtils.getCellValueAsString(cell9));
                     } else {
                        paraCommonTool = paraCommonToolList.get(0);
                     }
                  }
               }

               Cell cell0 = row.getCell(0);//序号(或企业内原代码)
               paraCommonTool.setSignCode(ExcelUtils.getCellValueAsString(cell0));
               Cell cell5 = row.getCell(5);//物品中文名称
               if (cell5 == null || cell5.getCellType() == CellType.BLANK) {
                  errorMessage.add("第" + currentRowNum + "行，数据导入失败,物品中文名称为空! ");
                  errorLines++;
                  continue;
               } else {
                  paraCommonTool.setChineseName(ExcelUtils.getCellValueAsString(cell5));
               }
               Cell cell6 = row.getCell(6);//物品外文名称
               paraCommonTool.setForeignLanguageName(ExcelUtils.getCellValueAsString(cell6));
               Cell cell7 = row.getCell(7);//标准级别
               paraCommonTool.setStandardLevel(ExcelUtils.getCellValueAsString(cell7));
               Cell cell8 = row.getCell(8);//标准号
               paraCommonTool.setStandardCode(ExcelUtils.getCellValueAsString(cell8));

               paraCommonToolImportList.add(paraCommonTool);
               successLines++;
            }
         }
         paraCommonToolService.saveOrUpdateBatch(paraCommonToolImportList);
         if (successLines == dataLines) {
            return Result.OK("共" + dataLines + "行数据全部导入成功！");
         } else {
            JSONObject result = new JSONObject(5);
            result.put("successCount", successLines);
            result.put("msg", "总上传行数：" + dataLines + "，成功导入行数：" + successLines);
            result.put("errorMsg", errorMessage);
            String fileUrl = PmsUtil.saveErrorMsgByList(errorMessage, "userImportExcelErrorLog");
            int lastIndex = fileUrl.lastIndexOf(File.separator);
            String fileName = fileUrl.substring(lastIndex + 1);
            result.put("fileUrl", "/sys/common/static/" + fileUrl);
            result.put("fileName", fileName);
            Result res = Result.ok(result);
            res.setCode(201);
            res.setMessage("文件导入成功，但有错误。");
            return res;
         }
      } catch (IOException e) {
         throw new RuntimeException(e);
      }
    }

}