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.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; 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; /** * 分页列表查询 * * @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 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 toolClassifyList = toolsClassifyService.list(new LambdaQueryWrapper() .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 baseToolsList = baseToolsService.list(new LambdaQueryWrapper() .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 paraCommonToolList = paraCommonToolService.list(new LambdaQueryWrapper() .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); } } }