From e48fa711a3664bece9b3e58840a75fe7c05bc47c Mon Sep 17 00:00:00 2001 From: panlinlin <648540858@qq.com> Date: 星期六, 08 五月 2021 17:14:05 +0800 Subject: [PATCH] 添加截图(快照)功能 --- src/main/java/com/genersoft/iot/vmp/service/impl/StreamProxyServiceImpl.java | 61 ++++++++++++++++++++++++------ 1 files changed, 49 insertions(+), 12 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/service/impl/StreamProxyServiceImpl.java b/src/main/java/com/genersoft/iot/vmp/service/impl/StreamProxyServiceImpl.java index ddf8183..a77371f 100644 --- a/src/main/java/com/genersoft/iot/vmp/service/impl/StreamProxyServiceImpl.java +++ b/src/main/java/com/genersoft/iot/vmp/service/impl/StreamProxyServiceImpl.java @@ -1,9 +1,11 @@ package com.genersoft.iot.vmp.service.impl; import com.alibaba.fastjson.JSONObject; -import com.genersoft.iot.vmp.conf.MediaServerConfig; +import com.genersoft.iot.vmp.gb28181.bean.GbStream; +import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig; import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils; import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem; +import com.genersoft.iot.vmp.service.IGbStreamService; import com.genersoft.iot.vmp.storager.IRedisCatchStorage; import com.genersoft.iot.vmp.storager.IVideoManagerStorager; import com.genersoft.iot.vmp.storager.dao.GbStreamMapper; @@ -13,6 +15,9 @@ import com.github.pagehelper.PageInfo; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; + +import java.util.ArrayList; +import java.util.List; /** * 瑙嗛浠g悊涓氬姟 @@ -38,25 +43,57 @@ @Autowired private PlatformGbStreamMapper platformGbStreamMapper; + @Autowired + private IGbStreamService gbStreamService; + @Override - public void save(StreamProxyItem param) { - MediaServerConfig mediaInfo = redisCatchStorage.getMediaInfo(); + public String save(StreamProxyItem param) { + ZLMServerConfig mediaInfo = redisCatchStorage.getMediaInfo(); String dstUrl = String.format("rtmp://%s:%s/%s/%s", "127.0.0.1", mediaInfo.getRtmpPort(), param.getApp(), param.getStream() ); param.setDst_url(dstUrl); + StringBuffer result = new StringBuffer(); + boolean streamLive = false; // 鏇存柊 if (videoManagerStorager.queryStreamProxy(param.getApp(), param.getStream()) != null) { - boolean result = videoManagerStorager.updateStreamProxy(param); - if (result && param.isEnable()) { - addStreamProxyToZlm(param); + if (videoManagerStorager.updateStreamProxy(param)) { + result.append("淇濆瓨鎴愬姛"); + if (param.isEnable()){ + JSONObject jsonObject = addStreamProxyToZlm(param); + if (jsonObject == null) { + result.append(", 浣嗘槸鍚敤澶辫触锛岃妫�鏌ユ祦鍦板潃鏄惁鍙敤"); + param.setEnable(false); + videoManagerStorager.updateStreamProxy(param); + } + } } }else { // 鏂板 - boolean result = videoManagerStorager.addStreamProxy(param); - if (result && param.isEnable()) { - addStreamProxyToZlm(param); + if (videoManagerStorager.addStreamProxy(param)){ + result.append("淇濆瓨鎴愬姛"); + streamLive = true; + if (param.isEnable()) { + JSONObject jsonObject = addStreamProxyToZlm(param); + if (jsonObject == null) { + streamLive = false; + result.append(", 浣嗘槸鍚敤澶辫触锛岃妫�鏌ユ祦鍦板潃鏄惁鍙敤"); + param.setEnable(false); + videoManagerStorager.updateStreamProxy(param); + } + } + } + + } + if (param.getPlatformGbId() != null && streamLive) { + List<GbStream> gbStreams = new ArrayList<>(); + gbStreams.add(param); + if (gbStreamService.addPlatformInfo(gbStreams, param.getPlatformGbId())){ + result.append(", 鍏宠仈鍥芥爣骞冲彴[ " + param.getPlatformGbId() + " ]鎴愬姛"); + }else { + result.append(", 鍏宠仈鍥芥爣骞冲彴[ " + param.getPlatformGbId() + " ]澶辫触"); } } + return result.toString(); } @Override @@ -105,6 +142,7 @@ StreamProxyItem streamProxy = videoManagerStorager.queryStreamProxy(app, stream); if (!streamProxy.isEnable() && streamProxy != null) { JSONObject jsonObject = addStreamProxyToZlm(streamProxy); + if (jsonObject == null) return false; if (jsonObject.getInteger("code") == 0) { result = true; streamProxy.setEnable(true); @@ -118,12 +156,11 @@ public boolean stop(String app, String stream) { boolean result = false; StreamProxyItem streamProxyDto = videoManagerStorager.queryStreamProxy(app, stream); - if (streamProxyDto.isEnable() && streamProxyDto != null) { + if (streamProxyDto != null && streamProxyDto.isEnable()) { JSONObject jsonObject = removeStreamProxyFromZlm(streamProxyDto); if (jsonObject.getInteger("code") == 0) { - result = true; streamProxyDto.setEnable(false); - videoManagerStorager.updateStreamProxy(streamProxyDto); + result = videoManagerStorager.updateStreamProxy(streamProxyDto); } } return result; -- Gitblit v1.8.0