|  |  | 
 |  |  | package com.dy.pipIrrRemote.rtuUpgrage; | 
 |  |  |  | 
 |  |  | import com.dy.common.contant.Constant; | 
 |  |  | import com.dy.common.multiDataSource.DataSourceContext; | 
 |  |  | import com.dy.common.softUpgrade.state.UpgradeInfo; | 
 |  |  | import com.dy.common.softUpgrade.state.UpgradeRtu; | 
 |  |  | import com.dy.common.softUpgrade.state.UpgradeState; | 
 |  |  | 
 |  |  | import com.dy.pipIrrGlobal.pojoUg.UgRtuTask; | 
 |  |  | import io.swagger.v3.oas.annotations.Hidden; | 
 |  |  | import io.swagger.v3.oas.annotations.tags.Tag; | 
 |  |  | import jakarta.servlet.http.HttpServletRequest; | 
 |  |  | import jakarta.servlet.http.HttpServletResponse; | 
 |  |  | import lombok.extern.slf4j.Slf4j; | 
 |  |  | import org.springframework.beans.factory.annotation.Autowired; | 
 |  |  | 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){ | 
 |  |  |         log.info("接收到的RTU远程升级状态数据为:{}", info.toString()); | 
 |  |  |         //进行排序 | 
 |  |  |         //Comparator<UpgradeRtu> comparator = Comparator.comparing(UpgradeRtu::getRtuAddr, Comparator.naturalOrder()); | 
 |  |  |         //info.ugRtuStateList = info.ugRtuStateList.stream().sorted(comparator).collect(Collectors.toList()); | 
 |  |  |  | 
 |  |  |     public BaseResponse<Boolean> receive(@RequestBody UpgradeInfo info, HttpServletRequest req, HttpServletResponse rep){ | 
 |  |  |         /* | 
 |  |  |         log.info("接收到的RTU远程升级状态数据为:{}", info.toString("532328000214")); | 
 |  |  |         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 == 4 | 
 |  |  |                 || n == 14 | 
 |  |  |                 || n == 24 | 
 |  |  |                 || n == 34 | 
 |  |  |                 || n == 44 | 
 |  |  |                 || n == 54 | 
 |  |  |                 || n == 64 | 
 |  |  |                 || n == 74 | 
 |  |  |                 || n == 84 | 
 |  |  |                 || n == 95 | 
 |  |  |                 || n == 45 | 
 |  |  |                 || n == 46 | 
 |  |  |                 || n == 47 | 
 |  |  |                 || n == 48 | 
 |  |  |                 || n == 49){ | 
 |  |  |  | 
 |  |  |         if(rtu.currentPackage == rtu.totalPackage){ | 
 |  |  |             //升级结束 | 
 |  |  |             rtu.state = UpgradeRtu.STATE_SUCCESS ; | 
 |  |  |             rtu.isOver = true ; | 
 |  |  |             return ; | 
 |  |  |         } | 
 |  |  |         if(rtu.reTryTimes >= 2){ | 
 |  |  |             //重试次数达到最大值 | 
 |  |  |             if(rtu.state == UpgradeRtu.STATE_FAILONE || | 
 |  |  |                     rtu.state == UpgradeRtu.STATE_FAIL){ | 
 |  |  |                 //又失败了,认为结束了 | 
 |  |  |                 rtu.isOver = true ; | 
 |  |  |                 return ; | 
 |  |  |             } | 
 |  |  |         } | 
 |  |  |  | 
 |  |  |         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 ; | 
 |  |  | 
 |  |  |             } | 
 |  |  |         } | 
 |  |  |  | 
 |  |  |         if(n == 45){ | 
 |  |  |         if(n == 450 || n == 451 || n == 452 || n == 453 || n == 454 || n == 455){ | 
 |  |  |             if(rtu.currentPackage != 1){ | 
 |  |  |                 //升死 | 
 |  |  |                 rtu.state = UpgradeRtu.STATE_FAIL ; | 
 |  |  | 
 |  |  |             } | 
 |  |  |         } | 
 |  |  |  | 
 |  |  |         if(rtu.currentPackage == rtu.totalPackage){ | 
 |  |  |             //升级结束 | 
 |  |  |             rtu.state = UpgradeRtu.STATE_SUCCESS ; | 
 |  |  |             rtu.isOver = true ; | 
 |  |  |             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 ; | 
 |  |  |             } | 
 |  |  |         } | 
 |  |  |  | 
 |  |  |         if(rtu.state != UpgradeRtu.STATE_FAILONE && | 
 |  |  | 
 |  |  |             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 ; | 
 |  |  |     } | 
 |  |  | } |