From dff994a28ff9be38450bc4e7ba540c2831d1206f Mon Sep 17 00:00:00 2001
From: zhangnaisong <2434969829@qq.com>
Date: 星期五, 21 七月 2023 09:53:55 +0800
Subject: [PATCH] 上传车辆图片id设为非必要提交

---
 ard-work/src/main/java/com/ruoyi/media/service/impl/MediaService.java |  140 +++++++++++++++++++++++++++++++++++-----------
 1 files changed, 106 insertions(+), 34 deletions(-)

diff --git a/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaService.java b/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaService.java
index bf043d6..baa5569 100644
--- a/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaService.java
+++ b/ard-work/src/main/java/com/ruoyi/media/service/impl/MediaService.java
@@ -1,10 +1,12 @@
 package com.ruoyi.media.service.impl;
 
 import com.alibaba.fastjson2.JSONObject;
+import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.media.domain.*;
 import com.ruoyi.media.service.IMediaService;
 import com.ruoyi.utils.forest.MediaClient;
 import com.ruoyi.utils.tools.ArdTool;
+import io.minio.messages.Item;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
@@ -26,13 +28,10 @@
 
     @Resource
     MediaClient mediaClient;
-
     @Value("${mediamtx.host}")
     String mediamtxHost;
-
     @Override
