| | |
| | | |
| | | import com.alibaba.fastjson2.JSONObject; |
| | | import com.dtflys.forest.exceptions.ForestNetworkException; |
| | | import com.dtflys.forest.exceptions.ForestRuntimeException; |
| | | import com.ruoyi.common.utils.StringUtils; |
| | | import com.ruoyi.media.domain.*; |
| | | import com.ruoyi.media.service.IMediaService; |
| | |
| | | * @Version: 1.0 |
| | | **/ |
| | | @Service |
| | | @Slf4j(topic = "cmd") |
| | | @Slf4j(topic = "vtdu") |
| | | @Order(2) |
| | | public class MediaServiceImpl implements IMediaService { |
| | | |
| | |
| | | */ |
| | | @Override |
| | | 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 conf = new Conf(); |
| | | String rootPath = System.getProperty("user.dir").replaceAll("\\\\", "/") + "/lib/mediamtx/"; |
| | | if (isCode.equals("1")) { |
| | | conf.setSource("publisher"); |
| | | //默认软解码 |
| | | String cmd = "ffmpeg -rtsp_transport tcp -i " + sourceUrl + " -vcodec libx264 -preset:v ultrafast -r 25 -keyint_min 25 -g 60 -sc_threshold 0 -threads 6 -b:v 2048k -acodec opus -strict -2 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; |
| | | if (mode.equals("0")) {//硬解码 |
| | | cmd = "ffmpeg -hwaccel cuvid -c:v hevc_cuvid -rtsp_transport tcp -i " + sourceUrl + " -c:v h264_nvenc -r 25 -g 60 -sc_threshold 0 -threads 6 -b:v 2048k -bf 0 -acodec opus -strict -2 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; |
| | | } |
| | | conf.setRunOnDemand(cmd); |
| | | conf.setRunOnDemandRestart(true); |
| | | conf.setRunOnDemandCloseAfter("5s"); |
| | | } else { |
| | | conf.setSource(sourceUrl); |
| | | conf.setSourceOnDemand(true); |
| | | } |
| | | conf.setMaxReaders(100); |
| | | conf.setSourceProtocol("tcp"); |
| | | |
| | | if (!checkNameExist(name)) { |
| | | mediaClient.addPath(name, conf); |
| | | } |
| | | Map<String, String> map = new HashMap<>(); |
| | | map.put("rtspUrl", rtspUrl); |
| | | map.put("rtmpUrl", rtmpUrl); |
| | | map.put("webrtcUrl", webrtcUrl); |
| | | try { |
| | | String rtspUrl = "rtsp://" + mediamtxHost + ":8554/" + name; |
| | | String rtmpUrl = "rtmp://" + mediamtxHost + ":1935/" + name; |
| | | String webrtcUrl = "http://" + mediamtxHost + ":8889/" + name; |
| | | |
| | | Conf conf = new Conf(); |
| | | String rootPath = System.getProperty("user.dir").replaceAll("\\\\", "/") + "/lib/mediamtx/"; |
| | | if (isCode.equals("1")) { |
| | | conf.setSource("publisher"); |
| | | //默认软解码 |
| | | String cmd = "ffmpeg -rtsp_transport tcp -i " + sourceUrl + " -vcodec libx264 -preset:v ultrafast -r 25 -keyint_min 25 -g 60 -sc_threshold 0 -threads 6 -b:v 2048k -acodec opus -strict -2 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; |
| | | if (mode.equals("0")) {//硬解码 |
| | | cmd = "ffmpeg -hwaccel cuvid -c:v hevc_cuvid -rtsp_transport tcp -i " + sourceUrl + " -c:v h264_nvenc -r 25 -g 60 -sc_threshold 0 -threads 6 -b:v 2048k -bf 0 -acodec opus -strict -2 -f rtsp rtsp://localhost:$RTSP_PORT/$MTX_PATH"; |
| | | } |
| | | conf.setRunOnDemand(cmd); |
| | | conf.setRunOnDemandRestart(true); |
| | | conf.setRunOnDemandCloseAfter("5s"); |
| | | } else { |
| | | conf.setSource(sourceUrl); |
| | | conf.setSourceOnDemand(true); |
| | | } |
| | | conf.setMaxReaders(100); |
| | | conf.setSourceProtocol("tcp"); |
| | | |
| | | if (!checkNameExist(name)) { |
| | | mediaClient.addPath(name, conf); |
| | | } |
| | | |
| | | map.put("rtspUrl", rtspUrl); |
| | | map.put("rtmpUrl", rtmpUrl); |
| | | map.put("webrtcUrl", webrtcUrl); |
| | | } catch (ForestRuntimeException ex) { |
| | | log.error("添加流媒体异常:"+ex.getMessage()); |
| | | } |
| | | return map; |
| | | } |
| | | |
| | |
| | | map.put("rtspUrl", rtspUrl); |
| | | map.put("rtmpUrl", rtmpUrl); |
| | | map.put("webrtcUrl", webrtcUrl); |
| | | } catch (ForestNetworkException ex) { |
| | | log.error(ex.getMessage()); |
| | | } catch (ForestRuntimeException ex) { |
| | | log.error("修改流媒体异常:"+ex.getMessage()); |
| | | } |
| | | return map; |
| | | } |
| | |
| | | |
| | | @Override |
| | | public void removePath(String name) { |
| | | if (checkNameExist(name)) { |
| | | mediaClient.removePath(name); |
| | | try { |
| | | if (checkNameExist(name)) { |
| | | mediaClient.removePath(name); |
| | | } |
| | | } catch (ForestRuntimeException ex) { |
| | | log.error("移除流媒体异常:"+ex.getMessage()); |
| | | } |
| | | } |
| | | |
| | |
| | | @Override |
| | | public List<String> getNameList() { |
| | | List<String> nameList = new ArrayList<>(); |
| | | String paths = mediaClient.paths(); |
| | | JsonsRoot jsonsRoot = JSONObject.parseObject(paths, JsonsRoot.class); |
| | | List<Items> items = jsonsRoot.getItems(); |
| | | for (Items item : items) { |
| | | nameList.add(item.getName()); |
| | | try { |
| | | String paths = mediaClient.paths(); |
| | | JsonsRoot jsonsRoot = JSONObject.parseObject(paths, JsonsRoot.class); |
| | | List<Items> items = jsonsRoot.getItems(); |
| | | for (Items item : items) { |
| | | nameList.add(item.getName()); |
| | | } |
| | | } catch (ForestRuntimeException ex) { |
| | | log.error("获取流媒体name列表异常:"+ex.getMessage()); |
| | | } |
| | | return nameList; |
| | | } |