From 92bd9ed78ad1da026b81fcd8e88445b6834bba76 Mon Sep 17 00:00:00 2001 From: zhangherong <571457620@qq.com> Date: 星期二, 01 四月 2025 17:12:11 +0800 Subject: [PATCH] art: 设备管理-保养标准-点检导入功能 --- lxzn-module-eam/src/main/java/org/jeecg/modules/eam/controller/EamMaintenanceStandardController.java | 461 +++++++++++++++++++++++++++++++++++++-------------------- 1 files changed, 296 insertions(+), 165 deletions(-) diff --git a/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/controller/EamMaintenanceStandardController.java b/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/controller/EamMaintenanceStandardController.java index c753d7e..d153584 100644 --- a/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/controller/EamMaintenanceStandardController.java +++ b/lxzn-module-eam/src/main/java/org/jeecg/modules/eam/controller/EamMaintenanceStandardController.java @@ -1,194 +1,325 @@ package org.jeecg.modules.eam.controller; -import java.util.Arrays; -import java.util.List; -import java.util.Map; -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 org.jeecg.common.api.vo.Result; -import org.jeecg.common.system.query.QueryGenerator; -import org.jeecg.common.aspect.annotation.AutoLog; -import org.jeecg.common.util.oConvertUtils; -import org.jeecg.modules.eam.constant.BusinessCodeConst; -import org.jeecg.modules.eam.entity.EamMaintenanceStandard; -import org.jeecg.modules.eam.request.EamMaintenanceStandardRequest; -import org.jeecg.modules.eam.service.IEamMaintenanceStandardService; -import java.util.Date; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import io.swagger.annotations.Api; +import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; +import org.apache.commons.lang3.StringUtils; +import org.apache.poi.hssf.usermodel.HSSFSheet; +import org.apache.poi.hssf.usermodel.HSSFWorkbook; +import org.apache.poi.ss.usermodel.*; +import org.apache.poi.xssf.usermodel.XSSFSheet; +import org.apache.poi.xssf.usermodel.XSSFWorkbook; +import org.jeecg.common.api.vo.FileUploadResult; +import org.jeecg.common.api.vo.Result; +import org.jeecg.common.aspect.annotation.AutoLog; +import org.jeecg.common.exception.JeecgBootException; import org.jeecg.common.system.base.controller.JeecgController; +import org.jeecg.common.system.query.QueryGenerator; +import org.jeecg.common.util.FileUtil; +import org.jeecg.modules.eam.constant.BusinessCodeConst; +import org.jeecg.modules.eam.constant.MaintenanceCategoryEnum; +import org.jeecg.modules.eam.dto.MaintenanceStandardImport; +import org.jeecg.modules.eam.entity.EamEquipment; +import org.jeecg.modules.eam.entity.EamMaintenanceStandard; +import org.jeecg.modules.eam.entity.EamMaintenanceStandardDetail; +import org.jeecg.modules.eam.request.EamMaintenanceStandardRequest; +import org.jeecg.modules.eam.service.IEamEquipmentService; +import org.jeecg.modules.eam.service.IEamMaintenanceStandardService; import org.jeecg.modules.system.service.ISysBusinessCodeRuleService; 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.jeecgframework.poi.util.PoiPublicUtil; 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 javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.nio.file.Files; +import java.nio.file.Paths; +import java.util.*; +import java.util.stream.Collectors; + +/** * @Description: 淇濆吇鏍囧噯 * @Author: jeecg-boot - * @Date: 2025-03-26 + * @Date: 2025-03-26 * @Version: V1.0 */ @Slf4j -@Api(tags="淇濆吇鏍囧噯") +@Api(tags = "淇濆吇鏍囧噯") @RestController @RequestMapping("/eam/maintenanceStandard") public class EamMaintenanceStandardController extends JeecgController<EamMaintenanceStandard, IEamMaintenanceStandardService> { - @Autowired - private IEamMaintenanceStandardService eamMaintenanceStandardService; - @Autowired - private ISysBusinessCodeRuleService businessCodeRuleService; - - /** - * 鍒嗛〉鍒楄〃鏌ヨ - * - * @param eamMaintenanceStandard - * @param pageNo - * @param pageSize - * @param req - * @return - */ - @AutoLog(value = "淇濆吇鏍囧噯-鍒嗛〉鍒楄〃鏌ヨ") - @ApiOperation(value="淇濆吇鏍囧噯-鍒嗛〉鍒楄〃鏌ヨ", notes="淇濆吇鏍囧噯-鍒嗛〉鍒楄〃鏌ヨ") - @GetMapping(value = "/list") - public Result<?> queryPageList(EamMaintenanceStandard eamMaintenanceStandard, - @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, - @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, - HttpServletRequest req) { - QueryWrapper<EamMaintenanceStandard> queryWrapper = QueryGenerator.initQueryWrapper(eamMaintenanceStandard, req.getParameterMap()); - Page<EamMaintenanceStandard> page = new Page<EamMaintenanceStandard>(pageNo, pageSize); - IPage<EamMaintenanceStandard> pageList = eamMaintenanceStandardService.page(page, queryWrapper); - return Result.OK(pageList); - } - - /** - * 娣诲姞 - * - * @param standardRequest - * @return - */ - @AutoLog(value = "淇濆吇鏍囧噯-娣诲姞") - @ApiOperation(value="淇濆吇鏍囧噯-娣诲姞", notes="淇濆吇鏍囧噯-娣诲姞") - @PostMapping(value = "/add") - public Result<?> add(@RequestBody EamMaintenanceStandardRequest standardRequest) { - if(standardRequest == null){ - return Result.error("娣诲姞鐨勫璞′笉鑳戒负绌猴紒"); - } - if(CollectionUtil.isEmpty(standardRequest.getTableDetailList())) { - return Result.error("淇濆吇椤逛笉鑳戒负绌猴紒"); - } - String codeSeq = businessCodeRuleService.generateBusinessCodeSeq(BusinessCodeConst.MAINTENANCE_STANDARD_CODE_RULE); - standardRequest.setStandardCode(codeSeq); - boolean b = eamMaintenanceStandardService.addMaintenanceStandard(standardRequest); - if(!b) { - return Result.error("娣诲姞澶辫触锛�"); - } - return Result.OK("娣诲姞鎴愬姛锛�"); - } - - /** - * 缂栬緫 - * - * @param standardRequest - * @return - */ - @AutoLog(value = "淇濆吇鏍囧噯-缂栬緫") - @ApiOperation(value="淇濆吇鏍囧噯-缂栬緫", notes="淇濆吇鏍囧噯-缂栬緫") - @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST}) - public Result<?> edit(@RequestBody EamMaintenanceStandardRequest standardRequest) { - if(standardRequest == null){ - return Result.error("娣诲姞鐨勫璞′笉鑳戒负绌猴紒"); - } - if(CollectionUtil.isEmpty(standardRequest.getTableDetailList())) { - return Result.error("淇濆吇椤逛笉鑳戒负绌猴紒"); - } - boolean b = eamMaintenanceStandardService.editMaintenanceStandard(standardRequest); - if(!b) { - return Result.error("缂栬緫澶辫触锛�"); - } - return Result.OK("缂栬緫鎴愬姛!"); - } - - /** - * 閫氳繃id鍒犻櫎 - * - * @param id - * @return - */ - @AutoLog(value = "淇濆吇鏍囧噯-閫氳繃id鍒犻櫎") - @ApiOperation(value="淇濆吇鏍囧噯-閫氳繃id鍒犻櫎", notes="淇濆吇鏍囧噯-閫氳繃id鍒犻櫎") - @DeleteMapping(value = "/delete") - public Result<?> delete(@RequestParam(name="id",required=true) String id) { - eamMaintenanceStandardService.removeById(id); - return Result.OK("鍒犻櫎鎴愬姛!"); - } - - /** - * 鎵归噺鍒犻櫎 - * - * @param ids - * @return - */ - @AutoLog(value = "淇濆吇鏍囧噯-鎵归噺鍒犻櫎") - @ApiOperation(value="淇濆吇鏍囧噯-鎵归噺鍒犻櫎", notes="淇濆吇鏍囧噯-鎵归噺鍒犻櫎") - @DeleteMapping(value = "/deleteBatch") - public Result<?> deleteBatch(@RequestParam(name="ids",required=true) String ids) { - this.eamMaintenanceStandardService.removeByIds(Arrays.asList(ids.split(","))); - return Result.OK("鎵归噺鍒犻櫎鎴愬姛锛�"); - } - - /** - * 閫氳繃id鏌ヨ - * - * @param id - * @return - */ - @AutoLog(value = "淇濆吇鏍囧噯-閫氳繃id鏌ヨ") - @ApiOperation(value="淇濆吇鏍囧噯-閫氳繃id鏌ヨ", notes="淇濆吇鏍囧噯-閫氳繃id鏌ヨ") - @GetMapping(value = "/queryById") - public Result<?> queryById(@RequestParam(name="id",required=true) String id) { - EamMaintenanceStandard eamMaintenanceStandard = eamMaintenanceStandardService.getById(id); - return Result.OK(eamMaintenanceStandard); - } + @Autowired + private IEamMaintenanceStandardService eamMaintenanceStandardService; + @Autowired + private ISysBusinessCodeRuleService businessCodeRuleService; + @Autowired + private IEamEquipmentService eamEquipmentService; - /** - * 瀵煎嚭excel - * - * @param request - * @param eamMaintenanceStandard - */ - @RequestMapping(value = "/exportXls") - public ModelAndView exportXls(HttpServletRequest request, EamMaintenanceStandard eamMaintenanceStandard) { - return super.exportXls(request, eamMaintenanceStandard, EamMaintenanceStandard.class, "淇濆吇鏍囧噯"); - } + /** + * 鍒嗛〉鍒楄〃鏌ヨ + * + * @param eamMaintenanceStandard + * @param pageNo + * @param pageSize + * @param req + * @return + */ + @ApiOperation(value = "淇濆吇鏍囧噯-鍒嗛〉鍒楄〃鏌ヨ", notes = "淇濆吇鏍囧噯-鍒嗛〉鍒楄〃鏌ヨ") + @GetMapping(value = "/list") + public Result<?> queryPageList(EamMaintenanceStandard eamMaintenanceStandard, + @RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo, + @RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize, + HttpServletRequest req) { + QueryWrapper<EamMaintenanceStandard> queryWrapper = QueryGenerator.initQueryWrapper(eamMaintenanceStandard, req.getParameterMap()); + Page<EamMaintenanceStandard> page = new Page<EamMaintenanceStandard>(pageNo, pageSize); + IPage<EamMaintenanceStandard> pageList = eamMaintenanceStandardService.page(page, queryWrapper); + return Result.OK(pageList); + } - /** - * 閫氳繃excel瀵煎叆鏁版嵁 - * - * @param request - * @param response - * @return - */ - @RequestMapping(value = "/importExcel", method = RequestMethod.POST) - public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) { - return super.importExcel(request, response, EamMaintenanceStandard.class); - } + /** + * 娣诲姞 + * + * @param standardRequest + * @return + */ + @AutoLog(value = "淇濆吇鏍囧噯-娣诲姞") + @ApiOperation(value = "淇濆吇鏍囧噯-娣诲姞", notes = "淇濆吇鏍囧噯-娣诲姞") + @PostMapping(value = "/add") + public Result<?> add(@RequestBody EamMaintenanceStandardRequest standardRequest) { + if (standardRequest == null) { + return Result.error("娣诲姞鐨勫璞′笉鑳戒负绌猴紒"); + } + if (CollectionUtil.isEmpty(standardRequest.getTableDetailList())) { + return Result.error("淇濆吇椤逛笉鑳戒负绌猴紒"); + } + String codeSeq = businessCodeRuleService.generateBusinessCodeSeq(BusinessCodeConst.MAINTENANCE_STANDARD_CODE_RULE); + standardRequest.setStandardCode(codeSeq); + boolean b = eamMaintenanceStandardService.addMaintenanceStandard(standardRequest); + if (!b) { + return Result.error("娣诲姞澶辫触锛�"); + } + return Result.OK("娣诲姞鎴愬姛锛�"); + } + /** + * 缂栬緫 + * + * @param standardRequest + * @return + */ + @AutoLog(value = "淇濆吇鏍囧噯-缂栬緫") + @ApiOperation(value = "淇濆吇鏍囧噯-缂栬緫", notes = "淇濆吇鏍囧噯-缂栬緫") + @RequestMapping(value = "/edit", method = {RequestMethod.PUT, RequestMethod.POST}) + public Result<?> edit(@RequestBody EamMaintenanceStandardRequest standardRequest) { + if (standardRequest == null) { + return Result.error("娣诲姞鐨勫璞′笉鑳戒负绌猴紒"); + } + if (CollectionUtil.isEmpty(standardRequest.getTableDetailList())) { + return Result.error("淇濆吇椤逛笉鑳戒负绌猴紒"); + } + boolean b = eamMaintenanceStandardService.editMaintenanceStandard(standardRequest); + if (!b) { + return Result.error("缂栬緫澶辫触锛�"); + } + return Result.OK("缂栬緫鎴愬姛!"); + } + + /** + * 閫氳繃id鍒犻櫎 + * + * @param id + * @return + */ + @AutoLog(value = "淇濆吇鏍囧噯-閫氳繃id鍒犻櫎") + @ApiOperation(value = "淇濆吇鏍囧噯-閫氳繃id鍒犻櫎", notes = "淇濆吇鏍囧噯-閫氳繃id鍒犻櫎") + @DeleteMapping(value = "/delete") + public Result<?> delete(@RequestParam(name = "id", required = true) String id) { + eamMaintenanceStandardService.removeById(id); + return Result.OK("鍒犻櫎鎴愬姛!"); + } + + /** + * 鎵归噺鍒犻櫎 + * + * @param ids + * @return + */ + @AutoLog(value = "淇濆吇鏍囧噯-鎵归噺鍒犻櫎") + @ApiOperation(value = "淇濆吇鏍囧噯-鎵归噺鍒犻櫎", notes = "淇濆吇鏍囧噯-鎵归噺鍒犻櫎") + @DeleteMapping(value = "/deleteBatch") + public Result<?> deleteBatch(@RequestParam(name = "ids", required = true) String ids) { + this.eamMaintenanceStandardService.removeByIds(Arrays.asList(ids.split(","))); + return Result.OK("鎵归噺鍒犻櫎鎴愬姛锛�"); + } + + /** + * 閫氳繃id鏌ヨ + * + * @param id + * @return + */ + @ApiOperation(value = "淇濆吇鏍囧噯-閫氳繃id鏌ヨ", notes = "淇濆吇鏍囧噯-閫氳繃id鏌ヨ") + @GetMapping(value = "/queryById") + public Result<?> queryById(@RequestParam(name = "id", required = true) String id) { + EamMaintenanceStandard eamMaintenanceStandard = eamMaintenanceStandardService.getById(id); + return Result.OK(eamMaintenanceStandard); + } + + /** + * 瀵煎嚭excel + * + * @param request + * @param eamMaintenanceStandard + */ + @RequestMapping(value = "/exportXls") + public ModelAndView exportXls(HttpServletRequest request, EamMaintenanceStandard eamMaintenanceStandard) { + return super.exportXls(request, eamMaintenanceStandard, EamMaintenanceStandard.class, "淇濆吇鏍囧噯"); + } + + /** + * 閫氳繃excel瀵煎叆鏁版嵁 + * + * @param request + * @param response + * @return + */ + @RequestMapping(value = "/inspectionImportExcel", method = RequestMethod.POST) + public Result<?> inspectionImportExcel(HttpServletRequest request, HttpServletResponse response) { + MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; + Map<String, MultipartFile> fileMap = multipartRequest.getFileMap(); + for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) { + // 鑾峰彇涓婁紶鏂囦欢瀵硅薄 + MultipartFile file = entity.getValue(); + ImportParams params = new ImportParams(); + params.setTitleRows(2); + params.setHeadRows(1); + params.setSheetNum(1); + params.setNeedSave(true); + EamMaintenanceStandardRequest standardRequest = new EamMaintenanceStandardRequest(); + try { + //璇诲彇璁惧缂栧彿锛屽浘鐗囩瓑 + readExcel(file, standardRequest); + EamEquipment equipment = eamEquipmentService.selectByEquipmentCode(standardRequest.getEquipmentCode()); + if(equipment == null) { + log.error("璁惧涓嶅瓨鍦細{}", standardRequest.getEquipmentCode()); + continue; + } + standardRequest.setStandardName(standardRequest.getEquipmentName() + "鐐规鏍囧噯"); + standardRequest.setMaintenanceCategory(MaintenanceCategoryEnum.POINT_INSPECTION.name()); + standardRequest.setEquipmentId(equipment.getId()); + //璇诲彇淇濆吇鏄庣粏鍐呭 + List<MaintenanceStandardImport> list = ExcelImportUtil.importExcel(file.getInputStream(), MaintenanceStandardImport.class, params); + //鏄庣粏椤� + List<EamMaintenanceStandardDetail> tableList = list.stream().map(EamMaintenanceStandardDetail::new).collect(Collectors.toList()); + standardRequest.setTableDetailList(tableList); + String codeSeq = businessCodeRuleService.generateBusinessCodeSeq(BusinessCodeConst.MAINTENANCE_STANDARD_CODE_RULE); + standardRequest.setStandardCode(codeSeq); + boolean b = eamMaintenanceStandardService.addMaintenanceStandard(standardRequest); + if (!b) { + log.error("淇濆瓨澶辫触锛� {}", standardRequest.getEquipmentCode()); + } + } catch (Exception e) { + //update-begin-author:taoyan date:20211124 for: 瀵煎叆鏁版嵁閲嶅澧炲姞鎻愮ず + String msg = e.getMessage(); + log.error("鏂囦欢 {} 澶勭悊寮傚父锛� {}", file.getOriginalFilename(), msg, e); + //update-end-author:taoyan date:20211124 for: 瀵煎叆鏁版嵁閲嶅澧炲姞鎻愮ず + } finally { + try { + file.getInputStream().close(); + } catch (IOException e) { + log.error(e.getMessage(), e); + } + } + } + return Result.ok("鏂囦欢瀵煎叆瀹屾垚锛�"); + } + + /** + * 璇诲彇Excel 绗竴琛岋紝 绗簩琛岀殑淇℃伅锛屽寘鎷浘鐗囦俊鎭� + * @param file + * @param request + */ + private void readExcel(MultipartFile file, EamMaintenanceStandardRequest request) { + Workbook book = null; + boolean isXSSFWorkbook = false; + try { + book = WorkbookFactory.create(file.getInputStream()); + if (book instanceof XSSFWorkbook) { + isXSSFWorkbook = true; + } + + Sheet sheet = book.getSheetAt(0); + //绗竴琛岃鍙� + Row row = sheet.getRow(0); + //璁惧缂栫爜 + Cell equipmentCode = row.getCell(15); + if(CellType.NUMERIC.equals(equipmentCode.getCellType())) { + request.setEquipmentCode(String.valueOf((int) equipmentCode.getNumericCellValue())); + }else if(CellType.STRING.equals(equipmentCode.getCellType())) { + request.setEquipmentCode(equipmentCode.getStringCellValue()); + } + if (StringUtils.isBlank(request.getEquipmentCode())) { + throw new JeecgBootException("Excel銆� " + file.getOriginalFilename() + "銆戞病鏈夎鍙栧埌璁惧缂栧彿锛屽鍏ュけ璐ワ紒"); + } + //鍒濆鏃ユ湡 + Cell initialDate = row.getCell(11); + if (DateUtil.isCellDateFormatted(initialDate)) { + request.setInitialDate(initialDate.getDateCellValue()); + } else { + request.setInitialDate(new Date()); + } + //璁惧鍚嶇О + Cell equipmentName = row.getCell(13); + request.setEquipmentName(equipmentName.getStringCellValue()); + + //绗簩琛岃鍙� + row = sheet.getRow(1); + //淇濆吇鍛ㄦ湡 + Cell period = row.getCell(11); + if (CellType.NUMERIC.equals(period.getCellType())) { + request.setMaintenancePeriod((int) period.getNumericCellValue()); + } else { + //榛樿鐐规鍛ㄦ湡 1 + request.setMaintenancePeriod(1); + } + //鏂囦欢缂栫爜 + Cell fileCode = row.getCell(13); + request.setFileCode(fileCode.getStringCellValue()); + + Map<String, PictureData> pictures; + if (isXSSFWorkbook) { + pictures = PoiPublicUtil.getSheetPictrues07((XSSFSheet) book.getSheetAt(0), (XSSFWorkbook) book); + } else { + pictures = PoiPublicUtil.getSheetPictrues03((HSSFSheet) book.getSheetAt(0), (HSSFWorkbook) book); + } + + if (CollectionUtil.isNotEmpty(pictures)) { + //鍙細瀛樺湪涓�寮犲浘鐗� + PictureData pictureData = pictures.get(pictures.keySet().iterator().next()); + byte[] data = pictureData.getData(); + String fileName = request.getEquipmentCode() + "[" + request.getFileCode() + "]" + "." + pictureData.suggestFileExtension(); + FileUploadResult fileUploadResult = FileUtil.uploadFile(data, fileName); + if(fileUploadResult != null) { + List<FileUploadResult> fileList = request.getFileList(); + if(fileList == null) { + fileList = new ArrayList<FileUploadResult>(); + } + fileList.add(fileUploadResult); + request.setFileList(fileList); + } + } + } catch (Exception e) { + log.error("璇诲彇Excel淇℃伅澶辫触锛歿}", e.getMessage(), e); + } + } } -- Gitblit v1.9.3