From 71dbac60afa54d2642c3471c080686da56579ca0 Mon Sep 17 00:00:00 2001
From: ‘liusuyi’ <1951119284@qq.com>
Date: 星期六, 09 九月 2023 16:37:13 +0800
Subject: [PATCH] 流媒体优化

---
 ard-work/src/main/java/com/ruoyi/media/controller/MediaController.java |   21 ++++++++++++++-------
 1 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/ard-work/src/main/java/com/ruoyi/media/controller/MediaController.java b/ard-work/src/main/java/com/ruoyi/media/controller/MediaController.java
index 411a27f..a4e65f3 100644
--- a/ard-work/src/main/java/com/ruoyi/media/controller/MediaController.java
+++ b/ard-work/src/main/java/com/ruoyi/media/controller/MediaController.java
@@ -18,6 +18,7 @@
 
 import javax.annotation.Resource;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 
 /**
@@ -51,14 +52,18 @@
         if (StringUtils.isEmpty(streamInfo.getRtspSource())) {
             return AjaxResult.error("rtsp鍦板潃涓嶈兘涓虹┖");
         }
-        String rtsp = mediaService.addPath(streamInfo.getName(), streamInfo.getRtspSource(), streamInfo.getMode(), streamInfo.getIsCode());
+        Map<String, String> map = mediaService.addPath(streamInfo.getName(), streamInfo.getRtspSource(), streamInfo.getMode(), streamInfo.getIsCode());
+        map.get("rtspUrl");
         Vtdu vtdu = new Vtdu();
         vtdu.setName(streamInfo.getName());
-        vtdu.setRtspUrl(streamInfo.getRtspSource());
+        vtdu.setSourceUrl(streamInfo.getRtspSource());
         vtdu.setIsCode(streamInfo.getIsCode());
         vtdu.setCodeType(streamInfo.getMode());
+        vtdu.setRtspUrl(map.get("rtspUrl"));
+        vtdu.setRtmpUrl(map.get("rtmpUrl"));
+        vtdu.setWebrtcUrl(map.get("webrtcUrl"));
         vtduService.insertVtdu(vtdu);
-        return AjaxResult.success(rtsp);
+        return AjaxResult.success(map);
     }
 
 
@@ -72,14 +77,16 @@
     public AjaxResult edit(@RequestBody StreamInfo streamInfo) {
         mediaService.removePath(new String[]{streamInfo.getName()});
         vtduService.deleteVtduByName(streamInfo.getName());
-        String rtsp = mediaService.addPath(streamInfo.getName(), streamInfo.getRtspSource(), streamInfo.getMode(), streamInfo.getIsCode());
+        Map<String, String> map = mediaService.addPath(streamInfo.getName(), streamInfo.getRtspSource(), streamInfo.getMode(), streamInfo.getIsCode());
         Vtdu vtdu = new Vtdu();
         vtdu.setName(streamInfo.getName());
-        vtdu.setRtspUrl(streamInfo.getRtspSource());
+        vtdu.setSourceUrl(streamInfo.getRtspSource());
         vtdu.setIsCode(streamInfo.getIsCode());
         vtdu.setCodeType(streamInfo.getMode());
-        vtduService.insertVtdu(vtdu);
-        return AjaxResult.success(rtsp);
+        vtdu.setRtspUrl(map.get("rtspUrl"));
+        vtdu.setRtmpUrl(map.get("rtmpUrl"));
+        vtdu.setWebrtcUrl(map.get("webrtcUrl"));
+        return AjaxResult.success(map);
     }
 
     /**

--
Gitblit v1.9.3