From fbdad00cdb294a857f55e33b2dd92c61dc37475c Mon Sep 17 00:00:00 2001
From: 648540858 <648540858@qq.com>
Date: 星期二, 13 十月 2020 18:55:42 +0800
Subject: [PATCH] 修复无音频通道开启通道音频后无法播放的bug 修复点播接口真的收到视频后回复

---
 src/main/java/com/genersoft/iot/vmp/vmanager/play/PlayController.java |   82 ++++++++++++++++++++++++++++++++++++++---
 1 files changed, 76 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/play/PlayController.java b/src/main/java/com/genersoft/iot/vmp/vmanager/play/PlayController.java
index 9fe78c3..996039e 100644
--- a/src/main/java/com/genersoft/iot/vmp/vmanager/play/PlayController.java
+++ b/src/main/java/com/genersoft/iot/vmp/vmanager/play/PlayController.java
@@ -1,19 +1,27 @@
 package com.genersoft.iot.vmp.vmanager.play;
 
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
+import com.genersoft.iot.vmp.common.StreamInfo;
+import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.CrossOrigin;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import com.alibaba.fastjson.JSONObject;
 import com.genersoft.iot.vmp.gb28181.bean.Device;
 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
 import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
 
+@CrossOrigin
 @RestController
 @RequestMapping("/api")
 public class PlayController {
@@ -25,22 +33,84 @@
 	
 	@Autowired
 	private IVideoManagerStorager storager;
+
+	@Autowired
+	private ZLMRESTfulUtils zlmresTfulUtils;
 	
-	@GetMapping("/play/{deviceId}_{channelId}")
+	@GetMapping("/play/{deviceId}/{channelId}")
 	public ResponseEntity<String> play(@PathVariable String deviceId,@PathVariable String channelId){
 		
 		Device device = storager.queryVideoDevice(deviceId);
-		String ssrc = cmder.playStreamCmd(device, channelId);
-		
+		StreamInfo streamInfo = cmder.playStreamCmd(device, channelId);
+		// 绛夊緟鎺ㄦ祦, TODO 榛樿瓒呮椂15s
+		boolean lockFlag = true;
+		long startTime = System.currentTimeMillis();
+		String streamId = String.format("%08x", Integer.parseInt(streamInfo.getSsrc())).toUpperCase();
+
+		// 鍒ゆ柇鎺ㄦ祦鏄惁瀛樺湪
+		while (lockFlag) {
+			try {
+				if (System.currentTimeMillis() - startTime > 15 * 1000) {
+					JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(streamId);
+					if (rtpInfo == null){
+						continue;
+					}else {
+						lockFlag = false;
+						streamInfo = storager.queryPlay(streamInfo);
+						// 鑾峰彇濯掍綋淇℃伅
+						JSONObject mediaList = zlmresTfulUtils.getMediaList("rtp", "rtmp");
+						if (mediaList.getInteger("code") == 0) {
+							JSONArray data = mediaList.getJSONArray("data");
+							if (data!= null) {
+								for (Object datum : data) {
+									JSONObject media = (JSONObject)datum;
+									if (streamId.equals(media.getString("stream"))) {
+										streamInfo.setTracks(media.getJSONArray("tracks"));
+										storager.startPlay(streamInfo);
+									}
+								}
+							}
+						}
+					};
+
+				}
+
+				Thread.sleep(200);
+			} catch (InterruptedException e) {
+				e.printStackTrace();
+			}
+		}
+
 		if (logger.isDebugEnabled()) {
 			logger.debug(String.format("璁惧棰勮 API璋冪敤锛宒eviceId锛�%s 锛宑hannelId锛�%s",deviceId, channelId));
-			logger.debug("璁惧棰勮 API璋冪敤锛宻src锛�"+ssrc+",ZLMedia streamId:"+Integer.toHexString(Integer.parseInt(ssrc)));
+			logger.debug("璁惧棰勮 API璋冪敤锛宻src锛�"+streamInfo.getSsrc()+",ZLMedia streamId:"+Integer.toHexString(Integer.parseInt(streamInfo.getSsrc())));
+		}
+		
+		if(streamInfo!=null) {
+			return new ResponseEntity<String>(JSON.toJSONString(streamInfo),HttpStatus.OK);
+		} else {
+			logger.warn("璁惧棰勮API璋冪敤澶辫触锛�");
+			return new ResponseEntity<String>(HttpStatus.INTERNAL_SERVER_ERROR);
+		}
+	}
+	
+	@PostMapping("/play/{ssrc}/stop")
+	public ResponseEntity<String> playStop(@PathVariable String ssrc){
+		
+		cmder.streamByeCmd(ssrc);
+		StreamInfo streamInfo = storager.queryPlayBySSRC(ssrc);
+		if (streamInfo == null) return new ResponseEntity<String>(HttpStatus.PAYMENT_REQUIRED);
+		storager.stopPlay(streamInfo);
+		if (logger.isDebugEnabled()) {
+			logger.debug(String.format("璁惧棰勮鍋滄API璋冪敤锛宻src锛�%s", ssrc));
 		}
 		
 		if(ssrc!=null) {
-			return new ResponseEntity<String>(ssrc,HttpStatus.OK);
+			JSONObject json = new JSONObject();
+			json.put("ssrc", ssrc);
+			return new ResponseEntity<String>(json.toString(),HttpStatus.OK);
 		} else {
-			logger.warn("璁惧棰勮API璋冪敤澶辫触锛�");
+			logger.warn("璁惧棰勮鍋滄API璋冪敤澶辫触锛�");
 			return new ResponseEntity<String>(HttpStatus.INTERNAL_SERVER_ERROR);
 		}
 	}

--
Gitblit v1.8.0