package com.dy.pipIrrRemote.rtuUpgrage; 
 | 
  
 | 
import com.alibaba.excel.EasyExcel; 
 | 
import com.alibaba.excel.support.ExcelTypeEnum; 
 | 
import com.dy.common.aop.SsoAop; 
 | 
import com.dy.common.softUpgrade.state.UpgradeTaskVo; 
 | 
import com.dy.common.springUtil.SpringContextUtil; 
 | 
import com.dy.common.webUtil.BaseResponse; 
 | 
import com.dy.common.webUtil.BaseResponseUtils; 
 | 
import com.dy.common.webUtil.QueryResultVo; 
 | 
import com.dy.common.webUtil.ResultCodeMsg; 
 | 
import com.dy.pipIrrGlobal.pojoUg.UgRtuProgram; 
 | 
import com.dy.pipIrrGlobal.pojoUg.UgRtuTask; 
 | 
import com.dy.pipIrrGlobal.rtuMw.Web2RtuMw; 
 | 
import com.dy.pipIrrGlobal.voUg.VoUgRtuResult4Failure; 
 | 
import com.dy.pipIrrGlobal.voUg.VoUgRtuResult4Success; 
 | 
import com.dy.pipIrrGlobal.voUg.VoWatch; 
 | 
import io.swagger.v3.oas.annotations.Operation; 
 | 
import io.swagger.v3.oas.annotations.media.Content; 
 | 
import io.swagger.v3.oas.annotations.media.Schema; 
 | 
import io.swagger.v3.oas.annotations.responses.ApiResponse; 
 | 
import io.swagger.v3.oas.annotations.responses.ApiResponses; 
 | 
import io.swagger.v3.oas.annotations.tags.Tag; 
 | 
import jakarta.servlet.http.HttpServletResponse; 
 | 
import lombok.extern.slf4j.Slf4j; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
import org.springframework.core.env.Environment; 
 | 
import org.springframework.http.MediaType; 
 | 
import org.springframework.web.bind.annotation.GetMapping; 
 | 
import org.springframework.web.bind.annotation.RequestMapping; 
 | 
import org.springframework.web.bind.annotation.RequestMethod; 
 | 
import org.springframework.web.bind.annotation.RestController; 
 | 
import org.springframework.web.client.RestTemplate; 
 | 
  
 | 
import java.io.UnsupportedEncodingException; 
 | 
import java.net.URLEncoder; 
 | 
import java.util.Base64; 
 | 
import java.util.List; 
 | 
  
 | 
/** 
 | 
 * @Author: liurunyu 
 | 
 * @Date: 2024/11/12 8:33 
 | 
 * @Description 
 | 
 */ 
 | 
@Slf4j 
 | 
@Tag(name = "rtu远程升级任务", description = "rtu远程升级任务相关操作") 
 | 
@RestController 
 | 
@RequestMapping(path = "rtuUpgrade") 
 | 
public class RtuUpgradeCtrl extends Web2RtuMw { 
 | 
    @Autowired 
 | 
    private RtuUpgradeSv sv ; 
 | 
  
 | 
    @Autowired 
 | 
    private RtuUpgradeResSv resSv ; 
 | 
  
 | 
    @Autowired 
 | 
    private Environment env; 
 | 
  
 | 
    @Autowired 
 | 
    private RestTemplate restTemplate; 
 | 
  
