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; import java.util.stream.Collectors; import java.io.IOException; import java.io.UnsupportedEncodingException; import java.net.URLDecoder; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; 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.*; import org.jeecg.modules.tms.enums.AccuracyClass; 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.service.IToolsConfigPropertyService; 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; import org.jeecgframework.poi.excel.entity.ImportParams; import org.jeecgframework.poi.excel.view.JeecgEntityExcelView; import org.jeecg.common.system.base.controller.JeecgController; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import org.springframework.web.servlet.ModelAndView; import com.alibaba.fastjson.JSON; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.jeecg.common.aspect.annotation.AutoLog; /** * @Description: tms_para_common_tool * @Author: jeecg-boot * @Date: 2025-05-12 * @Version: V1.0 */ @Api(tags="tms_para_common_tool") @RestController @RequestMapping("/tms/paraCommonTool") @Slf4j public class ParaCommonToolController extends JeecgController { @Autowired private IParaCommonToolService paraCommonToolService; @Autowired private IToolsClassifyService toolsClassifyService; @Autowired private IBaseToolsService baseToolsService; @Autowired private IToolsConfigPropertyService toolsConfigPropertyService; /** * 分页列表查询 * * @param paraCommonTool * @param pageNo * @param pageSize * @param req * @return */ //@AutoLog(value = "tms_para_common_tool-分页列表查询") @ApiOperation(value="tms_para_common_tool-分页列表查询", notes="tms_para_common_tool-分页列表查询") @GetMapping(value = "/list") public Result> queryPageList(ParaCommonTool paraCommonTool, @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, HttpServletRequest req) { QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(paraCommonTool, req.getParameterMap()); Page page = new Page(pageNo, pageSize); IPage pageList = paraCommonToolService.page(page, queryWrapper); return Result.OK(pageList); } /** * 添加 * * @param paraCommonTool * @return */ @AutoLog(value = "tms_para_common_tool-添加") @ApiOperation(value="tms_para_common_tool-添加", notes="tms_para_common_tool-添加") //@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:add") @PostMapping(value = "/add") public Result add(@RequestBody ParaCommonTool paraCommonTool) { paraCommonToolService.save(paraCommonTool); return Result.OK("添加成功!"); } /** * 编辑 * * @param paraCommonTool * @return */ @AutoLog(value = "tms_para_common_tool-编辑") @ApiOperation(value="tms_para_common_tool-编辑", notes="tms_para_common_tool-编辑") //@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:edit") @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST}) public Result edit(@RequestBody ParaCommonTool paraCommonTool) { paraCommonToolService.updateById(paraCommonTool); return Result.OK("编辑成功!"); } /** * 通过id删除 * * @param id * @return */ @AutoLog(value = "tms_para_common_tool-通过id删除") @ApiOperation(value="tms_para_common_tool-通过id删除", notes="tms_para_common_tool-通过id删除") //@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:delete") @DeleteMapping(value = "/delete") public Result delete(@RequestParam(name="id",required=true) String id) { paraCommonToolService.removeById(id); return Result.OK("删除成功!"); } /** * 批量删除 * * @param ids * @return */ @AutoLog(value = "tms_para_common_tool-批量删除") @ApiOperation(value="tms_para_common_tool-批量删除", notes="tms_para_common_tool-批量删除") //@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:deleteBatch") @DeleteMapping(value = "/deleteBatch") public Result deleteBatch(@RequestParam(name="ids",required=true) String ids) { this.paraCommonToolService.removeByIds(Arrays.asList(ids.split(","))); return Result.OK("批量删除成功!"); } /** * 通过id查询 * * @param id * @return */ //@AutoLog(value = "tms_para_common_tool-通过id查询") @ApiOperation(value="tms_para_common_tool-通过id查询", notes="tms_para_common_tool-通过id查询") @GetMapping(value = "/queryById") public Result queryById(@RequestParam(name="id",required=true) String id) { ParaCommonTool paraCommonTool = paraCommonToolService.getById(id); if(paraCommonTool==null) { return Result.error("未找到对应数据"); } return Result.OK(paraCommonTool); } /** * 导出excel * * @param request * @param paraCommonTool */ //@RequiresPermissions("org.jeecg.modules:tms_para_common_tool:exportXls") @RequestMapping(value = "/exportXls") public ModelAndView exportXls(HttpServletRequest request, ParaCommonTool paraCommonTool) { return super.exportXls(request, paraCommonTool, ParaCommonTool.class, "tms_para_common_tool"); } /** * 通过excel导入数据 * * @param request * @param response * @return */ //@RequiresPermissions("tms_para_common_tool:importExcel") @RequestMapping(value = "/importExcel", method = RequestMethod.POST) public Result importExcel(HttpServletRequest request, HttpServletResponse response) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; Map fileMap = multipartRequest.getFileMap(); // 错误信息 List errorMessage = new ArrayList<>(); int successLines = 0, errorLines = 0, dataLines = 0; List paraCommonToolImportList = new ArrayList<>(); try { for (Map.Entry 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 cell0 = row.getCell(0); if (cell0 == null || cell0.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,刀具编码为空! "); errorLines++; continue; } Cell cell1 = row.getCell(1); if (cell1 == null || cell1.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 cell5 = row.getCell(5); if (cell5 == null || cell5.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,三级分类名称为空! "); errorLines++; continue; } Cell cell6 = row.getCell(6); if (cell6 == null || cell6.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,四级分类名称为空! "); errorLines++; continue; } ToolsClassify classify; List class1List = toolsClassifyService.list(new LambdaQueryWrapper() .eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell3)) .eq(ToolsClassify::getStatus, CommonConstant.STATUS_1)); if (CollectionUtil.isEmpty(class1List)) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,系统不存在该一级分类名称! "); errorLines++; continue; } else { ToolsClassify classify1 = class1List.get(0); List class2List = toolsClassifyService.list(new LambdaQueryWrapper() .eq(ToolsClassify::getParentId, classify1.getId()) .eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell4)) .eq(ToolsClassify::getStatus, CommonConstant.STATUS_1)); if (CollectionUtil.isEmpty(class2List)) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,系统不存在该二级分类名称! "); errorLines++; continue; } else { ToolsClassify classify2 = class2List.get(0); List class3List = toolsClassifyService.list(new LambdaQueryWrapper() .eq(ToolsClassify::getParentId, classify2.getId()) .eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell5)) .eq(ToolsClassify::getStatus, CommonConstant.STATUS_1)); if (CollectionUtil.isEmpty(class3List)) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,系统不存在该三级分类名称! "); errorLines++; continue; } else { ToolsClassify classify3 = class3List.get(0); List class4List = toolsClassifyService.list(new LambdaQueryWrapper() .eq(ToolsClassify::getParentId, classify3.getId()) .eq(ToolsClassify::getTypeName, ExcelUtils.getCellValueAsString(cell6)) .eq(ToolsClassify::getStatus, CommonConstant.STATUS_1)); if (CollectionUtil.isEmpty(class4List)) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,系统不存在该四级分类名称! "); errorLines++; continue; } else { classify = class4List.get(0); paraCommonTool.setClassifyId(classify.getId()); } } } } Cell cell2 = row.getCell(2);//序号(或企业内原代码) paraCommonTool.setSignCode(ExcelUtils.getCellValueAsString(cell2)); Cell cell7 = row.getCell(7);//物品中文名称 if (cell7 == null || cell7.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,物品中文名称为空! "); errorLines++; continue; } else { paraCommonTool.setChineseName(ExcelUtils.getCellValueAsString(cell7)); } Cell cell8 = row.getCell(8);//物品外文名称 paraCommonTool.setForeignLanguageName(ExcelUtils.getCellValueAsString(cell8)); Cell cell9 = row.getCell(9);//标准级别 paraCommonTool.setStandardLevel(ExcelUtils.getCellValueAsString(cell9)); Cell cell10 = row.getCell(10);//标准号 paraCommonTool.setStandardCode(ExcelUtils.getCellValueAsString(cell10)); Cell cell11 = row.getCell(11); if (cell11 == null || cell11.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,型号/图号为空! "); errorLines++; continue; } else { paraCommonTool.setToolModel(ExcelUtils.getCellValueAsString(cell11)); } Cell cell12 = row.getCell(12); String accuracyClassCode; if (cell12 == null || cell12.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,精度等级为空! "); errorLines++; continue; } else { accuracyClassCode = AccuracyClass.getCodeFromValue(ExcelUtils.getCellValueAsString(cell12)); if (accuracyClassCode == null) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,精度等级不符合要求! "); errorLines++; continue; } } BaseTools baseTools = new BaseTools() .setToolCode(ExcelUtils.getCellValueAsString(cell0)) .setClassifyId(classify.getId()) .setSignCode(ExcelUtils.getCellValueAsString(cell2)) .setChineseName(ExcelUtils.getCellValueAsString(cell7)) .setForeignLanguageName(ExcelUtils.getCellValueAsString(cell8)) .setStandardLevel(ExcelUtils.getCellValueAsString(cell9)) .setStandardCode(ExcelUtils.getCellValueAsString(cell10)) .setToolModel(ExcelUtils.getCellValueAsString(cell11)) .setCurrentCode("0000")//当前编码初始化 .setParamaTableName("1")//1:通用参数类型 .setAccuracyClass(accuracyClassCode); //根据刀具编码、型号/图号、分类id查询刀具信息 List toolsList = baseToolsService.list(new LambdaQueryWrapper() .eq(BaseTools::getToolCode, ExcelUtils.getCellValueAsString(cell0)) .eq(BaseTools::getClassifyId, classify.getId()) .eq(BaseTools::getToolModel, ExcelUtils.getCellValueAsString(cell11))); if (CollectionUtil.isEmpty(toolsList)) { //不存在相应刀具,新增 baseToolsService.save(baseTools); } else { //存在相应刀具,更新 BaseTools tools = toolsList.get(0); baseTools.setId(tools.getId()); baseToolsService.updateById(baseTools); } paraCommonTool.setToolCode(baseTools.getId()); ToolsConfigProperty configProperty = new ToolsConfigProperty() .setChineseName(ExcelUtils.getCellValueAsString(cell7)) .setForeignLanguageName(ExcelUtils.getCellValueAsString(cell8)) .setStandardLevel(ExcelUtils.getCellValueAsString(cell9)) .setStandardCode(ExcelUtils.getCellValueAsString(cell10)) .setToolModel(ExcelUtils.getCellValueAsString(cell11)) .setPositionCode(ExcelUtils.getCellValueAsString(cell1)) .setApplicationType("1");//1:通用工具类型 //根据刀具id、分类id查询通用参数信息 List configPropertyList = toolsConfigPropertyService.list(new LambdaQueryWrapper() .eq(ToolsConfigProperty::getToolCode, baseTools.getId()) .eq(ToolsConfigProperty::getClassifyId, classify.getId()) .eq(ToolsConfigProperty::getStatus, CommonConstant.STATUS_1)); if (CollectionUtil.isEmpty(configPropertyList)) { //不存在通用参数信息,新增 configProperty.setStatus(CommonConstant.STATUS_1); configProperty.setToolCode(baseTools.getId()); configProperty.setClassifyId(classify.getId()); toolsConfigPropertyService.save(configProperty); } else { //存在通用参数信息,更新 ToolsConfigProperty property = configPropertyList.get(0); configProperty.setId(property.getId()); toolsConfigPropertyService.updateById(configProperty); } List paraCommonToolList = paraCommonToolService.list(new LambdaQueryWrapper() .eq(ParaCommonTool::getToolCode, baseTools.getId()) .eq(ParaCommonTool::getClassifyId, classify.getId()) .eq(ParaCommonTool::getToolModel, ExcelUtils.getCellValueAsString(cell11))); if (CollectionUtil.isNotEmpty(paraCommonToolList)) { //存在该通用参数信息,更新 ParaCommonTool commonTool = paraCommonToolList.get(0); paraCommonTool.setId(commonTool.getId()); } Cell cell13 = row.getCell(13);//刀具材料 if (cell13 == null || cell13.getCellType() == CellType.BLANK) { errorMessage.add("第" + currentRowNum + "行,数据导入失败,刀具材料为空! "); errorLines++; continue; } else { paraCommonTool.setToolMaterial(ExcelUtils.getCellValueAsString(cell13)); } Cell cell14 = row.getCell(14);//零件材料 paraCommonTool.setPartMaterial(ExcelUtils.getCellValueAsString(cell14)); Cell cell15 = row.getCell(15);//集团公司编码 paraCommonTool.setGroupCompanyCode(ExcelUtils.getCellValueAsString(cell15)); Cell cell16 = row.getCell(16);//集团公司标记 if (cell16 == null || cell16.getCellType() == CellType.BLANK) { //标记为空,拼接 String sign = ExcelUtils.getCellValueAsString(cell7)//中文名称 + "-" + ExcelUtils.getCellValueAsString(cell11);//型号/图号 paraCommonTool.setGroupCompanySign(sign); } else { paraCommonTool.setGroupCompanySign(ExcelUtils.getCellValueAsString(cell16)); } Cell cell17 = row.getCell(17);//工具简称 paraCommonTool.setShortCalled(ExcelUtils.getCellValueAsString(cell17)); 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); } } }