-    public String addPath(String name, String rtspPath) {
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
+    public String addPath(String name, String rtspPath,String mode) {
         String rtspUrl = "rtsp://" + mediamtxHost + ":8554/";
         Conf mediaInfo = new Conf();
         //-vcodec libx264 //鎸囧畾瑙嗛缂栫爜鍣ㄤ负 libx264锛屼娇鐢� H.264 缂栫爜鏍煎紡杩涜瑙嗛鍘嬬缉
@@ -46,48 +45,117 @@
         String cmd = "ffmpeg -rtsp_transport tcp -i \"" + rtspPath + "\" -vcodec libx264 -preset:v ultrafast -r 25 -threads 4  -b:v 4096k -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH";
         //GPU纭В鐮佺紪鐮� -hwaccel cuvid -c:v h264_cuvid  浣跨敤cuda瑙g爜   -c:v h264_nvenc 浣跨敤cuda缂栫爜
         //String cmd = "ffmpeg -hwaccel cuvid -c:v h264_cuvid  -rtsp_transport udp  -i \"" + rtspPath + "\" -c:v h264_nvenc  -r 25 -threads 4  -b:v 4096k -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH";
-//        mediaInfo.setRunoninit(cmd);
-//        mediaInfo.setRunoninitrestart(true);
-        mediaInfo.setRunondemand(cmd);
-        mediaInfo.setRunondemandrestart(true);
-        mediaClient.addPath(apiUrl, name, mediaInfo);
+
+        if(mode.equals("1"))
+        {
+            mediaInfo.setRunondemand(cmd);
+            mediaInfo.setRunondemandrestart(true);
+        }
+       else{
+           mediaInfo.setRunoninit(cmd);
+           mediaInfo.setRunoninitrestart(true);
+        }
+        mediaClient.addPath(name, mediaInfo);
         return rtspUrl + name;
     }
 
     @Override
-    public void removePath(String name) {
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        mediaClient.removePath(apiUrl, name);
+    public StreamInfo getPathInfo(String name) {
+        Items item = mediaClient.getPathInfo(name);
+        StreamInfo info = new StreamInfo();
+        //ID
+        info.setName(name);
+        String runoninit;
+        String runondemand = item.getConf().getRunondemand();
+        if (StringUtils.isNotEmpty(runondemand))
+        {
+            runoninit= item.getConf().getRunondemand();
+            info.setMode("1");
+        }
+        else
+        {
+            runoninit = item.getConf().getRunoninit();
+            info.setMode("2");
+        }
+        //RTSP婧愬湴鍧�
+        String regex = "rtsp://[^\\s\"]+";
+        Pattern pattern = Pattern.compile(regex);
+        Matcher matcher = pattern.matcher(runoninit);
+        if (matcher.find()) {
+            info.setRtspSource(matcher.group());
+        }
+        return info;
     }
 
     @Override
-    public List<Items> paths() {
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.paths(apiUrl);
+    public void removePath(String[] names) {
+        for(String name:names)
+        {
+            mediaClient.removePath(name);
+        }
+    }
+
+    @Override
+    public List<StreamInfo> paths() {
+
+        String list = mediaClient.paths();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
-        return jsonsRoot.getItems();
+        List<Items> items = jsonsRoot.getItems();
+        List<StreamInfo> pathInfoList = new ArrayList<>();
+        for (Items item : items) {
+            StreamInfo info = new StreamInfo();
+            //ID
+            String name = item.getName();
+            info.setName(name);
+            String runoninit;
+            String runondemand = item.getConf().getRunondemand();
+            if (StringUtils.isNotEmpty(runondemand))
+            {
+                runoninit= item.getConf().getRunondemand();
+                info.setMode("1");
+            }
+            else
+            {
+                runoninit = item.getConf().getRunoninit();
+                info.setMode("2");
+            }
+            //RTSP婧愬湴鍧�
+            String regex = "rtsp://[^\\s\"]+";
+            Pattern pattern = Pattern.compile(regex);
+            Matcher matcher = pattern.matcher(runoninit);
+            if (matcher.find()) {
+                info.setRtspSource(matcher.group());
+            }
+            //浼犺緭鍗忚
+            regex = "-rtsp_transport\\s+(\\w+)";
+            pattern = Pattern.compile(regex);
+            matcher = pattern.matcher(runoninit);
+            if (matcher.find()) {
+                info.setProtocol(matcher.group(1));
+            }
+
+            pathInfoList.add(info);
+        }
+        return pathInfoList;
     }
 
     @Override
     public List<Items> rtspconns() {
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.rtspconns(apiUrl);
+        String list = mediaClient.rtspconns();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
         return jsonsRoot.getItems();
     }
 
     @Override
     public List<Items> rtspsessions() {
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.rtspsessions(apiUrl);
+        String list = mediaClient.rtspsessions();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
         return jsonsRoot.getItems();
     }
 
     @Override
     public RtspSession getRtspSessionById(String sessionId) {
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.getRtspsessionById(apiUrl, sessionId);
+        String list = mediaClient.getRtspsessionById(sessionId);
         RtspSession rtspSession = JSONObject.parseObject(list, RtspSession.class);
         return rtspSession;
     }
@@ -95,8 +163,8 @@
     @Override
     public List<RtspSession> getPushStreams() {
         List<RtspSession> rtspSessions = new ArrayList<>();
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.paths(apiUrl);
+
+        String list = mediaClient.paths();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
         List<Items> items = jsonsRoot.getItems();
         for (Items item : items) {
@@ -111,8 +179,7 @@
     @Override
     public List<RtspSession> getPullStreams() {
         List<RtspSession> rtspSessions = new ArrayList<>();
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.paths(apiUrl);
+        String list = mediaClient.paths();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
         List<Items> items = jsonsRoot.getItems();
         for (Items item : items) {
@@ -129,8 +196,7 @@
     @Override
     public List<StreamInfo> getPushStreamList() {
         List<StreamInfo> PushStreamInfoList = new ArrayList<>();
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.paths(apiUrl);
+        String list = mediaClient.paths();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
         List<Items> items = jsonsRoot.getItems();
         for (Items item : items) {
@@ -160,8 +226,16 @@
             String formatSentSize = ArdTool.formatFileSize(bytesSent);
             info.setDownTraffic(formatSentSize);
             //RTSP婧愬湴鍧�
-            //String runoninit = item.getConf().getRunoninit();
-            String runoninit = item.getConf().getRunondemand();
+            String runondemand = item.getConf().getRunondemand();
+            String runoninit;
+            if(StringUtils.isNotEmpty(runondemand))
+            {
+                 runoninit = item.getConf().getRunondemand();
+            }
+            else
+            {
+                 runoninit = item.getConf().getRunoninit();
+            }
             String regex = "rtsp://[^\\s\"]+";
             Pattern pattern = Pattern.compile(regex);
             Matcher matcher = pattern.matcher(runoninit);
@@ -189,8 +263,7 @@
     @Override
     public List<StreamInfo> getPullStreamList() {
         List<StreamInfo> PullStreamInfoList = new ArrayList<>();
-        String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-        String list = mediaClient.paths(apiUrl);
+        String list = mediaClient.paths();
         JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
         List<Items> items = jsonsRoot.getItems();
         for (Items item : items) {
@@ -236,8 +309,7 @@
     @Override
     public Boolean kickRtspSession(String sessionId) {
         try {
-            String apiUrl = "http://" + mediamtxHost + ":9997/v2";
-            mediaClient.kick(apiUrl, sessionId);
+            mediaClient.kick(sessionId);
             return true;
         } catch (Exception ex) {
             return false;

--
Gitblit v1.9.3