 | 
    /** 
 | 
     * 下发rtu远程升级任务 
 | 
     * @param id 任务id 
 | 
     * @return 操作结果 
 | 
     */ 
 | 
    @Operation(summary = "下发rtu远程升级任务", description = "下发rtu远程升级任务") 
 | 
    @ApiResponses(value = { 
 | 
            @ApiResponse( 
 | 
                    responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, 
 | 
                    description = "返回操作成功与否数据(BaseResponse.content:Boolean)", 
 | 
                    content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, 
 | 
                            schema = @Schema(implementation = Boolean.class))} 
 | 
            ) 
 | 
    }) 
 | 
    @GetMapping(path = "/issuedTask") 
 | 
    @SsoAop() 
 | 
    public BaseResponse<Boolean> issuedTask(String id){ 
 | 
        if(id == null || id.trim().equals("")){ 
 | 
            return BaseResponseUtils.buildErrorMsg("任务id不能为空") ; 
 | 
        } 
 | 
        UgRtuTask tpo = this.sv.selectTaskById(id) ; 
 | 
        if(tpo == null){ 
 | 
            return BaseResponseUtils.buildErrorMsg("任务不存在") ; 
 | 
        } 
 | 
        if(tpo.isExecute == 1){ 
 | 
            return BaseResponseUtils.buildErrorMsg("任务已下发,不能重复下发任务") ; 
 | 
        } 
 | 
        UgRtuProgram ppo = this.sv.selectProgramById(tpo.programId) ; 
 | 
        if(ppo == null){ 
 | 
            return BaseResponseUtils.buildErrorMsg("任务对应的程序不存在") ; 
 | 
        } 
 | 
  
 | 
        List<String> taskRtuAddrs = this.sv.selectAllRtuAddrByTask(id) ; 
 | 
        if(taskRtuAddrs == null || taskRtuAddrs.size() == 0){ 
 | 
            return BaseResponseUtils.buildErrorMsg("任务所涉及的控制器还未设置") ; 
 | 
        } 
 | 
  
 | 
        String ugCallbackUrl_rm = env.getProperty("mw.ugCallbackUrl_rm" ); 
 | 
        if(ugCallbackUrl_rm == null || ugCallbackUrl_rm.trim().equals("")){ 
 | 
            return BaseResponseUtils.buildErrorMsg("未配置升级任务回调网址") ; 
 | 
        } 
 | 
  
 | 
        UpgradeTaskVo vo = new UpgradeTaskVo() ; 
 | 
        this.valueFromPo(vo, tpo, ppo) ; 
 | 
        vo.rtuAddrList = taskRtuAddrs ; 
 | 
        vo.callbackWebUrl = ugCallbackUrl_rm ; 
 | 
  
 | 
        String rqUrl = this.get2MwRequestUrl(this.env, ContextUgTaskSend) ; 
 | 
        BaseResponse res = sendPostRequest2Mw(restTemplate, rqUrl, vo) ; 
 | 
        if(res != null){ 
 | 
            if(res.isSuccess()){ 
 | 
                this.sv.setUpgradeTaskExecuted(id); 
 | 
                return BaseResponseUtils.buildSuccess(true) ; 
 | 
            }else{ 
 | 
                String msg = res.getContent()==null?null:(String)res.getContent() ; 
 | 
                if(msg == null){ 
 | 
                    msg = res.getMsg() ; 
 | 
                } 
 | 
                log.error("通信中间件执行下发升级任务失败," + msg) ; 
 | 
                return BaseResponseUtils.buildErrorMsg("通信中间件执行失败," + msg) ; 
 | 
            } 
 | 
        }else{ 
 | 
            log.error("通信中间件返回结果为null") ; 
 | 
            return BaseResponseUtils.buildErrorMsg("通信中间件返回结果为null") ; 
 | 
        } 
 | 
    } 
 | 
    /** 
 | 
     * 下发强制结束当前rtu远程升级任务 
 | 
     * @return 操作结果 
 | 
     */ 
 | 
    @Operation(summary = "下发强制结束当前rtu远程升级任务", description = "下发强制结束当前rtu远程升级任务") 
 | 
    @ApiResponses(value = { 
 | 
            @ApiResponse( 
 | 
                    responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, 
 | 
                    description = "返回操作成功与否数据(BaseResponse.content:Boolean)", 
 | 
                    content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, 
 | 
                            schema = @Schema(implementation = String.class))} 
 | 
            ) 
 | 
    }) 
 | 
    @GetMapping(path = "/forceOver") 
 | 
    @SsoAop() 
 | 
    public BaseResponse<String> forceOver() { 
 | 
        String rqUrl = this.get2MwRequestUrl(this.env, ContextUgForceOver) ; 
 | 
        BaseResponse res = sendGetRequest2Mw(restTemplate, rqUrl) ; 
 | 
        if(res != null){ 
 | 
            if(res.isSuccess()){ 
 | 
                RtuUpgradeStateReceiverCtrl.afterMwForceOverCurUgTask(); 
 | 
                String msg = res.getContent()==null?null:(String)res.getContent() ; 
 | 
                if(msg == null){ 
 | 
                    msg = res.getMsg() ; 
 | 
                } 
 | 
                return BaseResponseUtils.buildResult(ResultCodeMsg.RsCode.SUCCESS_CODE, msg, msg) ; 
 | 
            }else{ 
 | 
                String msg = res.getContent()==null?null:(String)res.getContent() ; 
 | 
                if(msg == null){ 
 | 
                    msg = res.getMsg() ; 
 | 
                } 
 | 
                log.error("通信中间件执行下发强制结束当前rtu远程升级任务失败:" + msg) ; 
 | 
                return BaseResponseUtils.buildErrorMsg("通信中间件执行强制结束当前rtu远程升级任务失败:" + msg) ; 
 | 
            } 
 | 
        }else{ 
 | 
            log.error("通信中间件返回结果为null") ; 
 | 
            return BaseResponseUtils.buildErrorMsg("通信中间件返回结果为null") ; 
 | 
        } 
 | 
    } 
 | 
  
 | 
  
 | 
    private void valueFromPo(UpgradeTaskVo vo, UgRtuTask tpo, UgRtuProgram ppo){ 
 | 
        vo.id = "" + tpo.id ; 
 | 
        vo.softFileName = ppo.hexFileName ; 
 | 
        vo.softStoreAddr = ppo.storeRamAddr ; 
 | 
        vo.softStartAddr = ppo.startRamAddr ; 
 | 
        vo.softFileData64 = Base64.getEncoder().encodeToString(ppo.programBytes) ; 
 | 
        //vo.softFileData = ppo.programBytes ; 
 | 
        vo.softBytesCalculate = ppo.programCalculateBytes ; 
 | 
        vo.softByteSrc16 = ppo.programCrc16 ; 
 | 
    } 
 | 
  
 | 
  
 | 
    /** 
 | 
     * 重置远程升级演示程序 
 | 
     * @return 操作结果 
 | 
     */ 
 | 
    @GetMapping(path = "/demoReset") 
 | 
    @SsoAop() 
 | 
    public BaseResponse<Boolean> demoReset(){ 
 | 
        if(RtuUpgradeStateReceiverCtrl.cache != null){ 
 | 
            //return BaseResponseUtils.buildError("当前没有升级任务") ; 
 | 
            //正式运行时,下面两行去掉,上面一行打开 
 | 
            RtuUpgradeStateReceiverCtrl ctrl = SpringContextUtil.getBean(RtuUpgradeStateReceiverCtrl.class); 
 | 
            ctrl.resetDemo(); 
 | 
            ctrl.demo(); 
 | 
        } 
 | 
        return BaseResponseUtils.buildSuccess(true) ; 
 | 
    } 
 | 
  
 | 
    /** 
 | 
     *  监视: 
 | 
     * 当前rtu远程升级任务执行状态 
 | 
     * @return 操作结果 
 | 
     */ 
 | 
    @Operation(summary = "当前rtu远程升级任务执行状态", description = "当前rtu远程升级任务执行状态") 
 | 
    @ApiResponses(value = { 
 | 
            @ApiResponse( 
 | 
                    responseCode = ResultCodeMsg.RsCode.SUCCESS_CODE, 
 | 
                    description = "返回操作成功与否数据(BaseResponse.content:Boolean)", 
 | 
                    content = {@Content(mediaType = MediaType.APPLICATION_JSON_VALUE, 
 | 
                            schema = @Schema(implementation = VoWatch.class))} 
 | 
            ) 
 | 
    }) 
 | 
    @GetMapping(path = "/curUpgradeState") 
 | 
    @SsoAop() 
 | 
    public BaseResponse<QueryResultVo<VoWatch> > curUpgradeState(QueryVo qvo){ 
 | 
        if(qvo == null){ 
 | 
            qvo = new QueryVo(); 
 | 
            qvo.pageCurr = 1 ; 
 | 
            qvo.pageSize = 49 ; 
 | 
        } 
 | 
        if(RtuUpgradeStateReceiverCtrl.cache == null){ 
 | 
            return BaseResponseUtils.buildErrorMsg("当前没有升级任务") ; 
 | 
            //正式运行时,下面两行去掉,上面一行打开 
 | 
            //RtuUpgradeStateReceiverCtrl ctrl = SpringContextUtil.getBean(RtuUpgradeStateReceiverCtrl.class); 
 | 
            //ctrl.demo(); 
 | 
        } 
 | 
        QueryResultVo<VoWatch> rsVo = this.resSv.curUpgradeState(qvo) ; 
 | 
        return BaseResponseUtils.buildSuccess(rsVo) ; 
 | 
    } 
 | 
  
 | 
    /** 
 | 
     * 导出升级失败RTU列表 
 | 
     * @param response 
 | 
     */ 
 | 
    @RequestMapping(value = "/exportUgFail", method = RequestMethod.GET) 
 | 
    public void exportUgFail(HttpServletResponse response) throws Exception { 
 | 
        setExcelRespProp(response, "升级失败RTU列表"); 
 | 
        List<VoUgRtuResult4Failure> rsList = resSv.exportUgFail(); 
 | 
        EasyExcel.write(response.getOutputStream()) 
 | 
                .head(VoUgRtuResult4Failure.class) 
 | 
                .excelType(ExcelTypeEnum.XLSX) 
 | 
                .sheet("升级失败RTU列表") 
 | 
                .doWrite(rsList); 
 | 
    } 
 | 
  
 | 
  
 | 
    /** 
 | 
     * 导出长级成功RTU列表 
 | 
     * @param response 
 | 
     */ 
 | 
    @RequestMapping(value = "/exportUgSuccess", method = RequestMethod.GET) 
 | 
    public void exportUgSuccess(HttpServletResponse response) throws Exception { 
 | 
        setExcelRespProp(response, "升级成功RTU列表"); 
 | 
        List<VoUgRtuResult4Success> rsList = resSv.exportUgSuccess(); 
 | 
        EasyExcel.write(response.getOutputStream()) 
 | 
                .head(VoUgRtuResult4Success.class) 
 | 
                .excelType(ExcelTypeEnum.XLSX) 
 | 
                .sheet("升级成功RTU列表") 
 | 
                .doWrite(rsList); 
 | 
    } 
 | 
  
 | 
  
 | 
    /** 
 | 
     * 设置excel下载响应头属性 
 | 
     */ 
 | 
    private void setExcelRespProp(HttpServletResponse response, String rawFileName) throws UnsupportedEncodingException { 
 | 
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"); 
 | 
        response.setCharacterEncoding("utf-8"); 
 | 
        String fileName = URLEncoder.encode(rawFileName, "UTF-8").replaceAll("\\+", "%20"); 
 | 
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx"); 
 | 
    } 
 | 
  
 | 
} 
 |