package com.zt.life.modules.mainPart.taskReliability.service; import cn.hutool.json.JSONArray; import cn.hutool.json.JSONObject; import cn.hutool.json.XML; import com.zt.common.exception.RenException; import com.zt.common.service.BaseService; import com.zt.common.servlet.Result; import com.zt.common.utils.CacheUtils; import com.zt.common.utils.CommonUtils; import com.zt.common.utils.TreeUtils; import com.zt.common.utils.UUIDUtil; import com.zt.core.shiro.ImportUtil; import com.zt.life.modules.mainPart.basicInfo.dao.ParamDataDao; import com.zt.life.modules.mainPart.basicInfo.dao.XhProductModelDao; import com.zt.life.modules.mainPart.basicInfo.model.ParamData; import com.zt.life.modules.mainPart.basicInfo.model.XhProductModel; import com.zt.life.modules.mainPart.taskReliability.dao.*; import com.zt.life.modules.mainPart.taskReliability.model.*; import org.apache.commons.io.IOUtils; import org.apache.commons.lang3.StringUtils; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.Workbook; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.dom4j.Document; import org.dom4j.DocumentHelper; import org.dom4j.Element; import org.dom4j.io.OutputFormat; import org.dom4j.io.XMLWriter; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.*; import java.util.*; import java.util.stream.Collectors; /** * product_model * * @author zt generator * @since 1.0.0 2024-02-29 */ @Service public class ReliabilityAssessService extends BaseService { private static final Logger logger = LoggerFactory.getLogger(ReliabilityAssessService.class); public static final String RELIA_ASSESS_TASK_TYPE_SIMULATION = "assessReq"; @Value("${spring.redis.host}") private String redisHost; @Value("${spring.redis.port}") private String redisPort; @Value("${data.assessLib.mainPy}") private String assessMain; @Value("${data.assessLib.resultHome}") private String resultHome; @Value("${data.assessLib.resultHome}") private String path; @Autowired private RedisTemplate redisTemplate; @Autowired private TaskService taskService; @Autowired private SimulatAssessService simulatAssessService; @Autowired private AssessResultDao assessResultDao; @Autowired private XhProductModelDao xhProductModelDao; @Autowired private ParamDataDao paramDataDao; @Autowired private ModelNodeAlgorithmDao modelNodeAlgorithmDao; @Autowired private AlgorithmDao algorithmDao; @Autowired private OperatConditModelDao operatConditModelDao; @Autowired private ReliabilityAssessDao reliabilityAssessDao; public List page(AssessResult assessResult) { Long assessId = baseDao.getAssessId(assessResult.getItemId(), assessResult.getProductId(), assessResult.getTaskId()); List list = this.getAssess(assessResult, assessId); if (list != null) { list = TreeUtils.build(list); } return list; } @Transactional(rollbackFor = Exception.class) public List assess(AssessResult assessResult) { Result result = null; Long assessId = UUIDUtil.generateId(); assessResult.setId(assessId); // 1. 组装供算法库评定的模型xml assembleModelXml(assessResult); // 2. 调用算法库,进行评定计算 result = callReliaAssessLib(assessResult); if (result.getCode() != 0) throw new RenException(result.getMsg()); assessResultDao.insert(assessResult); List list = this.getAssess(assessResult, assessId); if (list != null) { list = TreeUtils.build(list); } return list; } public List getAssess(AssessResult assessResult, Long assessId) { List list = baseDao.getProductList(assessResult.getProductId(), assessResult.getItemId()); String filePath = path + "/" + assessId + "/" + "result.xml"; String xml; InputStream in = null; File file = new File(filePath); if (file.exists()) { try { in = new FileInputStream(filePath); xml = IOUtils.toString(in); } catch (IOException e) { throw new RenException("文件不存在或者文件打不开"); } finally { try { in.close(); } catch (IOException e) { e.printStackTrace(); } } JSONObject xmlJSONObj = XML.toJSONObject(xml); //JSONObject jsonObject = xmlJSONObj.getJSONObject("Results"); JSONArray jsonArray = xmlJSONObj.getJSONObject("Results").getJSONArray("Result"); for (ReliabilityAssess assess : list) { Double result1 = null; for (int i = 0; i < jsonArray.size(); i++) { String nameValue = jsonArray.getJSONObject(i).get("name").toString(); if (nameValue.contains(assess.getId().toString())) { // 获取对应的值 if (jsonArray.getJSONObject(i).get("value") instanceof Double) { result1 = (Double) jsonArray.getJSONObject(i).get("value"); } else { String[] values = jsonArray.getJSONObject(i).get("value").toString().split(","); result1 = Double.valueOf(values[1]); } break; // 如果找到就退出循环 } } if (result1 != null) { result1 = Double.valueOf(new Formatter().format("%.2f", result1).toString()); } assess.setAssessResult(result1); } } else { list = null; } return list; } // 建立产品结构树时,相同的设备多个使用时,多个设备使用相同的名称。这样,在此处通过名称来判断是否是相同的设备,相同的设备,需要拿出来作为子标签嵌套进xml。 private void assembleModelXml(AssessResult assessResult) { Long productId = assessResult.getProductId(); Long taskId = assessResult.getTaskId(); Long itemId = assessResult.getItemId(); List assessDataList = reliabilityAssessDao.getProductList(productId, itemId); Task task = taskService.get(taskId); XhProductModel product = xhProductModelDao.getById(productId); List productList = xhProductModelDao.getByShipId(productId); List paramDataList = paramDataDao.getDeviceParams(productId); // 1. 计算各任务阶段的运行时长 List taskPhases = simulatAssessService.calcTaskPhaseDuration(task); // 2. 计算各工况模型的运行时长 List taskPhaseModelAll = new ArrayList<>(); for (TaskPhase taskPhase : taskPhases) { simulatAssessService.calcTaskPhaseModelDuration(taskPhase, taskPhaseModelAll); } // 3. 将各工况模型递归拆解为完整的可供算法包仿真计算的模型 try { Document document = DocumentHelper.createDocument(); // 添加root节点 Element root = document.addElement("des"); root.addAttribute("name", "General system"); root.addAttribute("gama", assessResult.getConfidence().toString()); addTasksTag(taskPhaseModelAll, root); addModelsTag(taskId, productId, product.getName(), product.getNamePath(), productList, paramDataList, taskPhaseModelAll, assessDataList, root); // 输出格式化xml XMLWriter xmlWriter = null; try { OutputFormat format = OutputFormat.createPrettyPrint(); format.setEncoding("UTF-8"); StringWriter writer = new StringWriter(); xmlWriter = new XMLWriter(writer, format); xmlWriter.write(document); assessResult.setXml(writer.toString()); } finally { if (xmlWriter != null) xmlWriter.close(); } } catch (Exception e) { e.printStackTrace(); throw new RenException("组装算法库仿真计算用模型XML失败: " + e.getMessage()); } } private void addModelsTag(Long taskId, Long productId, String productName, String productNamePath, List productList, List paramDataList, List taskPhaseModelAll, List assessDataList, Element root) { Map vCounter = new HashMap<>(); vCounter.put("v", 0); Element modelsTag = root.addElement("models"); for (TaskPhaseModel taskPhaseModel : taskPhaseModelAll) { Element modelTag = modelsTag.addElement("model"); modelTag.addAttribute("name", taskPhaseModel.getOperatConditId().toString()); List gkModelsAssembled = operatConditModelDao.getGKModelAssembled( taskPhaseModel.getOperatConditId()); OperatConditModel gkModelTop = gkModelsAssembled.stream().filter(item -> productId.equals(item.getProductId())).collect(Collectors.toList()).get(0); List modelNodeAndVnodeList = modelNodeAlgorithmDao.getListByModelId(gkModelTop.getModelId()); List algorithmList = algorithmDao.getListByModelId(gkModelTop.getModelId()); // 将模型转换为DOM,添加到model标签 Algorithm endAlgo = algorithmList.stream().filter(item -> "end".equals(item.getAlgorithmType())).collect(Collectors.toList()).get(0); ModelNode computerNode = modelNodeAndVnodeList.stream().filter(item -> endAlgo.getComputerList().equals(item.getId().toString())).collect(Collectors.toList()).get(0); List list = new ArrayList<>(); list.add(computerNode); node2DOM(taskId, taskPhaseModel, gkModelTop.getModelId(), productId, productName, productNamePath, productList, paramDataList, gkModelsAssembled, algorithmList, modelNodeAndVnodeList, assessDataList, false, null, list, null, null, null, null, modelTag, vCounter); } } // 递归函数 private void node2DOM(Long taskId, TaskPhaseModel taskPhaseModel, Long modelId, Long productId, String productName, String productNamePath, List productList, List paramDataList, List gkModelsAssembled, List algorithmList, List modelNodeAndVnodeList, List assessDataList, boolean sameNameGroup, String sameNameGroupAlgorithmType, List nodeList, Integer distType, String NoE, String ToE, String F, Element parent, Map vCounter) { if (sameNameGroup) { // 相同设备的分组 Long dataId = nodeList.get(0).getDataId(); XhProductModel product = productList.stream().filter(item -> dataId.equals(item.getId())).collect(Collectors.toList()).get(0); ParamData paramData = paramDataList.stream().filter(item -> dataId.equals(item.getProductId())).collect(Collectors.toList()).get(0); String NoEStr = ""; String ToEStr = ""; String FStr = ""; // for (ModelNode node : nodeList) { List assessDatas = assessDataList.stream().filter(item -> dataId.equals(item.getId())).collect(Collectors.toList()); NoEStr += ","; ToEStr += ","; FStr += ","; if (assessDatas.size() > 0) { NoEStr += assessDatas.get(0).getRunNum() != null ? assessDatas.get(0).getRunNum().toString() : "0"; ToEStr += assessDatas.get(0).getRunTimes() != null ? assessDatas.get(0).getRunTimes().toString() : "0"; FStr += assessDatas.get(0).getFailNum() != null ? assessDatas.get(0).getFailNum().toString() : "0"; } // } NoEStr = NoEStr.substring(1); ToEStr = ToEStr.substring(1); FStr = FStr.substring(1); int counter = vCounter.get("v"); counter++; vCounter.put("v", counter); Element sameNameGroupTag = parent.addElement("logic"); sameNameGroupTag.addAttribute("name", "v" + counter); if ("parallel".equals(sameNameGroupAlgorithmType)) { sameNameGroupTag.addAttribute("type", "parallel"); } else { sameNameGroupTag.addAttribute("type", "series"); } sameNameGroupTag.addAttribute("nums", String.valueOf(nodeList.size())); Element nodeTag = sameNameGroupTag.addElement("node"); nodeTag.addAttribute("name", nodeList.stream().map(item -> item.getDataId().toString() + "-" + item.getDeviceNo().toString()). collect(Collectors.joining(","))); nodeTag.addAttribute("real_name", product.getSameSbName()); nodeTag.addAttribute("name_path", product.getNamePath()); if (paramData.getReliabDistribType() == 3) { // 二项分布 nodeTag.addAttribute("distType", "ber"); nodeTag.addAttribute("NoE", NoEStr); } else { nodeTag.addAttribute("distType", "exp"); nodeTag.addAttribute("ToE", ToEStr); } nodeTag.addAttribute("F", FStr); } else { // 不同设备(只有1个) ModelNode node = nodeList.get(0); if ("node".equals(node.getNodeType())) { Long dataId = node.getDataId(); XhProductModel product = productList.stream().filter(item -> dataId.equals(item.getId())).collect(Collectors.toList()).get(0); if ("3,4,10".contains(product.getProductType())) { // 系统、分系统、虚单元 OperatConditModel gkModel = gkModelsAssembled.stream().filter(item -> dataId.equals(item.getProductId())).collect(Collectors.toList()).get(0); List algorithmListSub = algorithmDao.getListByModelId(gkModel.getModelId()); List modelNodeAndVnodeListSub = modelNodeAlgorithmDao.getListByModelId(gkModel.getModelId()); Algorithm endAlgo = algorithmListSub.stream().filter(item -> "end".equals(item.getAlgorithmType())).collect(Collectors.toList()).get(0); ModelNode computerNode = modelNodeAndVnodeListSub.stream().filter(item -> endAlgo.getComputerList().equals(item.getId().toString())).collect(Collectors.toList()).get(0); List list = new ArrayList<>(); list.add(computerNode); node2DOM(taskId, taskPhaseModel, gkModel.getModelId(), product.getId(), product.getName(), product.getNamePath(), productList, paramDataList, gkModelsAssembled, algorithmListSub, modelNodeAndVnodeListSub, assessDataList, false, null, list, null, null, null, null, parent, vCounter); } else if ("5".equals(product.getProductType())) { // 设备 Element nodeTag = parent.addElement("node"); nodeTag.addAttribute("name", dataId.toString()); nodeTag.addAttribute("real_name", product.getName()); nodeTag.addAttribute("name_path", product.getNamePath()); if (distType == 3) { // 二项分布 nodeTag.addAttribute("distType", "ber"); nodeTag.addAttribute("NoE", NoE); } else { nodeTag.addAttribute("distType", "exp"); nodeTag.addAttribute("ToE", ToE); } nodeTag.addAttribute("F", F); } } else { // vnode(运算节点) Algorithm algo = algorithmList.stream().filter(item -> node.getId().equals(item.getId())).collect(Collectors.toList()).get(0); if (!"series,parallel,vote".contains(algo.getAlgorithmType())) { throw new RenException("可靠性评定功能目前只支持串联、并联和表决。"); } Element element = parent.addElement("logic"); if (null == productId) { element.addAttribute("name", algo.getId().toString()); } else { element.addAttribute("name", productId.toString()); element.addAttribute("real_name", productName); element.addAttribute("name_path", productNamePath); } element.addAttribute("type", algo.getAlgorithmType()); String[] computerNodeListStr = algo.getComputerList().split(","); List distTypes = new ArrayList<>(); List nodes = new ArrayList<>(); List assessDatas = new ArrayList<>(); for (String nodeStr : computerNodeListStr) { ModelNode mn = modelNodeAndVnodeList.stream().filter(item -> nodeStr.equals(item.getId().toString())).collect(Collectors.toList()).get(0); nodes.add(mn); if (mn.getDataId() == null) { distTypes.add(null); assessDatas.add(null); } else { List paramDatas1 = paramDataList.stream().filter(item -> mn.getDataId().equals(item.getProductId())).collect(Collectors.toList()); if (paramDatas1.size() > 0) { distTypes.add(paramDatas1.get(0).getReliabDistribType()); } else { distTypes.add(null); } List assessDatas1 = assessDataList.stream().filter(item -> mn.getDataId().equals(item.getId())).collect(Collectors.toList()); if (assessDatas1.size() > 0) { assessDatas.add(assessDatas1.get(0)); } else { assessDatas.add(null); } } } // 根据同类设备名称进行分组处理(改为根据dataId进行分组) Map> groupByNameMap = new HashMap<>(); for (String nodeStr : computerNodeListStr) { ModelNode mn = modelNodeAndVnodeList.stream().filter(item -> nodeStr.equals(item.getId().toString())).collect(Collectors.toList()).get(0); String sameSbName = null; if (mn.getDataId() != null) { // XhProductModel product = productList.stream().filter(item -> // mn.getDataId().equals(item.getId())).collect(Collectors.toList()).get(0); // if ("5".equals(product.getProductType())) sameSbName = product.getSameSbName(); sameSbName = mn.getDataId().toString(); } if (sameSbName == null) { List list = new ArrayList<>(); list.add(mn); groupByNameMap.put(mn.getId().toString(), list); } else { if (groupByNameMap.containsKey(sameSbName)) { groupByNameMap.get(sameSbName).add(mn); } else { List list = new ArrayList<>(); list.add(mn); groupByNameMap.put(sameSbName, list); } } } // 表决 if ("vote".equals(algo.getAlgorithmType())) { element.addAttribute("k", algo.getVoteNum().toString()); if (groupByNameMap.size() == 1) { // 只有1个分组,说明设备都相同 element.addAttribute("nums", String.valueOf(computerNodeListStr.length)); Element nodeTag = element.addElement("node"); createNodeTagSameDevice(nodes, distTypes, assessDatas, nodeTag); } else { // 不全是相同设备,按不同设备处理 for (int i = 0; i < computerNodeListStr.length; i++) { List list = new ArrayList<>(); list.add(nodes.get(i)); String NoEStr = "0"; String ToEStr = "0"; String FStr = "0"; if (assessDatas.get(i) != null) { NoEStr = assessDatas.get(i).getRunNum() != null ? assessDatas.get(i).getRunNum().toString() : "0"; ToEStr = assessDatas.get(i).getRunTimes() != null ? assessDatas.get(i).getRunTimes().toString() : "0"; FStr = assessDatas.get(i).getFailNum() != null ? assessDatas.get(i).getFailNum().toString() : "0"; } node2DOM(taskId, taskPhaseModel, modelId, null, null, null, productList, paramDataList, gkModelsAssembled, algorithmList, modelNodeAndVnodeList, assessDataList, false, null, list, distTypes.get(i), NoEStr, ToEStr, FStr, element, vCounter); } } return; } if ("parallel".equals(algo.getAlgorithmType())) { if (groupByNameMap.size() == 1) { // 只有1个分组,说明设备都相同 element.addAttribute("nums", String.valueOf(computerNodeListStr.length)); Element nodeTag = element.addElement("node"); createNodeTagSameDevice(nodes, distTypes, assessDatas, nodeTag); return; } } Iterator iterator = groupByNameMap.keySet().iterator(); while (iterator.hasNext()) { String name = iterator.next(); List list = groupByNameMap.get(name); if (list.size() > 1) { // 相同名称的设备 node2DOM(taskId, taskPhaseModel, modelId, null, null, null, productList, paramDataList, gkModelsAssembled, algorithmList, modelNodeAndVnodeList, assessDataList, true, algo.getAlgorithmType(), list, null, null, null, null, element, vCounter); } else { // 不同名称节点(1个节点) Integer distTypeParam = null; String NoEStr = null; String ToEStr = null; String FStr = null; ModelNode oneNode = list.get(0); if (oneNode.getDataId() != null) { List paramDatas = paramDataList.stream().filter(item -> oneNode.getDataId().equals(item.getProductId())).collect(Collectors.toList()); if (paramDatas.size() > 0) { distTypeParam = paramDatas.get(0).getReliabDistribType(); } List assessDatas1 = assessDataList.stream().filter(item -> oneNode.getDataId().equals(item.getId())).collect(Collectors.toList()); if (assessDatas1.size() > 0) { NoEStr = assessDatas1.get(0).getRunNum() != null ? assessDatas1.get(0).getRunNum().toString() : "0"; ToEStr = assessDatas1.get(0).getRunTimes() != null ? assessDatas1.get(0).getRunTimes().toString() : "0"; FStr = assessDatas1.get(0).getFailNum() != null ? assessDatas1.get(0).getFailNum().toString() : "0"; } } node2DOM(taskId, taskPhaseModel, modelId, null, null, null, productList, paramDataList, gkModelsAssembled, algorithmList, modelNodeAndVnodeList, assessDataList, false, null, list, distTypeParam, NoEStr, ToEStr, FStr, element, vCounter); } } } } } private void createNodeTagSameDevice(List nodes, List distTypes, List assessDatas, Element nodeTag) { String NoEStr = ""; String ToEStr = ""; String FStr = ""; for (ReliabilityAssess assessData : assessDatas) { if (assessData != null) { NoEStr += "," + (assessData.getRunNum() != null ? assessData.getRunNum().toString() : "0"); ToEStr += "," + (assessData.getRunTimes() != null ? assessData.getRunTimes().toString() : "0"); FStr += "," + (assessData.getFailNum() != null ? assessData.getFailNum().toString() : "0"); break; } } NoEStr = NoEStr.substring(1); ToEStr = ToEStr.substring(1); FStr = FStr.substring(1); nodeTag.addAttribute("name", nodes.stream().map(item -> item.getDataId().toString() + "-" + item.getDeviceNo().toString()). collect(Collectors.joining(","))); if (distTypes.get(0) == 3) { // 二项分布 nodeTag.addAttribute("distType", "ber"); nodeTag.addAttribute("NoE", NoEStr); } else { nodeTag.addAttribute("distType", "exp"); nodeTag.addAttribute("ToE", ToEStr); } nodeTag.addAttribute("F", FStr); } private void addTasksTag(List taskPhaseModelAll, Element root) { Element tasksTag = root.addElement("tasks"); for (TaskPhaseModel taskPhaseModel : taskPhaseModelAll) { Element taskTag = tasksTag.addElement("task"); taskTag.addAttribute("duration", String.valueOf(taskPhaseModel.getOperatConditDuration())); taskTag.addAttribute("model", taskPhaseModel.getOperatConditId().toString()); } } private Result callReliaAssessLib(AssessResult assessResult) { Result result = null; InputStream is = null; BufferedReader br = null; try { try { setParamToRedis(assessResult); } catch (Exception e) { throw new RenException("访问Redis失败。请检查Redis是否已启动。"); } Process process = null; String command = "python " + assessMain; command += " -ip " + redisHost + " -port " + redisPort; command += " -taskType " + RELIA_ASSESS_TASK_TYPE_SIMULATION + " -taskId " + assessResult.getId().toString(); logger.info("cmd命令为:" + command); if (System.getProperty("os.name").toLowerCase().indexOf("windows") > -1) { process = Runtime.getRuntime().exec(new String[]{"cmd", "/c", command}); } else if (System.getProperty("os.name").toLowerCase().indexOf("linux") > -1) { process = Runtime.getRuntime().exec(new String[]{"/bin/sh", "-c", command}); } else { throw new RenException("暂不支持该操作系统,进行启动算法库计算!"); } is = process.getInputStream(); // 以命令行方式调用算法库时,接口约定返回的结果是utf-8编码 br = new BufferedReader(new InputStreamReader(is, "utf-8")); String line = br.readLine(); logger.info("算法库返回结果:" + line); int exitCode = process.waitFor(); // 同步方式,等待算法库计算完毕 if (line != null) { ReliaSimLibResult rtn = com.alibaba.fastjson.JSONObject.parseObject(line, ReliaSimLibResult.class); if ("0".equals(rtn.getCode())) { logger.info("启动可靠性评定算法库成功。"); if (exitCode == 0) { deleteAssessInfoInRedis(assessResult.getId()); result = Result.ok(); } else { result = Result.error("可靠性评定算法库计算失败: exitCode=" + exitCode); } } else { String errorMsg = rtn.getErrorMsg(); throw new RenException("启动可靠性评定算法库失败: errorMsg=" + errorMsg); } } } catch (Exception e) { logger.error("启动可靠性评定算法库时发生Exception:", e); e.printStackTrace(); throw new RenException("启动可靠性评定算法库失败: errorMsg=" + e.getMessage()); } finally { if (is != null) { try { is.close(); } catch (Exception e) { e.printStackTrace(); } } if (br != null) { try { br.close(); } catch (Exception e) { e.printStackTrace(); } } } return result; } private void setParamToRedis(AssessResult assessResult) { String key = assessResult.getId().toString() + RELIA_ASSESS_TASK_TYPE_SIMULATION; logger.info("redis key:" + key); com.alibaba.fastjson.JSONObject jsonObject = new com.alibaba.fastjson.JSONObject(); jsonObject.put("xmlfile", assessResult.getXml()); jsonObject.put("confidence_level", assessResult.getConfidence()); jsonObject.put("result_home", resultHome); redisTemplate.opsForValue().set(key, jsonObject.toJSONString()); } public void deleteAssessInfoInRedis(Long assessId) { redisTemplate.delete(assessId.toString() + RELIA_ASSESS_TASK_TYPE_SIMULATION); } public List> importProductExcel(MultipartFile mutFile, String progressId, Long shipId, Long itemId) { Date beginDate = new Date(); List> list = new ArrayList<>();//存储所有的导入状态 Map map;//存储每一行的状态 Integer sum = 0; // 总共的条数 Integer suc = 0; // 成功的条数 Integer err = 0; // 失败的条数 int row1 = 0; Map errMap = new HashMap<>(); try { // 获取导入文件的后缀名 String fileName = mutFile.getOriginalFilename(); Workbook workbook = null; //解决excel版本问题 if (fileName != null && fileName.endsWith(".xls")) { workbook = new HSSFWorkbook(mutFile.getInputStream()); } else if (fileName != null && fileName.endsWith(".xlsx")) { workbook = new XSSFWorkbook(mutFile.getInputStream()); } assert workbook != null; //断言如果[boolean表达式]为true,则程序继续执行。 如果为false,则程序抛出AssertionError,并终止执行。 int sheets = workbook.getNumberOfSheets(); Map modelMap = new HashMap<>(); List prductList = baseDao.getProductPath(shipId); for (XhProductModel product : prductList) { if (StringUtils.isNotBlank(product.getNamePath())) { modelMap.put(product.getNamePath(), product); } } String shipPath = ""; String xtPath = ""; String fxtPath = ""; String sbPath = ""; String path = ""; for (int i = 0; i < sheets; i++) { Sheet sheet = workbook.getSheetAt(i); int num = sheet.getLastRowNum(); // 一共有多少行 for (int j = 1; j <= num; j++) { CacheUtils.put(progressId, "speed", CommonUtils.getPercent(j, num)); CacheUtils.put(progressId, "msg", "共" + num + "行,已完成第" + j + "行"); row1 = j; String pattern = "yyyy-MM-dd"; Row row = sheet.getRow(row1); String name = ImportUtil.getCellValue(row, 1, pattern); //节点名称 String type = ImportUtil.getCellValue(row, 2, pattern);//节点类型 String runNumStr = ImportUtil.getCellValue(row, 3, pattern); //运行时长 String runTimeStr = ImportUtil.getCellValue(row, 4, pattern); //运行次数 String failNumStr = ImportUtil.getCellValue(row, 5, pattern); //失败次数 if (StringUtils.isBlank(name) || StringUtils.isBlank(type)) continue; Integer runNum = 0; Double runTime = 0.0; Integer failNum = 0; if (StringUtils.isNotBlank(runNumStr)) { runNum = Integer.valueOf(runNumStr); } if (StringUtils.isNotBlank(runTimeStr)) { runTime = Double.valueOf(runTimeStr); } if (StringUtils.isNotBlank(failNumStr)) { failNum = Integer.valueOf(failNumStr); } Long productId = null; if (type.equals("总体")) { path = shipPath = name; xtPath = ""; fxtPath = ""; productId = shipId; } else { if (type.equals("系统")) { if (StringUtils.isNotBlank(shipPath)) { path = xtPath = name; fxtPath = ""; } else { continue; } } if (type.equals("分系统")) { if (StringUtils.isNotBlank(xtPath)) { path = fxtPath = xtPath + "," + name; } else { continue; } } if (type.equals("设备")) { if ((StringUtils.isNotBlank(xtPath) || StringUtils.isNotBlank(fxtPath))) { path = StringUtils.isNotBlank(fxtPath) ? fxtPath + "," + name : xtPath + "," + name; } else { continue; } } if (modelMap.get(path) != null) { productId = modelMap.get(path).getId(); } } baseDao.deleteByItemId(itemId); ReliabilityAssess assess = new ReliabilityAssess(); assess.setFailNum(failNum); assess.setRunNum(runNum); assess.setRunTimes(runTime); assess.setItemId(itemId); assess.setProductId(productId); this.insert(assess); } } } catch (Exception e) { e.printStackTrace(); ImportUtil.updateErrMap(errMap, "导入异常" + e.getMessage(), "", row1); } Date nowDate = new Date(); String msg = "导入时间:" + CommonUtils.getDatePoor(nowDate, beginDate) + "\r\n"; System.out.println(msg); suc = sum - err; map = new HashMap<>(); map.put("sum", sum + ""); // 所有导入的条数 map.put("suc", suc + ""); // 成功的条数 map.put("err", err + ""); // 失败的条数 list.add(map); list.add(errMap); return list; } public List getAssessDataList(Long productId) { return baseDao.getAssessDataList(productId); } public void downloadXml(HttpServletRequest request, HttpServletResponse response, Long productId, Long taskId, Long itemId, Integer xml) { Long assessId = baseDao.getAssessId(itemId, productId, taskId); String json=""; if (xml == 1) { json = baseDao.getxml(assessId); } else { String filePath = path + "/" + assessId + "/" + "result.xml"; InputStream in = null; File file = new File(filePath); if (file.exists()) { try { in = new FileInputStream(filePath); json = IOUtils.toString(in); } catch (IOException e) { throw new RenException("文件不存在或者文件打不开"); } finally { try { in.close(); } catch (IOException e) { e.printStackTrace(); } } } } simulatAssessService.writeToTxt(request, response, json, "评定"); } }