|  |  | 
 |  |  | package com.zt.life.modules.mainPart.taskReliability.controller; | 
 |  |  |  | 
 |  |  |  | 
 |  |  | import cn.hutool.http.HttpUtil; | 
 |  |  | import cn.hutool.json.JSONArray; | 
 |  |  | import cn.hutool.json.JSONObject; | 
 |  |  | import cn.hutool.json.XML; | 
 |  |  | import com.fasterxml.jackson.databind.ObjectMapper; | 
 |  |  | import com.zt.common.annotation.LogOperation; | 
 |  |  | import com.zt.common.constant.Constant; | 
 |  |  | import com.zt.common.annotation.QueryParam; | 
 |  |  | import com.zt.common.constant.Constant; | 
 |  |  | import com.zt.common.db.query.QueryFilter; | 
 |  |  | import com.zt.common.servlet.Result; | 
 |  |  | import com.zt.common.exception.RenException; | 
 |  |  | import com.zt.common.servlet.PageResult; | 
 |  |  | import com.zt.common.utils.UUIDUtil; | 
 |  |  | import com.zt.common.servlet.Result; | 
 |  |  | import com.zt.common.validator.AssertUtils; | 
 |  |  | import com.zt.common.validator.ValidatorUtils; | 
 |  |  | import com.zt.common.validator.group.AddGroup; | 
 |  |  | import com.zt.common.validator.group.DefaultGroup; | 
 |  |  | import com.zt.common.validator.group.UpdateGroup; | 
 |  |  | import com.zt.life.modules.mainPart.taskReliability.model.SimulatAssess; | 
 |  |  | import com.zt.life.modules.mainPart.async.PythonLib; | 
 |  |  | import com.zt.life.modules.mainPart.taskReliability.dto.*; | 
 |  |  | import com.zt.life.modules.mainPart.taskReliability.model.*; | 
 |  |  | import com.zt.life.modules.mainPart.taskReliability.service.SimulatAssessService; | 
 |  |  | import com.zt.life.modules.mainPart.taskReliability.service.TaskService; | 
 |  |  | import com.zt.life.modules.mainPart.utils.OSUtils.ProcessUtils; | 
 |  |  | import io.swagger.annotations.Api; | 
 |  |  | import io.swagger.annotations.ApiImplicitParam; | 
 |  |  | import io.swagger.annotations.ApiImplicitParams; | 
 |  |  | import io.swagger.annotations.ApiOperation; | 
 |  |  | import org.apache.commons.io.IOUtils; | 
 |  |  | import org.apache.commons.lang3.StringUtils; | 
 |  |  | 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.web.bind.annotation.*; | 
 |  |  | import springfox.documentation.annotations.ApiIgnore; | 
 |  |  |  | 
 |  |  | import javax.servlet.http.HttpServletResponse; | 
 |  |  | import java.util.HashMap; | 
 |  |  | import java.util.List; | 
 |  |  | import java.io.File; | 
 |  |  | import java.io.FileInputStream; | 
 |  |  | import java.io.IOException; | 
 |  |  | import java.io.InputStream; | 
 |  |  | import java.util.*; | 
 |  |  |  | 
 |  |  |  | 
 |  |  | /** | 
 |  |  | 
 |  |  | @RequestMapping("/taskReliability/SimulatAssess/") | 
 |  |  | @Api(tags = "simulat_assess") | 
 |  |  | public class SimulatAssessController { | 
 |  |  |     private static final Logger logger = LoggerFactory.getLogger(SimulatAssessController.class); | 
 |  |  |  | 
 |  |  |     @Autowired | 
 |  |  |     private RedisTemplate redisTemplate; | 
 |  |  |  | 
 |  |  |  | 
 |  |  |     @Autowired | 
 |  |  |     private SimulatAssessService simulatAssessService; | 
 |  |  |  | 
 |  |  |     @GetMapping("page") | 
 |  |  |     @ApiOperation("分页") | 
 |  |  |     @ApiImplicitParams({ | 
 |  |  |             @ApiImplicitParam(name = Constant.Q.PAGE, value = Constant.QV.PAGE, required = true, dataType = Constant.QT.INT), | 
 |  |  |             @ApiImplicitParam(name = Constant.Q.LIMIT, value = Constant.QV.LIMIT, required = true, dataType = Constant.QT.INT), | 
 |  |  |             @ApiImplicitParam(name = Constant.Q.ORDER_FIELD, value = Constant.QV.ORDER_FIELD, dataType = Constant.QT.STRING), | 
 |  |  |             @ApiImplicitParam(name = Constant.Q.ORDER, value = Constant.QV.ORDER, dataType = Constant.QT.STRING), | 
 |  |  |     }) | 
 |  |  |     public PageResult<SimulatAssess> page(@ApiIgnore @QueryParam QueryFilter queryFilter) { | 
 |  |  |  | 
 |  |  |         return PageResult.ok(simulatAssessService.page(queryFilter)); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("{id}") | 
 |  |  |     @ApiOperation("信息") | 
 |  |  | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @PostMapping("analyze") | 
 |  |  |     public Result<Long> analyze(@RequestBody SimulatAssess simulatAssess) { | 
 |  |  |         //效验数据 | 
 |  |  |         ValidatorUtils.validateEntity(simulatAssess, AddGroup.class, DefaultGroup.class); | 
 |  |  |         Long taskId = UUIDUtil.generateId(); | 
 |  |  |         Integer time = simulatAssess.getSamplPeriod(); | 
 |  |  |         Integer Frequency = simulatAssess.getSimulatFrequency(); | 
 |  |  |         HashMap<String, Object> paramMap = new HashMap<>(); | 
 |  |  |         paramMap.put("taskId", taskId); | 
 |  |  |         paramMap.put("time", time); | 
 |  |  |         paramMap.put("Frequency", Frequency); | 
 |  |  |         paramMap.put("url", "http://localhost:8050/test-project//taskReliability/SimulatAssess/calculate"); | 
 |  |  |         String result1 = HttpUtil.get("http://localhost:8080/system/user/get", paramMap); | 
 |  |  |     @GetMapping("getSimulatList") | 
 |  |  |     @ApiOperation("信息") | 
 |  |  |     public Result<List<SimulatAssess>> getSimulatList(Long taskModelId) { | 
 |  |  |         List<SimulatAssess> data = simulatAssessService.getList(taskModelId); | 
 |  |  |  | 
 |  |  |         return Result.ok(taskId); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("calculate") | 
 |  |  |     public Result calculate(Long taskId, String List, String code, String errorMsg) { | 
 |  |  |         SimulatAssess simulatAssess = new SimulatAssess(); | 
 |  |  |         simulatAssess.setId(taskId); | 
 |  |  |         simulatAssess.setCurveData(List); | 
 |  |  |         simulatAssess.setCode(code); | 
 |  |  |         simulatAssess.setErrorMsg(errorMsg); | 
 |  |  |         simulatAssessService.insert(simulatAssess); | 
 |  |  |     @GetMapping("getSimulatParams") | 
 |  |  |     @ApiOperation("信息") | 
 |  |  |     public Result<SimulatAssess> getSimulatParams(Long id, Long taskModelId) { | 
 |  |  |         SimulatAssess data = simulatAssessService.getParams(id, taskModelId); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("verify") | 
 |  |  |     public Result<String> verify(Long taskId) { | 
 |  |  |         String result = simulatAssessService.verify(taskId); | 
 |  |  |         result = result.replaceAll("\n", "<br>"); | 
 |  |  |         return Result.ok(result); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @PostMapping("analyze") | 
 |  |  |     public Result<Object> analyze(@RequestBody SimulatAssess simulatAssess) { | 
 |  |  |         //效验数据 | 
 |  |  | //        Result result = pythonLib.callPython(simulatAssess); | 
 |  |  |         List<TaskModelCheckResultDto> result = simulatAssessService.simulateChecK(simulatAssess); | 
 |  |  |         if (result != null) { | 
 |  |  |             return Result.ok(result); | 
 |  |  |         } | 
 |  |  |         Boolean result2 = simulatAssessService.simulate(simulatAssess); | 
 |  |  |         if (result2) | 
 |  |  |             return Result.ok(); | 
 |  |  |         else | 
 |  |  |             return Result.error("算法库计算失败!"); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("process") | 
 |  |  |     @ApiOperation("获取算法库进程信息") | 
 |  |  |     @ApiImplicitParams({ | 
 |  |  |             @ApiImplicitParam(name = Constant.Q.PAGE, value = Constant.QV.PAGE, required = true, dataType = Constant.QT.INT), | 
 |  |  |             @ApiImplicitParam(name = Constant.Q.LIMIT, value = Constant.QV.LIMIT, required = true, dataType = Constant.QT.INT), | 
 |  |  |             @ApiImplicitParam(name = "isAlive", value = "存活", dataType = Constant.QT.Boolean), | 
 |  |  |     }) | 
 |  |  |     public PageResult<SimulatAssess> process(@ApiIgnore @QueryParam QueryFilter queryFilter) { | 
 |  |  |         List<SimulatAssess> page = simulatAssessService.process(queryFilter); | 
 |  |  |         return PageResult.ok(page); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |  | 
 |  |  |     @PostMapping("getResultXML") | 
 |  |  |     public Result<SimulaDataDto> getResultXML(@RequestBody SimulatAssess simulatAssess) { | 
 |  |  |         SimulaDataDto data = simulatAssessService.getResultXML(simulatAssess); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @PostMapping("getResultData") | 
 |  |  |     public Result<SimulaDataDto> getData(@RequestBody SimulatAssess simulatAssess) { | 
 |  |  |         SimulaDataDto data = simulatAssessService.getResultData(simulatAssess); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("getStatus") | 
 |  |  |     public Result<TimeDiagramDto> getTimeDiagram(Long productId, Long taskId, Long fzId, double smallWidth, double minPointNum) { | 
 |  |  |         TimeDiagramDto data = simulatAssessService.getTimeDiagram(productId, taskId, fzId, smallWidth, minPointNum); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("getDiagram") | 
 |  |  |     @ApiOperation("查询流程图") | 
 |  |  |     @LogOperation("查询流程图") | 
 |  |  |     public Result<TimeDiagram> getDiagram(String projectId, String diagramId, String showType, String isShow, String digramParams, String majorId, Integer winWidth, Integer winHeight) throws Exception { | 
 |  |  |         if (StringUtils.isBlank(diagramId)) | 
 |  |  |             diagramId = projectId; | 
 |  |  |         TimeDiagram diagram = simulatAssessService.getDiagram(projectId, diagramId, showType, isShow, digramParams, majorId, winWidth, winHeight); | 
 |  |  |         return Result.ok(diagram); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @PutMapping("saveDiagram") | 
 |  |  |     @ApiOperation("查询流程图") | 
 |  |  |     @LogOperation("查询流程图") | 
 |  |  |     public Result saveDiagram(@RequestBody TimeDiagram diagram) { | 
 |  |  |         // 效验数据 | 
 |  |  |         //ValidatorUtils.validateEntity(project.getProject(), UpdateGroup.class, DefaultGroup.class); | 
 |  |  |         if (diagram.getConfigChange() == null) | 
 |  |  |             diagram.setConfigChange(0); | 
 |  |  |         diagram.setSaveStatus(2); | 
 |  |  |         simulatAssessService.updateDiagram(diagram); | 
 |  |  |         //simulatAssessService.combineDiagram(diagram); | 
 |  |  |         return Result.ok(); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("getCalcProgress") | 
 |  |  |     public Result getCalcProgress(@RequestParam("taskList[]") Long[] taskList) { | 
 |  |  |         Integer progress = 0; | 
 |  |  |         String progress2 = ""; | 
 |  |  |         String key = null; | 
 |  |  |         if (taskList.length == 1) { | 
 |  |  |             Long taskId = taskList[0]; | 
 |  |  |             SimulatAssess simulatAssess = simulatAssessService.get(taskId); | 
 |  |  |             boolean isProcessAlive = ProcessUtils.isProcessAlive(simulatAssess.getPid()); | 
 |  |  |             key = taskId.toString() + simulatAssessService.RELIA_SIM_TASK_TYPE_PROGRESS; | 
 |  |  |             progress2 = (String) redisTemplate.opsForValue().get(key); | 
 |  |  |             if (progress2 == null) progress2 = "0"; | 
 |  |  |             progress = Integer.parseInt(progress2); | 
 |  |  |             if (progress == null) progress = 0; | 
 |  |  |             updateSimulateAssess(simulatAssess, isProcessAlive, progress); | 
 |  |  |             if (progress < 100 && !isProcessAlive) { | 
 |  |  |                 logger.error("算法库进程在计算途中异常终止了:taskId="+taskId+" pid="+simulatAssess.getPid()+" progress="+progress); | 
 |  |  |                 simulatAssessService.deleteSimInfoInRedis(taskId); | 
 |  |  |                 return Result.error("算法库进程在计算途中异常终止了!"); | 
 |  |  |             } | 
 |  |  |             if (progress == 100) { | 
 |  |  |                 try { | 
 |  |  |                     Thread.sleep(1000); | 
 |  |  |                 } catch (Exception e) { | 
 |  |  |                     e.printStackTrace(); | 
 |  |  |                 } | 
 |  |  |                 isProcessAlive = ProcessUtils.isProcessAlive(simulatAssess.getPid()); | 
 |  |  |                 simulatAssess.setProcessIsAlive(isProcessAlive); | 
 |  |  |                 simulatAssessService.update(simulatAssess); | 
 |  |  | //                simulatAssessService.deleteSimInfoInRedis(taskId); | 
 |  |  |             } | 
 |  |  |         } else { | 
 |  |  |             for (Long taskId : taskList) { | 
 |  |  |                 int progTmp = 0; | 
 |  |  |                 SimulatAssess simulatAssess = simulatAssessService.get(taskId); | 
 |  |  |                 boolean isProcessAlive = ProcessUtils.isProcessAlive(simulatAssess.getPid()); | 
 |  |  |                 key = taskId.toString() + simulatAssessService.RELIA_SIM_TASK_TYPE_PROGRESS; | 
 |  |  |                 progress2 = (String) redisTemplate.opsForValue().get(key); | 
 |  |  |                 if (progress2 == null) progress2 = "0"; | 
 |  |  |                 progTmp = Integer.parseInt(progress2); | 
 |  |  |                 updateSimulateAssess(simulatAssess, isProcessAlive, progTmp); | 
 |  |  |                 if (progTmp < 100 && !isProcessAlive) { | 
 |  |  |                     logger.error("算法库进程在计算途中异常终止了:taskId="+taskId+" pid="+simulatAssess.getPid()+" progress="+progTmp); | 
 |  |  | //                    simulatAssessService.deleteSimInfoInRedis(taskId); | 
 |  |  |                     return Result.error("算法库进程在计算途中异常终止了!"); | 
 |  |  |                 } | 
 |  |  |                 if (progTmp == 100) { | 
 |  |  |                     try { | 
 |  |  |                         Thread.sleep(500); | 
 |  |  |                     } catch (Exception e) { | 
 |  |  |                         e.printStackTrace(); | 
 |  |  |                     } | 
 |  |  |                     isProcessAlive = ProcessUtils.isProcessAlive(simulatAssess.getPid()); | 
 |  |  |                     simulatAssess.setProcessIsAlive(isProcessAlive); | 
 |  |  |                     simulatAssessService.update(simulatAssess); | 
 |  |  | //                    simulatAssessService.deleteSimInfoInRedis(taskId); | 
 |  |  |                 } | 
 |  |  |                 progress = progress + progTmp; | 
 |  |  |             } | 
 |  |  |             progress = progress / taskList.length; | 
 |  |  |         } | 
 |  |  |  | 
 |  |  |         return Result.ok(progress.toString()); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     private void updateSimulateAssess(SimulatAssess simulatAssess, boolean isProcessAlive, int progress) { | 
 |  |  |         if (!isProcessAlive) { | 
 |  |  |             simulatAssess.setProcessIsAlive(false); | 
 |  |  |             simulatAssess.setProgress(progress); | 
 |  |  |             simulatAssess.setProcessEndTime(new Date()); | 
 |  |  |             simulatAssessService.update(simulatAssess); | 
 |  |  |         } else { | 
 |  |  |             if (simulatAssess.getProgress() != progress) { | 
 |  |  |                 simulatAssess.setProgress(progress); | 
 |  |  |                 simulatAssessService.update(simulatAssess); | 
 |  |  |             } | 
 |  |  |         } | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("ReliabilityWeakness") | 
 |  |  |     public Result<List<WeakDto>> getReliabilityWeakness(Long fzId, Long taskId, Long productId) { | 
 |  |  |         List<WeakDto> data = simulatAssessService.getReliabilityWeakness(fzId, taskId, productId); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("SchemeCompar") | 
 |  |  |     public Result<SimulatResult> SchemeCompar(@RequestParam("taskList[]") String[] taskList, Long showProductId, Integer samplPeriod, Integer simulatFrequency) { | 
 |  |  |         SimulatResult data = simulatAssessService.SchemeCompar(taskList, showProductId, samplPeriod, simulatFrequency); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("getEcharts") | 
 |  |  |     public Result<SchemeComparDto> getEcharts(@RequestParam("taskList[]") String[] taskList, Long showProductId) { | 
 |  |  |         SchemeComparDto data = simulatAssessService.getEcharts(taskList, showProductId); | 
 |  |  |         return Result.ok(data); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     @GetMapping("killProcess") | 
 |  |  |     public Result<SchemeComparDto> killProcess(@RequestParam("taskId") Long taskId) { | 
 |  |  |         SimulatAssess simulatAssess = simulatAssessService.get(taskId); | 
 |  |  |         ProcessUtils.killProcess(simulatAssess.getPid()); | 
 |  |  |         boolean isProcessAlive = ProcessUtils.isProcessAlive(simulatAssess.getPid()); | 
 |  |  |         simulatAssess.setProcessIsAlive(isProcessAlive); | 
 |  |  |         simulatAssessService.update(simulatAssess); | 
 |  |  |         return Result.ok(); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  | 
 |  |  |         return Result.ok(); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |  | 
 |  |  | } |