From 985082d33930868c3cc1723f28fd9aaae9013a8f Mon Sep 17 00:00:00 2001 From: 648540858 <648540858@qq.com> Date: 星期三, 16 八月 2023 09:11:22 +0800 Subject: [PATCH] 添加缺少的内容 --- src/main/java/com/genersoft/iot/vmp/vmanager/streamProxy/StreamProxyController.java | 96 +++++++++++++++++++++++++++++++----------------- 1 files changed, 62 insertions(+), 34 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/streamProxy/StreamProxyController.java b/src/main/java/com/genersoft/iot/vmp/vmanager/streamProxy/StreamProxyController.java old mode 100644 new mode 100755 index 1ec4318..0689f42 --- a/src/main/java/com/genersoft/iot/vmp/vmanager/streamProxy/StreamProxyController.java +++ b/src/main/java/com/genersoft/iot/vmp/vmanager/streamProxy/StreamProxyController.java @@ -1,25 +1,31 @@ package com.genersoft.iot.vmp.vmanager.streamProxy; -import com.alibaba.fastjson.JSONObject; +import com.alibaba.fastjson2.JSONObject; import com.genersoft.iot.vmp.common.StreamInfo; +import com.genersoft.iot.vmp.conf.UserSetting; +import com.genersoft.iot.vmp.conf.exception.ControllerException; +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.MediaServerItem; import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem; import com.genersoft.iot.vmp.service.IMediaServerService; -import com.genersoft.iot.vmp.service.IMediaService; -import com.genersoft.iot.vmp.storager.IRedisCatchStorage; import com.genersoft.iot.vmp.service.IStreamProxyService; +import com.genersoft.iot.vmp.vmanager.bean.ErrorCode; +import com.genersoft.iot.vmp.vmanager.bean.StreamContent; import com.genersoft.iot.vmp.vmanager.bean.WVPResult; import com.github.pagehelper.PageInfo; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.Parameters; import io.swagger.v3.oas.annotations.tags.Tag; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; -import org.springframework.util.StringUtils; +import org.springframework.util.ObjectUtils; import org.springframework.web.bind.annotation.*; +import org.springframework.web.context.request.async.DeferredResult; + +import java.util.UUID; @SuppressWarnings("rawtypes") /** @@ -27,21 +33,23 @@ */ @Tag(name = "鎷夋祦浠g悊", description = "") @Controller -@CrossOrigin + @RequestMapping(value = "/api/proxy") public class StreamProxyController { private final static Logger logger = LoggerFactory.getLogger(StreamProxyController.class); @Autowired - private IRedisCatchStorage redisCatchStorage; - - - @Autowired private IMediaServerService mediaServerService; @Autowired private IStreamProxyService streamProxyService; + + @Autowired + private DeferredResultHolder resultHolder; + + @Autowired + private UserSetting userSetting; @Operation(summary = "鍒嗛〉鏌ヨ娴佷唬鐞�") @@ -64,18 +72,43 @@ }) @PostMapping(value = "/save") @ResponseBody - public WVPResult save(@RequestBody StreamProxyItem param){ + public DeferredResult<Object> save(@RequestBody StreamProxyItem param){ logger.info("娣诲姞浠g悊锛� " + JSONObject.toJSONString(param)); - if (StringUtils.isEmpty(param.getMediaServerId())) { + if (ObjectUtils.isEmpty(param.getMediaServerId())) { param.setMediaServerId("auto"); } - if (StringUtils.isEmpty(param.getType())) { + if (ObjectUtils.isEmpty(param.getType())) { param.setType("default"); } - if (StringUtils.isEmpty(param.getGbId())) { + if (ObjectUtils.isEmpty(param.getGbId())) { param.setGbId(null); } - WVPResult<StreamInfo> result = streamProxyService.save(param); + + RequestMessage requestMessage = new RequestMessage(); + String key = DeferredResultHolder.CALLBACK_CMD_PROXY + param.getApp() + param.getStream(); + requestMessage.setKey(key); + String uuid = UUID.randomUUID().toString(); + requestMessage.setId(uuid); + DeferredResult<Object> result = new DeferredResult<>(userSetting.getPlayTimeout().longValue()); + // 褰曞儚鏌ヨ浠hannelId浣滀负deviceId鏌ヨ + resultHolder.put(key, uuid, result); + result.onTimeout(()->{ + WVPResult<StreamInfo> wvpResult = new WVPResult<>(); + wvpResult.setCode(ErrorCode.ERROR100.getCode()); + wvpResult.setMsg("瓒呮椂"); + requestMessage.setData(wvpResult); + resultHolder.invokeAllResult(requestMessage); + }); + + streamProxyService.save(param, (code, msg, streamInfo) -> { + logger.info("[鎷夋祦浠g悊] {}", code == ErrorCode.SUCCESS.getCode()? "鎴愬姛":"澶辫触锛� " + msg); + if (code == ErrorCode.SUCCESS.getCode()) { + requestMessage.setData(new StreamContent(streamInfo)); + }else { + requestMessage.setData(WVPResult.fail(code, msg)); + } + resultHolder.invokeAllResult(requestMessage); + }); return result; } @@ -83,16 +116,14 @@ @ResponseBody @Operation(summary = "鑾峰彇ffmpeg.cmd妯℃澘") @Parameter(name = "mediaServerId", description = "娴佸獟浣揑D", required = true) - public WVPResult getFFmpegCMDs(@RequestParam String mediaServerId){ + public JSONObject getFFmpegCMDs(@RequestParam String mediaServerId){ logger.debug("鑾峰彇鑺傜偣[ {} ]ffmpeg.cmd妯℃澘", mediaServerId ); MediaServerItem mediaServerItem = mediaServerService.getOne(mediaServerId); - JSONObject data = streamProxyService.getFFmpegCMDs(mediaServerItem); - WVPResult<JSONObject> result = new WVPResult<>(); - result.setCode(0); - result.setMsg("success"); - result.setData(data); - return result; + if (mediaServerItem == null) { + throw new ControllerException(ErrorCode.ERROR100.getCode(), "娴佸獟浣擄細 " + mediaServerId + "鏈壘鍒�"); + } + return streamProxyService.getFFmpegCMDs(mediaServerItem); } @DeleteMapping(value = "/del") @@ -100,18 +131,13 @@ @Operation(summary = "绉婚櫎浠g悊") @Parameter(name = "app", description = "搴旂敤鍚�", required = true) @Parameter(name = "stream", description = "娴乮d", required = true) - public WVPResult del(@RequestParam String app, @RequestParam String stream){ + public void del(@RequestParam String app, @RequestParam String stream){ logger.info("绉婚櫎浠g悊锛� " + app + "/" + stream); - WVPResult<Object> result = new WVPResult<>(); if (app == null || stream == null) { - result.setCode(400); - result.setMsg(app == null ?"app涓嶈兘涓簄ull":"stream涓嶈兘涓簄ull"); + throw new ControllerException(ErrorCode.ERROR400.getCode(), app == null ?"app涓嶈兘涓簄ull":"stream涓嶈兘涓簄ull"); }else { streamProxyService.del(app, stream); - result.setCode(0); - result.setMsg("success"); } - return result; } @GetMapping(value = "/start") @@ -119,13 +145,12 @@ @Operation(summary = "鍚敤浠g悊") @Parameter(name = "app", description = "搴旂敤鍚�", required = true) @Parameter(name = "stream", description = "娴乮d", required = true) - public Object start(String app, String stream){ + public void start(String app, String stream){ logger.info("鍚敤浠g悊锛� " + app + "/" + stream); boolean result = streamProxyService.start(app, stream); if (!result) { - logger.info("鍚敤浠g悊澶辫触锛� " + app + "/" + stream); + throw new ControllerException(ErrorCode.ERROR100); } - return result?"success":"fail"; } @GetMapping(value = "/stop") @@ -133,9 +158,12 @@ @Operation(summary = "鍋滅敤浠g悊") @Parameter(name = "app", description = "搴旂敤鍚�", required = true) @Parameter(name = "stream", description = "娴乮d", required = true) - public Object stop(String app, String stream){ + public void stop(String app, String stream){ logger.info("鍋滅敤浠g悊锛� " + app + "/" + stream); boolean result = streamProxyService.stop(app, stream); - return result?"success":"fail"; + if (!result) { + logger.info("鍋滅敤浠g悊澶辫触锛� " + app + "/" + stream); + throw new ControllerException(ErrorCode.ERROR100); + } } } -- Gitblit v1.8.0