|  |  |  | 
|---|
|  |  |  | import org.springframework.web.bind.annotation.*; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import java.util.ArrayList; | 
|---|
|  |  |  | import java.util.Comparator; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.stream.Collectors; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | private RtuUpgradeSv sv ; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 强制结束升级任务通信中间件成功执行后 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | public static void afterMwForceOverCurUgTask(){ | 
|---|
|  |  |  | cache = null ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * rtu远程升级任务通信中间件执行情况统计回收 | 
|---|
|  |  |  | * @param info 数据 | 
|---|
|  |  |  | * @return 操作结果 | 
|---|
|  |  |  | 
|---|
|  |  |  | @Hidden //不公开接口,其只有通信中间件调用 | 
|---|
|  |  |  | @PostMapping(path = "/receive") | 
|---|
|  |  |  | public BaseResponse<Boolean> receive(@RequestBody UpgradeInfo info, HttpServletRequest req, HttpServletResponse rep){ | 
|---|
|  |  |  | /* | 
|---|
|  |  |  | log.info("接收到的RTU远程升级状态数据为:{}", info.toString("532328000214")); | 
|---|
|  |  |  | //进行排序 | 
|---|
|  |  |  | //Comparator<UpgradeRtu> comparator = Comparator.comparing(UpgradeRtu::getRtuAddr, Comparator.naturalOrder()); | 
|---|
|  |  |  | //info.ugRtuStateList = info.ugRtuStateList.stream().sorted(comparator).collect(Collectors.toList()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //通信中间件传过来的机构tag,以用于查找数据源 | 
|---|
|  |  |  | String token = req.getHeader(Constant.UserTokenKeyInHeader); | 
|---|
|  |  |  | DataSourceContext.set(token); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(info.ugRtuStateList != null && info.ugRtuStateList.size() > 0){ | 
|---|
|  |  |  | if(info.ugRtuStateList.size() < 10){ | 
|---|
|  |  |  | for (UpgradeRtu rtuVo : info.ugRtuStateList) { | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | */ | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //进行排序 | 
|---|
|  |  |  | Comparator<UpgradeRtu> comparator = Comparator.comparing(UpgradeRtu::getRtuAddr, Comparator.naturalOrder()); | 
|---|
|  |  |  | info.ugRtuStateList = info.ugRtuStateList.stream().sorted(comparator).collect(Collectors.toList()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //通信中间件传过来的机构tag,以用于查找数据源 | 
|---|
|  |  |  | String token = req.getHeader(Constant.TokenKeyInHeader); | 
|---|
|  |  |  | DataSourceContext.set(token); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(cache == null){ | 
|---|
|  |  |  | cache = info; | 
|---|
|  |  |  | //此时不做任务操作,只保障cache不为空, 等待下次发来数据 | 
|---|
|  |  |  | if(info.ugRtuStateList != null && info.ugRtuStateList.size() > 0){ | 
|---|
|  |  |  | List<UpgradeRtu> overList = info.ugRtuStateList.stream().filter(itemVo -> itemVo.isOver).collect(Collectors.toList()) ; | 
|---|
|  |  |  | if(overList != null && overList.size() > 0){ | 
|---|
|  |  |  | cache = info; | 
|---|
|  |  |  | this.save2Db(info.ugTaskId, overList); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | if(info.ugRtuStateList != null && info.ugRtuStateList.size() > 0){ | 
|---|
|  |  |  | //此时保证两个集合都不为null | 
|---|
|  |  |  | this.save2Db(info.ugTaskId, info.ugRtuStateList, cache.ugRtuStateList); | 
|---|
|  |  |  | cache = info; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //cache赋值必须放在上面处理的后面,否则上面的比较不成功 | 
|---|
|  |  |  | cache = info; | 
|---|
|  |  |  | if(info.ugOverallState != null && info.ugOverallState.allOver){ | 
|---|
|  |  |  | this.saveTaskOver(info.ugTaskId) ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return null; | 
|---|
|  |  |  | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private void save2Db(String taskId, List<UpgradeRtu> newList, List<UpgradeRtu> oldList){ | 
|---|
|  |  |  | List<UpgradeRtu> newOverList = newList.stream().filter(vo -> vo.isOver).collect(Collectors.toList()) ; | 
|---|
|  |  |  | List<UpgradeRtu> oldNoOverList = newList.stream().filter(vo -> !vo.isOver).collect(Collectors.toList()) ; | 
|---|
|  |  |  | boolean oldExist = false ; | 
|---|
|  |  |  | for(UpgradeRtu nvo : newOverList){ | 
|---|
|  |  |  | if(nvo != null) { | 
|---|
|  |  |  | if(oldList.stream().anyMatch(vo -> vo.rtuAddr.equals(nvo.rtuAddr) && vo.isOver == false)){ | 
|---|
|  |  |  | //上次没有升级结束,而当前升级结束了 | 
|---|
|  |  |  | this.sv.saveRtuUpgradeState(Long.parseLong(taskId), nvo); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | oldExist = false ; | 
|---|
|  |  |  | if(oldNoOverList.stream().anyMatch(vo -> vo.rtuAddr.equals(nvo.rtuAddr))){ | 
|---|
|  |  |  | oldExist = true ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if(!oldExist){ | 
|---|
|  |  |  | //上次没有升级结束,而当前升级结束了 | 
|---|
|  |  |  | this.sv.saveRtuUpgradeState(Long.parseLong(taskId), nvo); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 保存升级任务已经执行完成 | 
|---|
|  |  |  | * @param taskId | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private void saveTaskOver(String taskId){ | 
|---|
|  |  |  | this.sv.updateTaskOver(taskId) ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | ///////////////////////////////////////////////////// | 
|---|
|  |  |  | // | 
|---|
|  |  |  | // 以下模拟数据 | 
|---|
|  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public Object execute() throws Exception { | 
|---|
|  |  |  | while(!this.stop){ | 
|---|
|  |  |  | if(!runDemo()){ | 
|---|
|  |  |  | if(!runInDemo()){ | 
|---|
|  |  |  | this.stop = true ; | 
|---|
|  |  |  | }else{ | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | private boolean runDemo(){ | 
|---|
|  |  |  | private boolean runInDemo(){ | 
|---|
|  |  |  | for(UpgradeRtu rtu : cache.ugRtuStateList){ | 
|---|
|  |  |  | this.rtuUpgrade(rtu) ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | rtu.currentPackage = 0 ; | 
|---|
|  |  |  | rtu.currentRamAddr = 0x00 ; | 
|---|
|  |  |  | rtu.lastDownDt = "" ; | 
|---|
|  |  |  | rtu.lastDownDtAt = 0L ; | 
|---|
|  |  |  | rtu.reTryTimes = 0 ; | 
|---|
|  |  |  | rtu.isOver = false ; | 
|---|
|  |  |  | }else{ | 
|---|
|  |  |  | 
|---|
|  |  |  | rtu.currentPackage = 1 ; | 
|---|
|  |  |  | rtu.currentRamAddr = 0x00 ; | 
|---|
|  |  |  | rtu.lastDownDt = DateTime.yyyy_MM_dd_HH_mm_ss() ; | 
|---|
|  |  |  | rtu.lastDownDtAt = System.currentTimeMillis() ; | 
|---|
|  |  |  | rtu.reTryTimes = 0 ; | 
|---|
|  |  |  | rtu.isOver = false ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | //离线的,不处理 | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | int n = Integer.parseInt(new CreateRandom().create(2)) ; | 
|---|
|  |  |  | if(n == 54){ | 
|---|
|  |  |  | if(rtu.currentPackage == 1){ | 
|---|
|  |  |  | //1包死 | 
|---|
|  |  |  | rtu.state = UpgradeRtu.STATE_FAILONE ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(n == 45){ | 
|---|
|  |  |  | if(rtu.currentPackage != 1){ | 
|---|
|  |  |  | //升死 | 
|---|
|  |  |  | rtu.state = UpgradeRtu.STATE_FAIL ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(rtu.currentPackage == rtu.totalPackage){ | 
|---|
|  |  |  | //升级结束 | 
|---|
|  |  |  | 
|---|
|  |  |  | rtu.isOver = true ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if(rtu.reTryTimes >= 2){ | 
|---|
|  |  |  | //重试次数达到最大值 | 
|---|
|  |  |  | if(rtu.state == UpgradeRtu.STATE_FAILONE || | 
|---|
|  |  |  | rtu.state == UpgradeRtu.STATE_FAIL){ | 
|---|
|  |  |  | //又失败了,认为结束了 | 
|---|
|  |  |  | rtu.isOver = true ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(rtu.state != UpgradeRtu.STATE_FAILONE && | 
|---|
|  |  |  | rtu.state != UpgradeRtu.STATE_FAIL){ | 
|---|
|  |  |  | int n = Integer.parseInt(new CreateRandom().create(3)) ; | 
|---|
|  |  |  | if(n == 540 || n == 541 || n == 542 || n == 543 || n == 544 || n == 545 || n == 546 || n == 547 || n == 548 || n == 549 ){ | 
|---|
|  |  |  | if(rtu.currentPackage == 1){ | 
|---|
|  |  |  | //1包死 | 
|---|
|  |  |  | rtu.state = UpgradeRtu.STATE_FAILONE ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(n == 450 || n == 451 || n == 452 || n == 453 || n == 454 || n == 455){ | 
|---|
|  |  |  | if(rtu.currentPackage != 1){ | 
|---|
|  |  |  | //升死 | 
|---|
|  |  |  | rtu.state = UpgradeRtu.STATE_FAIL ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if(rtu.state == UpgradeRtu.STATE_FAILONE || | 
|---|
|  |  |  | rtu.state == UpgradeRtu.STATE_FAIL){ | 
|---|
|  |  |  | if(rtu.reTryTimes < 2){ | 
|---|
|  |  |  | rtu.state = UpgradeRtu.STATE_RUNNING ; | 
|---|
|  |  |  | rtu.currentPackage = 1 ; | 
|---|
|  |  |  | rtu.currentRamAddr = 0x00 ; | 
|---|
|  |  |  | rtu.lastDownDt = DateTime.yyyy_MM_dd_HH_mm_ss() ; | 
|---|
|  |  |  | rtu.lastDownDtAt = System.currentTimeMillis() ; | 
|---|
|  |  |  | rtu.reTryTimes++ ; | 
|---|
|  |  |  | rtu.isOver = false ; | 
|---|
|  |  |  | return ; | 
|---|
|  |  |  | 
|---|
|  |  |  | rtu.currentPackage += 1 ; | 
|---|
|  |  |  | rtu.currentRamAddr = 0x00 + UpgradeRtu.RAMADDRADD ; | 
|---|
|  |  |  | rtu.lastDownDt = DateTime.yyyy_MM_dd_HH_mm_ss() ; | 
|---|
|  |  |  | rtu.reTryTimes = 0 ; | 
|---|
|  |  |  | rtu.lastDownDtAt = System.currentTimeMillis() ; | 
|---|
|  |  |  | rtu.isOver = false ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | for(UpgradeRtu rtu : cache.ugRtuStateList){ | 
|---|
|  |  |  | if(rtu.state == UpgradeRtu.STATE_OFFLINE){ | 
|---|
|  |  |  | cache.ugOverallState.offLineTotal ++ ; | 
|---|
|  |  |  | cache.ugOverallState.failTotal++; | 
|---|
|  |  |  | }else if(rtu.state == UpgradeRtu.STATE_UNSTART){ | 
|---|
|  |  |  | cache.ugOverallState.unStartTotal ++ ; | 
|---|
|  |  |  | }else if(rtu.state == UpgradeRtu.STATE_RUNNING){ | 
|---|
|  |  |  | 
|---|
|  |  |  | }else if(rtu.state == UpgradeRtu.STATE_SUCCESS) { | 
|---|
|  |  |  | cache.ugOverallState.successTotal++; | 
|---|
|  |  |  | }else if(rtu.state == UpgradeRtu.STATE_FAILONE) { | 
|---|
|  |  |  | cache.ugOverallState.failOneTotal++; | 
|---|
|  |  |  | cache.ugOverallState.dieOneTotal++; | 
|---|
|  |  |  | cache.ugOverallState.failTotal++; | 
|---|
|  |  |  | }else if(rtu.state == UpgradeRtu.STATE_FAIL) { | 
|---|
|  |  |  | cache.ugOverallState.dieMultiTotal++; | 
|---|
|  |  |  | cache.ugOverallState.failTotal++; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if(rtu.isOver){ | 
|---|
|  |  |  | 
|---|
|  |  |  | if(!hasRunning){ | 
|---|
|  |  |  | cache.ugOverallState.allOver = true ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if(cache.ugOverallState.allOver){ | 
|---|
|  |  |  | cache.ugOverallState.overTotal = 0; | 
|---|
|  |  |  | if(cache.ugRtuStateList != null && cache.ugRtuStateList.size() > 0){ | 
|---|
|  |  |  | for(UpgradeRtu rtu : cache.ugRtuStateList){ | 
|---|
|  |  |  | rtu.isOver = true ; | 
|---|
|  |  |  | cache.ugOverallState.overTotal++; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return hasRunning ; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|