From 1dad50645c6e61980f15b2be4505f75e7c922ee7 Mon Sep 17 00:00:00 2001 From: xyc <jc_xiong@hotmail.com> Date: 星期二, 22 十月 2024 14:39:33 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- modules/mainPart/src/main/java/com/zt/life/modules/mainPart/basicInfo/service/XhProductModelService.java | 260 ++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 202 insertions(+), 58 deletions(-) diff --git a/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/basicInfo/service/XhProductModelService.java b/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/basicInfo/service/XhProductModelService.java index b62fc7b..f9a0192 100644 --- a/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/basicInfo/service/XhProductModelService.java +++ b/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/basicInfo/service/XhProductModelService.java @@ -1,5 +1,7 @@ package com.zt.life.modules.mainPart.basicInfo.service; +import com.alibaba.excel.EasyExcelFactory; +import com.alibaba.excel.ExcelWriter; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.zt.common.constant.Constant; import com.zt.common.service.BaseService; @@ -8,14 +10,17 @@ import com.zt.common.utils.TreeUtils; import com.zt.common.utils.UUIDUtil; import com.zt.core.shiro.ImportUtil; +import com.zt.life.export.service.DownloadService; import com.zt.life.modules.mainPart.basicInfo.dao.XhProductModelDao; +import com.zt.life.modules.mainPart.basicInfo.dto.ProductDto; import com.zt.life.modules.mainPart.basicInfo.model.ParamData; import com.zt.life.modules.mainPart.basicInfo.model.ProductImg; import com.zt.life.modules.mainPart.basicInfo.model.XhProductModel; import com.zt.life.modules.mainPart.sysPictureBase.service.SysPictureBaseService; -import com.zt.modules.sys.dto.DictTypeDto; import org.apache.commons.lang3.StringUtils; import org.apache.poi.hssf.usermodel.HSSFWorkbook; +import org.apache.poi.openxml4j.exceptions.InvalidFormatException; +import org.apache.poi.ss.usermodel.Cell; import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.Workbook; @@ -25,6 +30,9 @@ import com.zt.common.db.query.QueryFilter; import org.springframework.web.multipart.MultipartFile; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.*; import java.util.*; @@ -194,23 +202,37 @@ String xt = com.zt.core.shiro.ImportUtil.getCellValue(row, 1, pattern); //绯荤粺 String fxt = com.zt.core.shiro.ImportUtil.getCellValue(row, 2, pattern); String sb = com.zt.core.shiro.ImportUtil.getCellValue(row, 3, pattern); //璁惧 - String sameSbName = com.zt.core.shiro.ImportUtil.getCellValue(row, 4, pattern); //鍚岀被璁惧鍚嶇О + String sameSbNumStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 4, pattern); //鍚岀被璁惧鏁伴噺 String equipType = com.zt.core.shiro.ImportUtil.getCellValue(row, 5, pattern); //璁惧绫诲瀷 String reliabDistribType = com.zt.core.shiro.ImportUtil.getCellValue(row, 6, pattern); //鍙潬鎬у垎甯冪被鍨� String basicJoinCompute = com.zt.core.shiro.ImportUtil.getCellValue(row, 7, pattern); //鍙傚姞璁$畻 - String basicMtbfRegulate = com.zt.core.shiro.ImportUtil.getCellValue(row, 8, pattern); //mtbf - String basicMtbfRegulSuccRate = com.zt.core.shiro.ImportUtil.getCellValue(row, 9, pattern); //鎴愬姛鐜� - String basicMtbfOperatingRatio = com.zt.core.shiro.ImportUtil.getCellValue(row, 10, pattern); //杩愯姣� - String taskMtbcfRegulate = com.zt.core.shiro.ImportUtil.getCellValue(row, 11, pattern); //mtbcf - String taskMtbcfOtherParams2 = com.zt.core.shiro.ImportUtil.getCellValue(row, 12, pattern); //mtbcf鍏朵粬鍙傛暟2 - String taskMtbcfOtherParams3 = com.zt.core.shiro.ImportUtil.getCellValue(row, 13, pattern); //mtbcf鍏朵粬鍙傛暟3 - String taskMtbcfRegulSuccRate = com.zt.core.shiro.ImportUtil.getCellValue(row, 14, pattern); //mtbcf鎴愬姛鐜� - String taskMtbcfOperatingRatio = com.zt.core.shiro.ImportUtil.getCellValue(row, 15, pattern); //mtbcf杩愯姣� + String basicMtbfRegulateStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 8, pattern); //mtbf + String basicMtbfRegulSuccRateStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 9, pattern); //鎴愬姛鐜� + String basicMtbfOperatingRatioStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 10, pattern); //杩愯姣� + String taskMtbcfRegulateStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 11, pattern); //mtbcf + String taskMtbcfOtherParams2Str = com.zt.core.shiro.ImportUtil.getCellValue(row, 12, pattern); //mtbcf鍏朵粬鍙傛暟2 + String taskMtbcfOtherParams3Str = com.zt.core.shiro.ImportUtil.getCellValue(row, 13, pattern); //mtbcf鍏朵粬鍙傛暟3 + String taskMtbcfRegulSuccRateStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 14, pattern); //mtbcf鎴愬姛鐜� + String taskMtbcfOperatingRatioStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 15, pattern); //mtbcf杩愯姣� String repairable = com.zt.core.shiro.ImportUtil.getCellValue(row, 16, pattern); //鏄惁鍙淮淇� String repairDistribType = com.zt.core.shiro.ImportUtil.getCellValue(row, 17, pattern); //缁翠慨鍙戝竷绫诲瀷 - String repairMttcr = com.zt.core.shiro.ImportUtil.getCellValue(row, 18, pattern); //mttcr - String repairMttcrOtherParams2 = com.zt.core.shiro.ImportUtil.getCellValue(row, 19, pattern); //mttcr鍏朵粬鍙傛暟2 - String repairMttcrOtherParams3 = com.zt.core.shiro.ImportUtil.getCellValue(row, 20, pattern); //mttcr鍏朵粬鍙傛暟3 + String repairMttcrStr = com.zt.core.shiro.ImportUtil.getCellValue(row, 18, pattern); //mttcr + String repairMttcrOtherParams2Str = com.zt.core.shiro.ImportUtil.getCellValue(row, 19, pattern); //mttcr鍏朵粬鍙傛暟2 + String repairMttcrOtherParams3Str = com.zt.core.shiro.ImportUtil.getCellValue(row, 20, pattern); //mttcr鍏朵粬鍙傛暟3 + + Integer sameSbNum = StringUtils.isNotBlank(sameSbNumStr) ? Integer.parseInt(sameSbNumStr) : 1; + Double basicMtbfRegulate = StringUtils.isNotBlank(basicMtbfRegulateStr) ? Double.valueOf(basicMtbfRegulateStr) : null; + Double basicMtbfRegulSuccRate = StringUtils.isNotBlank(basicMtbfRegulSuccRateStr) ? Double.valueOf(basicMtbfRegulSuccRateStr) : null; + Double basicMtbfOperatingRatio = StringUtils.isNotBlank(basicMtbfOperatingRatioStr) ? Double.parseDouble(basicMtbfOperatingRatioStr) : 1.0; + Double taskMtbcfRegulate = StringUtils.isNotBlank(taskMtbcfRegulateStr) ? Double.parseDouble(taskMtbcfRegulateStr) : null; + Double taskMtbcfOtherParams2 = StringUtils.isNotBlank(taskMtbcfOtherParams2Str) ? Double.parseDouble(taskMtbcfOtherParams2Str) : null; + Double taskMtbcfOtherParams3 = StringUtils.isNotBlank(taskMtbcfOtherParams3Str) ? Double.parseDouble(taskMtbcfOtherParams3Str) : null; + Double taskMtbcfRegulSuccRate = StringUtils.isNotBlank(taskMtbcfRegulSuccRateStr) ? Double.parseDouble(taskMtbcfRegulSuccRateStr) : null; + Double taskMtbcfOperatingRatio = StringUtils.isNotBlank(taskMtbcfOperatingRatioStr) ? Double.parseDouble(taskMtbcfOperatingRatioStr) : 1.0; + Double repairMttcr = StringUtils.isNotBlank(repairMttcrStr) ? Double.parseDouble(repairMttcrStr) : null; + Double repairMttcrOtherParams2 = StringUtils.isNotBlank(repairMttcrOtherParams2Str) ? Double.parseDouble(repairMttcrOtherParams2Str) : null; + Double repairMttcrOtherParams3 = StringUtils.isNotBlank(repairMttcrOtherParams3Str) ? Double.parseDouble(repairMttcrOtherParams3Str) : null; + if (StringUtils.isBlank(sb)) continue; @@ -253,10 +275,6 @@ } else { curSbId = modelMap.get(sbPath).getId(); paramId = modelMap.get(sbPath).getParamId(); - } - - if(StringUtils.isBlank(sameSbName)){ - sameSbName = sb; } if (StringUtils.isEmpty(sb)) { @@ -311,7 +329,7 @@ xhProductModel.setPid(curFxtId == null ? curXtId : curFxtId); xhProductModel.setId(curSbId); xhProductModel.setName(sb); - xhProductModel.setSameSbName(sameSbName); + //xhProductModel.setSameSbName(sameSbName); xhProductModel.setEquipType(equipType); xhProductModel.setProductType("5"); xhProductModel.setSort(Integer.valueOf(sort)); @@ -320,7 +338,7 @@ xhProductModel.setNamePath(sbPath); if (insertSb) { this.insert(xhProductModel); - }else{ + } else { this.update(xhProductModel); } @@ -331,45 +349,21 @@ itemEntity.setProductId(curSbId); itemEntity.setReliabDistribType(reliabType.get(reliabDistribType)); itemEntity.setBasicJoinCompute(isOrNot.get(basicJoinCompute)); - if (StringUtils.isNotBlank(basicMtbfRegulate)) { - itemEntity.setBasicMtbfRegulate(Double.valueOf(basicMtbfRegulate)); - } - if (StringUtils.isNotBlank(basicMtbfRegulSuccRate)) { - itemEntity.setBasicMtbfRegulSuccRate(Double.valueOf(basicMtbfRegulSuccRate)); - } - if (StringUtils.isNotBlank(basicMtbfOperatingRatio)) { - itemEntity.setBasicMtbfOperatingRatio(Double.valueOf(basicMtbfOperatingRatio)); - } else { - itemEntity.setBasicMtbfOperatingRatio(1.0); - } - if (StringUtils.isNotBlank(taskMtbcfRegulate)) { - itemEntity.setTaskMtbcfRegulate(Double.valueOf(taskMtbcfRegulate)); - } - if (StringUtils.isNotBlank(taskMtbcfOtherParams2)) { - itemEntity.setTaskMtbcfOtherParams2(Double.valueOf(taskMtbcfOtherParams2)); - } - if (StringUtils.isNotBlank(taskMtbcfOtherParams3)) { - itemEntity.setTaskMtbcfOtherParams3(Double.valueOf(taskMtbcfOtherParams3)); - } - if (StringUtils.isNotBlank(taskMtbcfRegulSuccRate)) { - itemEntity.setTaskMtbcfRegulSuccRate(Double.valueOf(taskMtbcfRegulSuccRate)); - } - if (StringUtils.isNotBlank(taskMtbcfOperatingRatio)) { - itemEntity.setTaskMtbcfOperatingRatio(Double.valueOf(taskMtbcfOperatingRatio)); - } else { - itemEntity.setTaskMtbcfOperatingRatio(1.0); - } + itemEntity.setBasicUnitNum(sameSbNum); + itemEntity.setBasicMtbfRegulate(basicMtbfRegulate); + itemEntity.setBasicMtbfRegulSuccRate(basicMtbfRegulSuccRate); + itemEntity.setBasicMtbfOperatingRatio(basicMtbfOperatingRatio); + itemEntity.setTaskMtbcfRegulate(taskMtbcfRegulate); + itemEntity.setTaskMtbcfOtherParams2(taskMtbcfOtherParams2); + itemEntity.setTaskMtbcfOtherParams3(taskMtbcfOtherParams3); + itemEntity.setTaskMtbcfRegulSuccRate(taskMtbcfRegulSuccRate); + itemEntity.setTaskMtbcfOperatingRatio(taskMtbcfOperatingRatio); itemEntity.setRepairable(isOrNot.get(repairable)); itemEntity.setRepairDistribType(repairType.get(repairDistribType)); - if (StringUtils.isNotBlank(repairMttcr)) { - itemEntity.setRepairMttcr(Double.valueOf(repairMttcr)); - } - if (StringUtils.isNotBlank(repairMttcrOtherParams2)) { - itemEntity.setRepairMttcrOtherParams2(Double.valueOf(repairMttcrOtherParams2)); - } - if (StringUtils.isNotBlank(repairMttcrOtherParams3)) { - itemEntity.setRepairMttcrOtherParams3(Double.valueOf(repairMttcrOtherParams3)); - } + itemEntity.setRepairMttcr(repairMttcr); + itemEntity.setRepairMttcrOtherParams2(repairMttcrOtherParams2); + itemEntity.setRepairMttcrOtherParams3(repairMttcrOtherParams3); + if (modelMap.get(sbPath) == null) { paramDataService.insert(itemEntity); } else { @@ -414,7 +408,157 @@ return baseDao.getByPid(id); } - public void exportDataExcel(Long shipId) { - baseDao.getProductByShip(shipId); + public void exportDataExcel(HttpServletRequest request, HttpServletResponse response, Long shipId, String name) { + Map<Integer, String> reliabType = new HashMap<>(); + reliabType.put(1, "鎸囨暟鍒嗗竷"); + reliabType.put(3, "浜岄」鍒嗗竷"); + reliabType.put(2, "濞佸竷灏斿垎甯�"); + Map<Integer, String> repairType = new HashMap<>(); + repairType.put(1, "鎸囨暟鍒嗗竷"); + repairType.put(2, "濞佸竷灏斿垎甯�"); + Map<Integer, String> isOrNot = new HashMap<>(); + isOrNot.put(1, "鏄�"); + isOrNot.put(0, "鍚�"); + + List<ProductDto> data = baseDao.getProductByShip(shipId); + data = TreeUtils.build(data); + List<ProductDto> List = new ArrayList<>(); + this.writeExcel(data, List); + // 璇诲彇妯℃澘鏂囦欢 + try (InputStream inputStream = getClass().getResourceAsStream("/浜у搧缁撴瀯鏍戝鍑烘ā鏉�.xlsx")) { + Workbook workbook = new XSSFWorkbook(inputStream); + // 鑾峰彇绗竴涓伐浣滆〃 + Sheet sheet = workbook.getSheetAt(0); + int num = 0; + String xt = ""; + String fxt = ""; + String sb = ""; + for (int i = 0; i < List.size(); i++) { + if (List.get(i).getProductType() == 5) { + num = num + 1; + String[] parentList = List.get(i).getNamePath().split(","); + Row row = sheet.createRow(sheet.getLastRowNum() + 1); // 鍒涘缓鏂拌 + Cell cell = row.createCell(0); + cell.setCellValue(num); + if (parentList.length > 2) { + if (!xt.equals(parentList[0])) { + xt = parentList[0]; + cell = row.createCell(1); + cell.setCellValue(xt); + } + if (!fxt.equals(parentList[1])) { + fxt = parentList[1]; + cell = row.createCell(2); + cell.setCellValue(fxt); + } + if (!sb.equals(parentList[2])) { + sb = parentList[2]; + cell = row.createCell(3); + cell.setCellValue(sb); + } + } else { + if (!xt.equals(parentList[0])) { + xt = parentList[0]; + cell = row.createCell(1); + cell.setCellValue(xt); + } + + if (!sb.equals(parentList[1])) { + sb = parentList[1]; + cell = row.createCell(3); + cell.setCellValue(sb); + } + } + cell = row.createCell(4); + if (StringUtils.isNotBlank(List.get(i).getSameSbName())) { + cell.setCellValue(List.get(i).getSameSbName()); + } + cell = row.createCell(5); + if (StringUtils.isNotBlank(List.get(i).getEquipType())) { + cell.setCellValue(List.get(i).getEquipType()); + } + cell = row.createCell(6); + String type = reliabType.get(List.get(i).getReliabDistribType()); + cell.setCellValue(type); + cell = row.createCell(7); + String isNot = isOrNot.get(List.get(i).getBasicJoinCompute()); + cell.setCellValue(isNot); + cell = row.createCell(8); + if (List.get(i).getBasicMtbfRegulate() != null) { + cell.setCellValue(List.get(i).getBasicMtbfRegulate()); + } + cell = row.createCell(9); + if (List.get(i).getBasicMtbfRegulSuccRate() != null) { + cell.setCellValue(List.get(i).getBasicMtbfRegulSuccRate()); + } + cell = row.createCell(10); + if (List.get(i).getBasicMtbfOperatingRatio() != null) { + cell.setCellValue(List.get(i).getBasicMtbfOperatingRatio()); + } else { + cell.setCellValue(1.0); + } + cell = row.createCell(11); + if (List.get(i).getTaskMtbcfRegulate() != null) { + cell.setCellValue(List.get(i).getTaskMtbcfRegulate()); + } + cell = row.createCell(12); + if (List.get(i).getTaskMtbcfOtherParams2() != null) { + cell.setCellValue(List.get(i).getTaskMtbcfOtherParams2()); + } + cell = row.createCell(13); + if (List.get(i).getTaskMtbcfOtherParams3() != null) { + cell.setCellValue(List.get(i).getTaskMtbcfOtherParams3()); + } + cell = row.createCell(14); + if (List.get(i).getTaskMtbcfRegulSuccRate() != null) { + cell.setCellValue(List.get(i).getTaskMtbcfRegulSuccRate()); + } + cell = row.createCell(15); + if (List.get(i).getTaskMtbcfOperatingRatio() != null) { + cell.setCellValue(List.get(i).getTaskMtbcfOperatingRatio()); + } else { + cell.setCellValue(1.0); + } + cell = row.createCell(16); + if (List.get(i).getRepairable() != null) { + isNot = isOrNot.get(List.get(i).getRepairable()); + cell.setCellValue(isNot); + } + cell = row.createCell(17); + if (List.get(i).getRepairDistribType() != null) { + type = repairType.get(List.get(i).getRepairDistribType()); + cell.setCellValue(type); + } + cell = row.createCell(18); + if (List.get(i).getRepairMttcr() != null) { + cell.setCellValue(List.get(i).getRepairMttcr()); + } + cell = row.createCell(19); + if (List.get(i).getRepairMttcrOtherParams2() != null) { + cell.setCellValue(List.get(i).getRepairMttcrOtherParams2()); + } + cell = row.createCell(20); + if (List.get(i).getRepairMttcrOtherParams3() != null) { + cell.setCellValue(List.get(i).getRepairMttcrOtherParams3()); + } + } + } + response.setContentType("application/octet-stream;charset=UTF-8"); + String encodedFilename = DownloadService.getNameEncoder(request, name + "_浜у搧缁撴瀯鏍�.xlsx"); + response.addHeader("Content-Disposition", "attachment;filename=" + encodedFilename); + workbook.write(response.getOutputStream()); + + } catch ( + IOException e) { + e.printStackTrace(); + } + + } + + private void writeExcel(List<ProductDto> data, List<ProductDto> List) { + for (ProductDto dto : data) { + List.add(dto); + writeExcel(dto.getChildren(), List); + } } } -- Gitblit v1.9.1