From 33aed56cfbaad870dbf4780688ba6b08d0ee806c Mon Sep 17 00:00:00 2001 From: ‘liusuyi’ <1951119284@qq.com> Date: 星期四, 28 九月 2023 11:02:13 +0800 Subject: [PATCH] 修改雷达引导持续报警每5分钟再次引导一次 --- ard-work/src/main/java/com/ruoyi/media/service/impl/MediaServiceImpl.java | 190 +++++++++++++++++++++++++++-------------------- 1 files changed, 109 insertions(+), 81 deletions(-) diff --git a/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaServiceImpl.java b/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaServiceImpl.java index 4966e7e..af45f2b 100644 --- a/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaServiceImpl.java +++ b/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaServiceImpl.java @@ -1,6 +1,8 @@ package com.ruoyi.media.service.impl; import com.alibaba.fastjson2.JSONObject; +import com.ruoyi.common.utils.DateUtils; +import com.ruoyi.common.utils.SecurityUtils; import com.ruoyi.common.utils.StringUtils; import com.ruoyi.media.domain.*; import com.ruoyi.media.mapper.VtduMapper; @@ -20,10 +22,7 @@ import javax.annotation.PreDestroy; import javax.annotation.Resource; import java.io.File; -import java.util.ArrayList; -import java.util.Collections; -import java.util.Comparator; -import java.util.List; +import java.util.*; import java.util.regex.Matcher; import java.util.regex.Pattern; @@ -45,68 +44,37 @@ @Value("${mediamtx.host}") String mediamtxHost; - @Value("${mediamtx.enabled}") - Boolean mediamtxEnabled; @Value("${mediamtx.software_decoding}") Boolean softwareDecoding; - - String processName = "mediamtx.exe"; @Override public void run(ApplicationArguments args) { try { - log.info("寮�濮嬪姞杞芥祦濯掍綋鍒楄〃"); + log.debug("寮�濮嬪姞杞芥祦濯掍綋鍒楄〃"); List<StreamInfo> paths = paths(); for (StreamInfo path : paths) { mediaClient.removePath(path.getName()); } List<Vtdu> vtduList = vtduMapper.selectVtduList(new Vtdu()); - for (Vtdu v : vtduList) { - addPath(v.getName(), v.getRtspUrl(), v.getCodeType(), v.getIsCode()); + for (Vtdu vtdu : vtduList) { + Map<String, String> map = addPath(vtdu.getName(), vtdu.getSourceUrl(), vtdu.getCodeType(), vtdu.getIsCode()); + vtdu.setRtspUrl(map.get("rtspUrl")); + vtdu.setRtmpUrl(map.get("rtmpUrl")); + vtdu.setWebrtcUrl(map.get("webrtcUrl")); + vtdu.setUpdateTime(DateUtils.getNowDate()); + vtduMapper.updateVtdu(vtdu); } } catch (Exception ex) { log.error("鍔犺浇娴佸獟浣撳垪琛ㄥ紓甯�:" + ex.getMessage()); } } - @PostConstruct - public void initMediaMtx() { - if (mediamtxEnabled) { - log.info("鍒濆鍖栧惎鍔╩ediaMTX"); - if (Platform.isWindows()) { - String exePath = System.getProperty("user.dir") + File.separator + "lib" + File.separator + "mediamtx" + File.separator + "mediamtx.exe"; - String ymlPath = System.getProperty("user.dir") + File.separator + "lib" + File.separator + "mediamtx" + File.separator + "mediamtx.yml"; - - List<String> cmd = new ArrayList<>(); - cmd.add(exePath); - cmd.add(ymlPath); - if (CmdUtils.isProcessRunning(processName)) { - // 杩涚▼宸茬粡鍦ㄨ繍琛岋紝缁撴潫璇ヨ繘绋� - CmdUtils.stopProcess(processName); - } - // 鍚姩鍚庡彴杩涚▼ - CmdUtils.commandStart(processName, cmd, null); - // 鍚姩cmd绐楀彛 -// String[] command = {"cmd","/c","start",exePath,ymlPath}; -// CmdUtils.commandStart(command); - } - } - } - - @PreDestroy - public void destroyMediaMtx() { - if (mediamtxEnabled) { - log.info("閿�姣乵ediaMtx"); - if (CmdUtils.isProcessRunning(processName)) { - // 杩涚▼宸茬粡鍦ㄨ繍琛岋紝缁撴潫璇ヨ繘绋� - CmdUtils.stopProcess(processName); - } - } - } - @Override - public String addPath(String name, String rtspPath, String mode, String isCode) { - String rtspUrl = "rtsp://" + mediamtxHost + ":7554/" + name; + public Map<String,String> addPath(String name, String sourceUrl, String mode, String isCode) { + String rtspUrl = "rtsp://" + mediamtxHost + ":8554/" + name; + String rtmpUrl = "rtmp://" + mediamtxHost + ":1935/" + name; + String webrtcUrl = "http://" + mediamtxHost + ":8889/" + name; + Conf mediaInfo = new Conf(); String rootPath = System.getProperty("user.dir").replaceAll("\\\\", "/") + "/lib/mediamtx/"; //-vcodec libx264 //鎸囧畾瑙嗛缂栫爜鍣ㄤ负 libx264锛屼娇鐢� H.264 缂栫爜鏍煎紡杩涜瑙嗛鍘嬬缉 @@ -119,28 +87,97 @@ // -i //鐢ㄤ簬鎸囧畾杈撳叆濯掍綋鏂囦欢鎴栬緭鍏ユ祦鐨勫湴鍧� // -bf 0 绂佺敤B甯э紝鍥犱负webrtc鍦ㄧ綉椤佃皟鐢ㄦ椂鎺у埗鍙颁竴鐩磋緭鍑� WebRTC doesn鈥檛 support H264 streams with B-frames //-f rtsp //杩欎釜閫夐」鍛婅瘔 FFmpeg 杈撳嚭涓� RTSP 鏍煎紡銆� + //-acodec opus //闊抽杞爜opus + //-acodec pcm_alaw //闊抽杞爜PCMA(G711A) + //-strict 閫夐」鐢ㄤ簬璁剧疆闊抽缂栬В鐮佺殑涓ユ牸绾у埆锛岃�� -2 鏄叾鍙傛暟涔嬩竴銆傚湪鏃╂湡鐗堟湰鐨� FFmpeg 涓紝-strict 閫夐」鐢ㄤ簬璁剧疆闊抽缂栬В鐮佸櫒鐨勮涓猴紝鍏朵腑 -2 琛ㄧず浣跨敤闈炲父涓ユ牸鐨勬ā寮忋�� //CPU杞В鐮佺紪鐮� //String cmd = rootPath + "/lib/mediamtx/" +"ffmpeg -rtsp_transport tcp -i " + rtspPath + " -vcodec libx264 -preset:v ultrafast -r 25 -threads 4 -b:v 2048k -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; //GPU纭В鐮佺紪鐮� -hwaccel cuvid -c:v h264_cuvid 浣跨敤cuda瑙g爜 -c:v h264_nvenc 浣跨敤cuda缂栫爜 //String cmd = rootPath + "/lib/mediamtx/" + "ffmpeg -hwaccel cuvid -c:v h264_cuvid -rtsp_transport udp -i " + rtspPath + " -c:v h264_nvenc -r 25 -threads 4 -b:v 2048k -bf 0 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; if (isCode.equals("1")) { - String cmd = rootPath + "ffmpeg -rtsp_transport tcp -i " + rtspPath + " -vcodec libx264 -preset:v ultrafast -r 25 -keyint_min 20 -g 20 -sc_threshold 0 -threads 6 -b:v 2048k -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; + String cmd = "ffmpeg -rtsp_transport tcp -i " + sourceUrl + " -vcodec libx264 -preset:v ultrafast -r 25 -keyint_min 25 -g 25 -sc_threshold 0 -threads 6 -b:v 2048k -acodec opus -strict -2 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; if (!softwareDecoding) { - cmd = rootPath + "ffmpeg -hwaccel cuvid -c:v h264_cuvid -rtsp_transport tcp -i " + rtspPath + " -c:v h264_nvenc -r 25 -g 60 -threads 6 -b:v 2048k -bf 0 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; + cmd = "ffmpeg -hwaccel cuvid -c:v h264_cuvid -rtsp_transport tcp -i " + sourceUrl + " -c:v h264_nvenc -r 25 -g 60 -threads 6 -b:v 2048k -bf 0 -acodec opus -strict -2 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; } if (mode.equals("0")) { mediaInfo.setRunondemand(cmd); mediaInfo.setRunondemandrestart(true); + mediaInfo.setRunondemandcloseafter("5s"); } else { mediaInfo.setRunoninit(cmd); mediaInfo.setRunoninitrestart(true); } } else { - mediaInfo.setSource(rtspPath); + mediaInfo.setSource(sourceUrl); + mediaInfo.setSourceondemand(true); } + mediaInfo.setMaxReaders(100); mediaInfo.setSourceprotocol("tcp"); mediaClient.addPath(name, mediaInfo); - return rtspUrl; + Map<String,String> map=new HashMap<>(); + map.put("rtspUrl",rtspUrl); + map.put("rtmpUrl",rtmpUrl); + map.put("webrtcUrl",webrtcUrl); + return map; + } + + @Override + public Map<String, String> editPath(String name, String sourceUrl, String mode, String isCode) { + String rtspUrl = "rtsp://" + mediamtxHost + ":8554/" + name; + String rtmpUrl = "rtmp://" + mediamtxHost + ":1935/" + name; + String webrtcUrl = "http://" + mediamtxHost + ":8889/" + name; + + Conf mediaInfo = new Conf(); + String rootPath = System.getProperty("user.dir").replaceAll("\\\\", "/") + "/lib/mediamtx/"; + //-vcodec libx264 //鎸囧畾瑙嗛缂栫爜鍣ㄤ负 libx264锛屼娇鐢� H.264 缂栫爜鏍煎紡杩涜瑙嗛鍘嬬缉 + //-preset ultrafast //--preset鐨勫弬鏁颁富瑕佽皟鑺傜紪鐮侀�熷害鍜岃川閲忕殑骞宠 锛屾湁ultrafast锛堣浆鐮侀�熷害鏈�蹇紝瑙嗛寰�寰�涔熸渶妯$硦锛夈�乻uperfast銆乿eryfast銆乫aster銆乫ast銆乵edium銆乻low銆乻lower銆乿eryslow銆乸lacebo杩�10涓�夐」锛屼粠蹇埌鎱� + //-r 25 //璁剧疆杈撳嚭瑙嗛鐨勫抚鐜囦负 25 甯�/绉� + //-g 20 //鍏抽敭甯ч棿闅�20 + //-sc_threshold 0 //灏嗗叾璁剧疆涓�0锛�-sc_threshold 0锛夌鐢ㄥ満鏅彉鍖栨娴� + //-rtsp_transport tcp //杩欎釜閫夐」鍛婅瘔 FFmpeg 浣跨敤 TCP 浣滀负 RTSP 鐨勪紶杈撳崗璁� + //-threads 4: 鎸囧畾瑕佷娇鐢ㄧ殑绾跨▼鏁颁负 4銆�//杩欏厑璁� FFmpeg 鍦ㄥ鏍稿鐞嗗櫒涓婁娇鐢ㄥ涓嚎绋嬫潵杩涜瑙嗛缂栫爜锛屼互鍔犲揩閫熷害銆� + // -i //鐢ㄤ簬鎸囧畾杈撳叆濯掍綋鏂囦欢鎴栬緭鍏ユ祦鐨勫湴鍧� + // -bf 0 绂佺敤B甯э紝鍥犱负webrtc鍦ㄧ綉椤佃皟鐢ㄦ椂鎺у埗鍙颁竴鐩磋緭鍑� WebRTC doesn鈥檛 support H264 streams with B-frames + //-f rtsp //杩欎釜閫夐」鍛婅瘔 FFmpeg 杈撳嚭涓� RTSP 鏍煎紡銆� + //-acodec opus //闊抽杞爜opus + //CPU杞В鐮佺紪鐮� + //String cmd = rootPath + "/lib/mediamtx/" +"ffmpeg -rtsp_transport tcp -i " + rtspPath + " -vcodec libx264 -preset:v ultrafast -r 25 -threads 4 -b:v 2048k -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; + //GPU纭В鐮佺紪鐮� -hwaccel cuvid -c:v h264_cuvid 浣跨敤cuda瑙g爜 -c:v h264_nvenc 浣跨敤cuda缂栫爜 + //String cmd = rootPath + "/lib/mediamtx/" + "ffmpeg -hwaccel cuvid -c:v h264_cuvid -rtsp_transport udp -i " + rtspPath + " -c:v h264_nvenc -r 25 -threads 4 -b:v 2048k -bf 0 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; + if (isCode.equals("1")) { + mediaInfo.setSource("publisher"); + String cmd = "ffmpeg -rtsp_transport tcp -i " + sourceUrl + " -vcodec libx264 -preset:v ultrafast -r 25 -keyint_min 25 -g 25 -sc_threshold 0 -threads 6 -b:v 2048k -acodec opus -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; + if (!softwareDecoding) { + cmd = "ffmpeg -hwaccel cuvid -c:v h264_cuvid -rtsp_transport tcp -i " + sourceUrl + " -c:v h264_nvenc -r 25 -g 60 -threads 6 -b:v 2048k -bf 0 -acodec opus -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; + } + if (mode.equals("0")) { + mediaInfo.setRunondemand(cmd); + mediaInfo.setRunondemandcloseafter("5s"); + mediaInfo.setRunondemandrestart(true); + mediaInfo.setRunoninit(""); + mediaInfo.setRunoninitrestart(false); + } else { + mediaInfo.setRunoninit(cmd); + mediaInfo.setRunoninitrestart(true); + mediaInfo.setRunondemand(""); + mediaInfo.setRunondemandrestart(false); + } + } else { + mediaInfo.setSource(sourceUrl); + mediaInfo.setSourceondemand(true); + mediaInfo.setRunondemand(""); + mediaInfo.setRunondemandrestart(false); + mediaInfo.setRunoninit(""); + mediaInfo.setRunoninitrestart(false); + } + mediaInfo.setMaxReaders(100); + mediaInfo.setSourceprotocol("tcp"); + mediaClient.editPath(name, mediaInfo); + Map<String,String> map=new HashMap<>(); + map.put("rtspUrl",rtspUrl); + map.put("rtmpUrl",rtmpUrl); + map.put("webrtcUrl",webrtcUrl); + return map; } @Override @@ -149,19 +186,17 @@ StreamInfo info = new StreamInfo(); //ID info.setName(name); - String runoninit; - String runondemand = item.getConf().getRunondemand(); - if (StringUtils.isNotEmpty(runondemand)) { - runoninit = item.getConf().getRunondemand(); + String runOn; + if (StringUtils.isNotEmpty(item.getConf().getRunondemand())) { + runOn = item.getConf().getRunondemand(); info.setMode("0"); } else { - runoninit = item.getConf().getRunoninit(); + //runOn = item.getConf().getRunonready(); + runOn = item.getConf().getRunoninit(); info.setMode("1"); } //RTSP婧愬湴鍧� - String regex = "rtsp://[^\\s\"]+"; - Pattern pattern = Pattern.compile(regex); - Matcher matcher = pattern.matcher(runoninit); + Matcher matcher = Pattern.compile("rtsp://[^\\s\"]+").matcher(runOn); if (matcher.find()) { info.setRtspSource(matcher.group()); info.setIsCode("1"); @@ -190,19 +225,17 @@ //ID String name = item.getName(); info.setName(name); - String runoninit; - String runondemand = item.getConf().getRunondemand(); - if (StringUtils.isNotEmpty(runondemand)) { - runoninit = item.getConf().getRunondemand(); + String runOn; + if (StringUtils.isNotEmpty(item.getConf().getRunondemand())) { + runOn = item.getConf().getRunondemand(); info.setMode("0"); } else { - runoninit = item.getConf().getRunoninit(); + runOn = item.getConf().getRunoninit(); + //runOn = item.getConf().getRunonready(); info.setMode("1"); } //RTSP婧愬湴鍧� - String regex = "rtsp://[^\\s\"]+"; - Pattern pattern = Pattern.compile(regex); - Matcher matcher = pattern.matcher(runoninit); + Matcher matcher = Pattern.compile("rtsp://[^\\s\"]+").matcher(runOn); if (matcher.find()) { info.setRtspSource(matcher.group()); info.setIsCode("1"); @@ -211,9 +244,7 @@ info.setIsCode("0"); } //浼犺緭鍗忚 - regex = "-rtsp_transport\\s+(\\w+)"; - pattern = Pattern.compile(regex); - matcher = pattern.matcher(runoninit); + matcher = Pattern.compile("-rtsp_transport\\s+(\\w+)").matcher(runOn); if (matcher.find()) { info.setProtocol(matcher.group(1)); } @@ -265,7 +296,7 @@ String rtmpUrl = "rtmp://" + mediamtxHost + ":1935/" + name; info.setRtmpUrl(rtmpUrl); //RTSP鎾斁鍦板潃 - String rtspUrl = "rtsp://" + mediamtxHost + ":7554/" + name; + String rtspUrl = "rtsp://" + mediamtxHost + ":8554/" + name; info.setRtspUrl(rtspUrl); //WEBRTC鎾斁鍦板潃 String webrtcUrl = "http://" + mediamtxHost + ":8889/" + name; @@ -296,16 +327,14 @@ info.setRemoteAddr(rtspSession.getRemoteAddr()); } //RTSP婧愬湴鍧� - String runondemand = item.getConf().getRunondemand(); - String runoninit; - if (StringUtils.isNotEmpty(runondemand)) { - runoninit = item.getConf().getRunondemand(); + String runOn; + if (StringUtils.isNotEmpty(item.getConf().getRunondemand())) { + runOn = item.getConf().getRunondemand(); } else { - runoninit = item.getConf().getRunoninit(); + runOn = item.getConf().getRunoninit(); + //runOn = item.getConf().getRunonready(); } - String regex = "rtsp://[^\\s\"]+"; - Pattern pattern = Pattern.compile(regex); - Matcher matcher = pattern.matcher(runoninit); + Matcher matcher = Pattern.compile("rtsp://[^\\s\"]+").matcher(runOn); if (matcher.find()) { info.setRtspSource(matcher.group()); } else { @@ -316,7 +345,6 @@ //鎷夋祦鏁伴噺 List<Readers> readers = item.getReaders(); info.setNum(readers.size()); - PushStreamInfoList.add(info); } -- Gitblit v1.9.3