From 195bb5267a6ece13363303e177fee7d1fa3941aa Mon Sep 17 00:00:00 2001
From: jinlin <jinlin>
Date: 星期二, 19 十一月 2024 15:25:25 +0800
Subject: [PATCH] 修改

---
 modules/mainPart/src/main/java/com/zt/life/modules/mainPart/taskReliability/service/SimulatAssessService.java |  212 +++++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 180 insertions(+), 32 deletions(-)

diff --git a/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/taskReliability/service/SimulatAssessService.java b/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/taskReliability/service/SimulatAssessService.java
index d8696a3..ba5f857 100644
--- a/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/taskReliability/service/SimulatAssessService.java
+++ b/modules/mainPart/src/main/java/com/zt/life/modules/mainPart/taskReliability/service/SimulatAssessService.java
@@ -5,11 +5,13 @@
 import cn.hutool.json.XML;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.spire.pdf.tables.table.convert.Convert;
+import com.zt.common.db.query.QueryFilter;
 import com.zt.common.exception.RenException;
 import com.zt.common.service.BaseService;
 import com.zt.common.servlet.Result;
 import com.zt.common.utils.TreeUtils;
 import com.zt.common.utils.UUIDUtil;
+import com.zt.life.export.service.DownloadService;
 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;
@@ -24,6 +26,7 @@
 import com.zt.life.modules.mainPart.taskReliability.dto.TaskPhaseConstraintDto;
 import com.zt.life.modules.mainPart.taskReliability.model.*;
 import com.zt.life.modules.mainPart.utils.GetStringSpace;
+import com.zt.life.modules.mainPart.utils.OSUtils.ProcessUtils;
 import io.swagger.models.auth.In;
 import org.apache.commons.io.IOUtils;
 import org.apache.commons.lang3.StringUtils;
@@ -44,6 +47,9 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.io.*;
 import java.nio.file.FileAlreadyExistsException;
 import java.util.*;
@@ -115,7 +121,6 @@
     Map<Integer, String> templetsStrMap2 = new HashMap<>();
 
     private JSONObject xmlJSONObj;
