From 252e5230703e4355d25274d21b84f2d060907416 Mon Sep 17 00:00:00 2001
From: liusuyi <13324259@qq.com>
Date: 星期五, 14 七月 2023 21:49:39 +0800
Subject: [PATCH] 提交拉流取流查询接口
---
ard-work/src/main/java/com/ruoyi/media/service/impl/MediaService.java | 79 ++++++++++++++++++++++++++++++++++-----
1 files changed, 69 insertions(+), 10 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 b360b82..00e5413 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,15 +1,14 @@
package com.ruoyi.media.service.impl;
import com.alibaba.fastjson2.JSONObject;
-import com.ruoyi.media.domain.Conf;
-import com.ruoyi.media.domain.Items;
-import com.ruoyi.media.domain.JsonsRoot;
+import com.ruoyi.media.domain.*;
import com.ruoyi.media.service.IMediaService;
import com.ruoyi.utils.forest.MediaClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
+import java.util.ArrayList;
import java.util.List;
/**
@@ -41,26 +40,86 @@
// -i //鐢ㄤ簬鎸囧畾杈撳叆濯掍綋鏂囦欢鎴栬緭鍏ユ祦鐨勫湴鍧�
//-f rtsp //杩欎釜閫夐」鍛婅瘔 FFmpeg 杈撳嚭涓� RTSP 鏍煎紡銆�
//CPU杞В鐮佺紪鐮�
- //String cmd = "ffmpeg -rtsp_transport udp -i \"" + rtspPath + "\" -vcodec libx264 -preset:v veryfast -r 25 -threads 4 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH";
- //GPU纭В鐮佺紪鐮�
- String cmd = "ffmpeg -hwaccel cuvid -c:v h264_cuvid -rtsp_transport udp -i \"" + rtspPath + "\" -c:v h264_nvenc -r 25 -threads 4 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH";
+ 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瑙ｇ爜 -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 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH";
mediaInfo.setRunoninit(cmd);
mediaInfo.setRunoninitrestart(true);
- mediaClient.add(apiUrl, name, mediaInfo);
+ mediaClient.addPath(apiUrl, name, mediaInfo);
return rtspUrl + name;
}
@Override
public void removePath(String name) {
String apiUrl="http://"+mediamtxHost+":9997/v2";
- mediaClient.remove(apiUrl, name);
+ mediaClient.removePath(apiUrl, name);
}
@Override
- public List<Items> list() {
+ public List<Items> paths() {
String apiUrl="http://"+mediamtxHost+":9997/v2";
- String list = mediaClient.list(apiUrl);
+ String list = mediaClient.paths(apiUrl);
JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
return jsonsRoot.getItems();
}
+
+ @Override
+ public List<Items> rtspconns() {
+ String apiUrl="http://"+mediamtxHost+":9997/v2";
+ String list = mediaClient.rtspconns(apiUrl);
+ 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);
+ 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);
+ RtspSession rtspSession = JSONObject.parseObject(list, RtspSession.class);
+ return rtspSession;
+ }
+
+ @Override
+ public List<RtspSession> getPushStreams() {
+ List<RtspSession> rtspSessions=new ArrayList<>();
+ String apiUrl="http://"+mediamtxHost+":9997/v2";
+ String list = mediaClient.paths(apiUrl);
+ JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
+ List<Items> items = jsonsRoot.getItems();
+ for(Items item:items)
+ {
+ Source source = item.getSource();
+ RtspSession rtspSession = getRtspSessionById(source.getId());
+ rtspSession.setCameraId(item.getName());
+ rtspSessions.add(rtspSession);
+ }
+ return rtspSessions;
+ }
+
+ @Override
+ public List<RtspSession> getPullStreams() {
+ List<RtspSession> rtspSessions=new ArrayList<>();
+ String apiUrl="http://"+mediamtxHost+":9997/v2";
+ String list = mediaClient.paths(apiUrl);
+ JsonsRoot jsonsRoot = JSONObject.parseObject(list, JsonsRoot.class);
+ List<Items> items = jsonsRoot.getItems();
+ for(Items item:items)
+ {
+ List<Readers> readers = item.getReaders();
+ for(Readers reader:readers) {
+ RtspSession rtspSession = getRtspSessionById(reader.getId());
+ rtspSession.setCameraId(item.getName());
+ rtspSessions.add(rtspSession);
+ }
+ }
+ return rtspSessions;
+ }
}
--
Gitblit v1.9.3