From e29d94c83f62960bb1d6dacac4c978debc85c5ef Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期五, 22 七月 2022 16:02:14 +0800
Subject: [PATCH] 增加推流添加功能，修复级联目录推送
---
src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java | 185 ++++++++++++++++++++++++++++++++++++++-------
1 files changed, 155 insertions(+), 30 deletions(-)
diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java b/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java
index c0fafd3..1f0884f 100644
--- a/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java
+++ b/src/main/java/com/genersoft/iot/vmp/vmanager/streamPush/StreamPushController.java
@@ -3,13 +3,19 @@
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelReader;
import com.alibaba.excel.read.metadata.ReadSheet;
+import com.genersoft.iot.vmp.common.StreamInfo;
+import com.genersoft.iot.vmp.conf.security.SecurityUtils;
+import com.genersoft.iot.vmp.conf.security.dto.LoginUser;
import com.genersoft.iot.vmp.gb28181.bean.GbStream;
import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
+import com.genersoft.iot.vmp.media.zlm.dto.StreamAuthorityInfo;
import com.genersoft.iot.vmp.media.zlm.dto.StreamPushItem;
import com.genersoft.iot.vmp.service.IMediaServerService;
+import com.genersoft.iot.vmp.service.IMediaService;
import com.genersoft.iot.vmp.service.IStreamPushService;
import com.genersoft.iot.vmp.service.impl.StreamPushUploadFileHandler;
+import com.genersoft.iot.vmp.vmanager.bean.BatchGBStreamParam;
import com.genersoft.iot.vmp.vmanager.bean.StreamPushExcelDto;
import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
import com.github.pagehelper.PageInfo;
@@ -24,10 +30,12 @@
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
+import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.request.async.DeferredResult;
import org.springframework.web.multipart.MultipartFile;
+import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
@@ -52,21 +60,32 @@
@Autowired
private DeferredResultHolder resultHolder;
+ @Autowired
+ private IMediaService mediaService;
+
@ApiOperation("鎺ㄦ祦鍒楄〃鏌ヨ")
@ApiImplicitParams({
@ApiImplicitParam(name="page", value = "褰撳墠椤�", required = true, dataTypeClass = Integer.class),
@ApiImplicitParam(name="count", value = "姣忛〉鏌ヨ鏁伴噺", required = true, dataTypeClass = Integer.class),
@ApiImplicitParam(name="query", value = "鏌ヨ鍐呭", dataTypeClass = String.class),
- @ApiImplicitParam(name="online", value = "鏄惁鍦ㄧ嚎", dataTypeClass = Boolean.class),
+ @ApiImplicitParam(name="pushing", value = "鏄惁姝ｅ湪鎺ㄦ祦", dataTypeClass = Boolean.class),
+ @ApiImplicitParam(name="mediaServerId", value = "娴佸獟浣揑D", dataTypeClass = String.class),
})
@GetMapping(value = "/list")
@ResponseBody
public PageInfo<StreamPushItem> list(@RequestParam(required = false)Integer page,
@RequestParam(required = false)Integer count,
@RequestParam(required = false)String query,
- @RequestParam(required = false)Boolean online ){
+ @RequestParam(required = false)Boolean pushing,
+ @RequestParam(required = false)String mediaServerId ){
- PageInfo<StreamPushItem> pushList = streamPushService.getPushList(page, count);
+ if (StringUtils.isEmpty(query)) {
+ query = null;
+ }
+ if (StringUtils.isEmpty(mediaServerId)) {
+ mediaServerId = null;
+ }
+ PageInfo<StreamPushItem> pushList = streamPushService.getPushList(page, count, query, pushing, mediaServerId);
return pushList;
}
@@ -107,29 +126,54 @@
})
@PostMapping(value = "/stop")
@ResponseBody
- public Object removeFormGB(@RequestParam(required = true)String app, @RequestParam(required = true)String streamId){
+ public Object stop(String app, String streamId){
if (streamPushService.stop(app, streamId)){
return "success";
}else {
return "fail";
}
}
+
+ @ApiOperation("涓澶氫釜鎺ㄦ祦")
+ @ApiImplicitParams({
+ @ApiImplicitParam(name = "app", value = "搴旂敤鍚�", required = true, dataTypeClass = String.class),
+ @ApiImplicitParam(name = "streamId", value = "娴両D", required = true, dataTypeClass = String.class),
+ })
+ @DeleteMapping(value = "/batchStop")
+ @ResponseBody
+ public Object batchStop(@RequestBody BatchGBStreamParam batchGBStreamParam){
+ if (batchGBStreamParam.getGbStreams().size() == 0) {
+ return "fail";
+ }
+ if (streamPushService.batchStop(batchGBStreamParam.getGbStreams())){
+ return "success";
+ }else {
+ return "fail";
+ }
+ }
+
@PostMapping(value = "upload")
@ResponseBody
public DeferredResult<ResponseEntity<WVPResult<Object>>> uploadChannelFile(@RequestParam(value = "file") MultipartFile file){
-
// 鏈�澶氬鐞嗘枃浠朵竴涓皬鏃�
DeferredResult<ResponseEntity<WVPResult<Object>>> result = new DeferredResult<>(60*60*1000L);
// 褰曞儚鏌ヨ浠hannelId浣滀负deviceId鏌ヨ
String key = DeferredResultHolder.UPLOAD_FILE_CHANNEL;
String uuid = UUID.randomUUID().toString();
-
+ logger.info("閫氶亾瀵煎叆鏂囦欢绫诲瀷: {}",file.getContentType() );
if (file.isEmpty()) {
logger.warn("閫氶亾瀵煎叆鏂囦欢涓虹┖");
WVPResult<Object> wvpResult = new WVPResult<>();
wvpResult.setCode(-1);
wvpResult.setMsg("鏂囦欢涓虹┖");
+ result.setResult(ResponseEntity.status(HttpStatus.BAD_REQUEST).body(wvpResult));
+ return result;
+ }
+ if (file.getContentType() == null) {
+ WVPResult<Object> wvpResult = new WVPResult<>();
+ wvpResult.setCode(-1);
+ wvpResult.setMsg("鏃犳硶璇嗗埆鏂囦欢绫诲瀷");
result.setResult(ResponseEntity.status(HttpStatus.BAD_REQUEST).body(wvpResult));
return result;
}
@@ -162,32 +206,113 @@
} catch (IOException e) {
e.printStackTrace();
}
- //浼犲叆鍙傛暟
- ExcelReader excelReader = EasyExcel.read(inputStream, StreamPushExcelDto.class,
- new StreamPushUploadFileHandler(streamPushService, mediaServerService.getDefaultMediaServer().getId(), (errorStreams, errorGBs)->{
- logger.info("閫氶亾瀵煎叆鎴愬姛锛屽瓨鍦ㄩ噸澶岮pp+Stream涓簕}涓紝瀛樺湪鍥芥爣ID涓簕}涓�", errorStreams.size(), errorGBs.size());
- RequestMessage msg = new RequestMessage();
- msg.setKey(key);
- WVPResult<Map<String, List<String>>> wvpResult = new WVPResult<>();
- if (errorStreams.size() == 0 && errorGBs.size() == 0) {
- wvpResult.setCode(0);
- wvpResult.setMsg("鎴愬姛");
- }else {
- wvpResult.setCode(1);
- wvpResult.setMsg("瀵煎叆鎴愬姛銆備絾鏄瓨鍦ㄩ噸澶嶆暟鎹�");
- Map<String, List<String>> errorData = new HashMap<>();
- errorData.put("gbId", errorGBs);
- errorData.put("stream", errorStreams);
- wvpResult.setData(errorData);
- }
- msg.setData(wvpResult);
- resultHolder.invokeAllResult(msg);
- })).build();
- ReadSheet readSheet = EasyExcel.readSheet(0).build();
- excelReader.read(readSheet);
- excelReader.finish();
+ try {
+ //浼犲叆鍙傛暟
+ ExcelReader excelReader = EasyExcel.read(inputStream, StreamPushExcelDto.class,
+ new StreamPushUploadFileHandler(streamPushService, mediaServerService.getDefaultMediaServer().getId(), (errorStreams, errorGBs)->{
+ logger.info("閫氶亾瀵煎叆鎴愬姛锛屽瓨鍦ㄩ噸澶岮pp+Stream涓簕}涓紝瀛樺湪鍥芥爣ID涓簕}涓�", errorStreams.size(), errorGBs.size());
+ RequestMessage msg = new RequestMessage();
+ msg.setKey(key);
+ WVPResult<Map<String, List<String>>> wvpResult = new WVPResult<>();
+ if (errorStreams.size() == 0 && errorGBs.size() == 0) {
+ wvpResult.setCode(0);
+ wvpResult.setMsg("鎴愬姛");
+ }else {
+ wvpResult.setCode(1);
+ wvpResult.setMsg("瀵煎叆鎴愬姛銆備絾鏄瓨鍦ㄩ噸澶嶆暟鎹�");
+ Map<String, List<String>> errorData = new HashMap<>();
+ errorData.put("gbId", errorGBs);
+ errorData.put("stream", errorStreams);
+ wvpResult.setData(errorData);
+ }
+ msg.setData(wvpResult);
+ resultHolder.invokeAllResult(msg);
+ })).build();
+ ReadSheet readSheet = EasyExcel.readSheet(0).build();
+ excelReader.read(readSheet);
+ excelReader.finish();
+ }catch (Exception e) {
+ logger.warn("閫氶亾瀵煎叆澶辫触锛�", e);
+ RequestMessage msg = new RequestMessage();
+ msg.setKey(key);
+ WVPResult<Object> wvpResult = new WVPResult<>();
+ wvpResult.setCode(-1);
+ wvpResult.setMsg("閫氶亾瀵煎叆澶辫触: " + e.getMessage() );
+ msg.setData(wvpResult);
+ resultHolder.invokeAllResult(msg);
+ }
+
+
return result;
}
+ /**
+ * 鑾峰彇鎺ㄦ祦鎾斁鍦板潃
+ * @param app 搴旂敤鍚�
+ * @param stream 娴乮d
+ * @return
+ */
+ @ApiOperation("鑾峰彇鎺ㄦ祦鎾斁鍦板潃")
+ @ApiImplicitParams({
+ @ApiImplicitParam(name = "app", value = "搴旂敤鍚�", dataTypeClass = String.class),
+ @ApiImplicitParam(name = "stream", value = "娴乮d", dataTypeClass = String.class),
+ @ApiImplicitParam(name = "mediaServerId", value = "濯掍綋鏈嶅姟鍣╥d", dataTypeClass = String.class, required = false),
+ })
+ @GetMapping(value = "/getPlayUrl")
+ @ResponseBody
+ public WVPResult<StreamInfo> getPlayUrl(HttpServletRequest request, @RequestParam String app,
+ @RequestParam String stream,
+ @RequestParam(required = false) String mediaServerId){
+ boolean authority = false;
+ // 鏄惁鐧婚檰鐢ㄦ埛, 鐧婚檰鐢ㄦ埛杩斿洖瀹屾暣淇℃伅
+ LoginUser userInfo = SecurityUtils.getUserInfo();
+ if (userInfo!= null) {
+ authority = true;
+ }
+ StreamInfo streamInfo = mediaService.getStreamInfoByAppAndStreamWithCheck(app, stream, mediaServerId, authority);
+
+ WVPResult<StreamInfo> result = new WVPResult<>();
+ if (streamInfo != null){
+ result.setCode(0);
+ result.setMsg("scccess");
+ result.setData(streamInfo);
+ }else {
+ result.setCode(-1);
+ result.setMsg("fail");
+ }
+ return result;
+ }
+
+ /**
+ * 鑾峰彇鎺ㄦ祦鎾斁鍦板潃
+ * @param stream 鎺ㄦ祦淇℃伅
+ * @return
+ */
+ @ApiOperation("鑾峰彇鎺ㄦ祦鎾斁鍦板潃")
+ @ApiImplicitParams({
+ @ApiImplicitParam(name = "stream", value = "鎺ㄦ祦淇℃伅", dataTypeClass = StreamPushItem.class),
+ })
+ @PostMapping(value = "/add")
+ @ResponseBody
+ public WVPResult<StreamInfo> add(@RequestBody StreamPushItem stream){
+ if (StringUtils.isEmpty(stream.getGbId())) {
+
+ return new WVPResult<>(400, "鍥芥爣ID涓嶅彲涓虹┖", null);
+ }
+ if (StringUtils.isEmpty(stream.getApp()) && StringUtils.isEmpty(stream.getStream())) {
+ return new WVPResult<>(400, "app鎴杝tream涓嶅彲涓虹┖", null);
+ }
+ stream.setStatus(false);
+ stream.setPushIng(false);
+ stream.setAliveSecond(0L);
+ stream.setTotalReaderCount("0");
+ boolean result = streamPushService.add(stream);
+
+ if (result) {
+ return new WVPResult<>(0, "success", null);
+ }else {
+ return new WVPResult<>(-1, "fail", null);
+ }
+ }
}
--
Gitblit v1.8.0