-
 
     /**
      * 鍒嗛〉鏌ヨ
@@ -354,9 +359,9 @@
                         currentStatusDto.setX2(nextStatusDto.getX1());
                         statusDtoListNew.add(new TimeDiagramStatusDto(nextStatusDto.getX1(), nextStatusDto.getX2(), nextStatusDto.getStatus()));
                     } else if (nextStatusDto.getX1() > currentStatusDto.getX1() && nextStatusDto.getX2() < currentStatusDto.getX2()) {
-                        currentStatusDto.setX2(nextStatusDto.getX1());
                         statusDtoListNew.add(new TimeDiagramStatusDto(nextStatusDto.getX1(), nextStatusDto.getX2(), nextStatusDto.getStatus()));
                         statusDtoListOld.add(new TimeDiagramStatusDto(nextStatusDto.getX2(), currentStatusDto.getX2(), currentStatusDto.getStatus()));
+                        currentStatusDto.setX2(nextStatusDto.getX1());
                     }
                 } else {
                     if (nextStatusDto.getX1() >= currentStatusDto.getX2()) {
@@ -556,22 +561,30 @@
         List<ProductStatusDto> dbList = this.getProduct(productId, taskId);
         List<ProductStatusDto> childList = new ArrayList<>();
         for (ProductStatusDto item : dbList) {
-            if (item.getSameSbNum() > 1) {
-                for (int i = 1; i <= item.getSameSbNum(); i++) {
-                    try {
-                        ProductStatusDto newRow = item.clone();
-                        newRow.setDeviceNo(i);
-                        newRow.setName(item.getName() + "-" + i);
-                        newRow.setDataId(item.getId().toString() + "-" + i);
-                        childList.add(newRow);
-                    } catch (CloneNotSupportedException e) {
-                        e.printStackTrace();
+            int sameNum = item.getSameSbNum() == null ? 1 : item.getSameSbNum();
+            for (int i = 1; i <= sameNum; i++) {
+                try {
+                    ProductStatusDto newRow = null;
+                    if (sameNum > 1) {
+                        newRow = item.clone();
                     }
+                    else
+                        newRow = item;
+                    if ("5".equals(newRow.getProductType())){
+                        newRow.setDeviceNo(i);
+                        newRow.setDataId(item.getId().toString() + "-" + i);
+                        if (i>1){
+                            newRow.setName(item.getName() + "-" + i);
+                        }
+                    }
+                    else{
+                        newRow.setDeviceNo(0);
+                        newRow.setDataId(item.getId().toString());
+                    }
+                    childList.add(newRow);
+                } catch (CloneNotSupportedException e) {
+                    e.printStackTrace();
                 }
-            } else {
-                item.setDeviceNo(0);
-                item.setDataId(item.getId().toString());
-                childList.add(item);
             }
         }
         childList.sort(Comparator.comparing(ProductStatusDto::getDeviceNo, Comparator.naturalOrder()));
@@ -581,6 +594,7 @@
                 this.processJSONArray(jsonArray, item, taskId);
             }
         }
+        childList.removeIf(item -> item.getStatusList().size() == 0);
         return childList;
     }
 
@@ -655,7 +669,6 @@
         return result;
     }
 
-    @Transactional(rollbackFor = Exception.class)
     public Boolean simulate(SimulatAssess simulatAssess) {
         Boolean result = false;
         // 2. 缁勮渚涚畻娉曞簱浠跨湡鐨勬ā鍨媥ml
@@ -665,6 +678,11 @@
         result = callReliaSimLib(simulatAssess);
 
         return result;
+    }
+
+    public List<SimulatAssess> process(QueryFilter queryFilter) {
+        List<SimulatAssess> page = baseDao.getProcessList(queryFilter.getParams());
+        return queryFilter.getPageList(page);
     }
 
     private Boolean callReliaSimLib(SimulatAssess simulatAssess) {
@@ -690,6 +708,8 @@
             } else {
                 throw new RenException("鏆備笉鏀寔璇ユ搷浣滅郴缁燂紝杩涜鍚姩绠楁硶搴撹绠楋紒");
             }
+            long pid = ProcessUtils.getProcessId(process);
+            logger.info("绠楁硶搴揚ID锛�" + pid);
             is = process.getInputStream();
             // 浠ュ懡浠よ鏂瑰紡璋冪敤绠楁硶搴撴椂锛屾帴鍙g害瀹氳繑鍥炵殑缁撴灉鏄痷tf-8缂栫爜
             br = new BufferedReader(new InputStreamReader(is, "utf-8"));
@@ -700,6 +720,8 @@
                 ReliaSimLibResult rtn = com.alibaba.fastjson.JSONObject.parseObject(line, ReliaSimLibResult.class);
                 if ("0".equals(rtn.getCode())) {
                     logger.info("鍚姩鍙潬鎬т豢鐪熻瘎浼扮畻娉曞簱鎴愬姛銆�");
+                    newProcess(simulatAssess, pid, command);
+                    update(simulatAssess);
                     result = true;
                 } else {
                     String errorMsg = rtn.getErrorMsg();
@@ -728,6 +750,14 @@
         }
 
         return result;
+    }
+
+    public void newProcess(SimulatAssess simulatAssess, long pid, String command) {
+        simulatAssess.setPid(pid);
+        simulatAssess.setCommandLine(command);
+        simulatAssess.setProcessIsAlive(true);
+        simulatAssess.setProcessStartTime(new Date());
+        simulatAssess.setProgress(0);
     }
 
     private void setParamToRedis(SimulatAssess simulatAssess) {
@@ -784,7 +814,8 @@
             Element root = document.addElement("des");
             root.addAttribute("name", "General system");
             addTasksTag(taskPhaseModelAll, root);
-            addModelsTag(taskId,
+            addModelsTag(simulatAssess,
+                    taskId,
                     productId,
                     product.getName(),
                     product.getNamePath(),
@@ -921,6 +952,10 @@
                     ele = repairModelTag.addElement("shape");
                     ele.addAttribute("value", String.valueOf(repairModel.getParam3()));
                     break;
+                case RepairModel.TYPE_DIRAC:
+                    ele = repairModelTag.addElement("repair-time");
+                    ele.addAttribute("value", String.valueOf(repairModel.getParam1()));
+                    break;
                 default:
                     break;
             }
@@ -963,7 +998,8 @@
         return taskPhases;
     }
 
-    private void addModelsTag(Long taskId,
+    private void addModelsTag(SimulatAssess simulatAssess,
+                              Long taskId,
                               Long productId,
                               String productName,
                               String productNamePath,
@@ -989,7 +1025,8 @@
                     "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);
-            node2DOM(taskId,
+            node2DOM(simulatAssess,
+                    taskId,
                     taskPhaseModel,
                     gkModelTop.getModelId(),
                     productId,
@@ -1007,7 +1044,8 @@
     }
 
     // 閫掑綊鍑芥暟
-    private void node2DOM(Long taskId,
+    private void node2DOM(SimulatAssess simulatAssess,
+                          Long taskId,
                           TaskPhaseModel taskPhaseModel,
                           Long modelId,
                           Long productId,
@@ -1037,7 +1075,8 @@
                         "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);
-                node2DOM(taskId,
+                node2DOM(simulatAssess,
+                        taskId,
                         taskPhaseModel,
                         gkModel.getModelId(),
                         product.getId(),
@@ -1055,8 +1094,8 @@
                 // 璁惧
                 Integer deviceNo = node.getDeviceNo();
                 Element nodeTag = parent.addElement("node");
-                nodeTag.addAttribute("name", deviceNo == 0 ? dataId.toString() : dataId.toString() + "-" + deviceNo);
-                nodeTag.addAttribute("real_name", deviceNo == 0 ? product.getName() : product.getName() + "-" + deviceNo);
+                nodeTag.addAttribute("name", dataId.toString() + "-" + deviceNo);
+                nodeTag.addAttribute("real_name", product.getName() + "-" + deviceNo);
                 nodeTag.addAttribute("name_path", product.getNamePath());
                 nodeTag.addAttribute("type", "node");
 
@@ -1077,7 +1116,7 @@
                             .filter(item -> taskId.equals(item.getTaskId()))
                             .filter(item -> taskPhaseModel.getPhaseId().equals(item.getPhaseId()))
                             .filter(item -> taskPhaseModel.getOperatConditId().equals(item.getOperatConditId()))
-                            .filter(item -> modelId.equals(item.getModelId()))
+//                            .filter(item -> modelId.equals(item.getModelId()))
                             .filter(item -> dataId.equals(item.getProductId()))
                             .collect(Collectors.toList()).get(0);
                 }
@@ -1088,7 +1127,12 @@
 
                 if (1 == paramData.getRepairable()) {
                     // 鍙淮淇�
-                    RepairModel repairModel = createRepairModel(paramData);
+                    RepairModel repairModel = null;
+                    if (isRepairDirac(simulatAssess.getRepairDiracFlag(), taskPhaseModel, product.getId())) {
+                        repairModel = createRepairModelDirac(0);
+                    } else {
+                        repairModel = createRepairModel(paramData);
+                    }
                     repairModels.add(repairModel);
                     Element repairTag = nodeTag.addElement("repair_model");
                     repairTag.addAttribute("name", repairModel.getId().toString());
@@ -1123,7 +1167,8 @@
             for (String nodeStr : computerNodeListStr) {
                 ModelNode mn = modelNodeAndVnodeList.stream().filter(item ->
                         nodeStr.equals(item.getId().toString())).collect(Collectors.toList()).get(0);
-                node2DOM(taskId,
+                node2DOM(simulatAssess,
+                        taskId,
                         taskPhaseModel,
                         modelId,
                         null,
@@ -1178,23 +1223,46 @@
         return failureModel;
     }
 
+    private boolean isRepairDirac(Boolean repairDiracFlag, TaskPhaseModel taskPhaseModel, Long productId) {
+        boolean result = false;
+
+        if (repairDiracFlag != null && repairDiracFlag) {
+            if (taskPhaseModel.getPhaseConstraints() == null) {
+                result = true;
+            } else {
+                List<TaskPhaseConstraintDto> list = taskPhaseModel.getPhaseConstraints().stream().filter(item ->
+                        productId.equals(item.getProductId())).collect(Collectors.toList());
+                if (list.size() == 0) {
+                    result = true;
+                }
+            }
+        }
+
+        return result;
+    }
+
+    private RepairModel createRepairModelDirac(double time) {
+        RepairModel repairModel = new RepairModel();
+        repairModel.setId(UUIDUtil.generateId());
+        repairModel.setType(RepairModel.TYPE_DIRAC);
+        repairModel.setParam1(0);
+        return repairModel;
+    }
+
     private RepairModel createRepairModel(ParamData paramData) {
         RepairModel repairModel = new RepairModel();
         repairModel.setId(UUIDUtil.generateId());
         switch (paramData.getRepairDistribType()) {
             case 1:
                 // 鎸囨暟鍒嗗竷
-                repairModel.setType(FailureModel.TYPE_EXP);
+                repairModel.setType(RepairModel.TYPE_EXP);
                 repairModel.setParam1(1.0 / paramData.getRepairMttcr());
                 break;
             case 2:
                 // 濞佸竷灏斿垎甯�
-                repairModel.setType(FailureModel.TYPE_WBL);
+                repairModel.setType(RepairModel.TYPE_WBL);
                 repairModel.setParam2(paramData.getRepairMttcrOtherParams2());
                 repairModel.setParam3(paramData.getRepairMttcrOtherParams3());
-                break;
-            case 3:
-                // 浜岄」鍒嗗竷锛堢淮淇病鏈夛級
                 break;
             default:
                 break;
@@ -1551,4 +1619,84 @@
         dto.setCurveList(curveList);
         return dto;
     }
+
+    public void downloadXml(HttpServletRequest request, HttpServletResponse response, String type, Long id, Integer xml) {
+        if (type.equals("fz")) {
+            if (xml == 1) {
+                SimulatAssess simulatAssess = this.get(id);
+                String xml1 = simulatAssess.getXml();
+                writeToTxt(request, response, xml1, "浠跨湡杈撳叆");
+            } else {
+                String filePath = path + "/" + id + "/" + "result.xml";
+                String xml1;
+                InputStream in = null;
+
+                try {
+                    in = new FileInputStream(filePath);
+                    xml1 = IOUtils.toString(in);
+                } catch (IOException e) {
+                    e.printStackTrace();
+                    throw new RenException("鏂囦欢涓嶅瓨鍦ㄦ垨鑰呮枃浠舵墦涓嶅紑");
+                } finally {
+                    try {
+                        if (in != null) {
+                            in.close();
+                        }
+                    } catch (IOException e) {
+                        e.printStackTrace();
+                    }
+                }
+                writeToTxt(request, response, xml1, "浠跨湡杈撳嚭");
+            }
+        } else if (type.equals("sxt")) {
+            if (xml == 1) {
+                String filePath = path + "/" + id + "/" + "status.json";
+                ObjectMapper mapper = new ObjectMapper();
+                String jsonStr = null;
+                try {
+                    // 浣跨敤 ObjectMapper 鐨� readValue 鏂规硶锛屽皢鏂囦欢涓殑 JSON 鏁版嵁杞崲涓轰竴涓� Java 瀵硅薄
+                    // 杩欓噷浣跨敤 Object 绫讳綔涓烘硾鍨嬪弬鏁帮紝琛ㄧず浠绘剰绫诲瀷鐨勫璞�
+                    Object obj = mapper.readValue(new File(filePath), Object.class);
+                    // 浣跨敤 ObjectMapper 鐨� writeValueAsString 鏂规硶锛屽皢 Java 瀵硅薄杞崲涓� JSON 瀛楃涓�
+                    jsonStr = mapper.writeValueAsString(obj);
+
+                } catch (IOException e) {
+                    // 澶勭悊寮傚父
+                    e.printStackTrace();
+                    throw new RenException("鏂囦欢涓嶅瓨鍦ㄦ垨鑰呮枃浠舵墦涓嶅紑");
+                }
+                writeToTxt(request, response, jsonStr, "鏃跺簭鍥捐緭鍏�");
+            } else {
+                SimulatAssess simulatAssess = this.get(id);
+                TimeDiagramDto timeDiagram = getTimeDiagram(simulatAssess.getProductId(), simulatAssess.getTaskModelId(), simulatAssess.getId(), 1200, 5);
+                String xml2 = timeDiagram.getDiagramJson();
+                writeToTxt(request, response, xml2, "鏃跺簭鍥捐緭鍑�");
+            }
+        }
+    }
+
+    public void writeToTxt(HttpServletRequest request, HttpServletResponse response, String jsonString, String name) {//璁剧疆鍝嶅簲鐨勫瓧绗﹂泦
+        //璁剧疆鍝嶅簲鍐呭鐨勭被鍨�
+        BufferedOutputStream buff = null;
+        ServletOutputStream outStr = null;
+        try {
+            response.setContentType("text/plain;charset=UTF-8");
+            String encodedFilename = DownloadService.getNameEncoder(request, name + "xml.txt");
+            response.addHeader("Content-Disposition", "attachment;filename=" + encodedFilename);
+            outStr = response.getOutputStream();
+            buff = new BufferedOutputStream(outStr);
+            buff.write(jsonString.getBytes("UTF-8"));
+            buff.flush();
+            buff.close();
+        } catch (Exception e) {
+            e.printStackTrace();
+        } finally {
+            try {
+                buff.close();
+                outStr.close();
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+    }
 }

--
Gitblit v1.9.1