old mode 100644
new mode 100755
|  |  |  | 
|---|
|  |  |  | package com.genersoft.iot.vmp.vmanager.gb28181.play; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.alibaba.fastjson.JSONArray; | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSONArray; | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSONObject; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.InviteInfo; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.InviteSessionStatus; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.InviteSessionType; | 
|---|
|  |  |  | 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.conf.exception.SsrcTransactionNotFoundException; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.security.JwtUtils; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.Device; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.SsrcTransaction; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.Device; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.IInviteStreamService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.IMediaServerService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.vmanager.bean.WVPResult; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.vmanager.gb28181.play.bean.PlayResult; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.IMediaService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.IPlayService; | 
|---|
|  |  |  | import io.swagger.annotations.Api; | 
|---|
|  |  |  | import io.swagger.annotations.ApiImplicitParam; | 
|---|
|  |  |  | import io.swagger.annotations.ApiImplicitParams; | 
|---|
|  |  |  | import io.swagger.annotations.ApiOperation; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.bean.InviteErrorCode; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.utils.DateUtil; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.vmanager.bean.*; | 
|---|
|  |  |  | 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 io.swagger.v3.oas.annotations.Operation; | 
|---|
|  |  |  | import io.swagger.v3.oas.annotations.Parameter; | 
|---|
|  |  |  | import io.swagger.v3.oas.annotations.security.SecurityRequirement; | 
|---|
|  |  |  | 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.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.transmit.cmd.impl.SIPCommander; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorager; | 
|---|
|  |  |  | import org.springframework.web.bind.annotation.*; | 
|---|
|  |  |  | import org.springframework.web.context.request.async.DeferredResult; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import javax.servlet.http.HttpServletRequest; | 
|---|
|  |  |  | import javax.sip.InvalidArgumentException; | 
|---|
|  |  |  | import javax.sip.SipException; | 
|---|
|  |  |  | import java.net.MalformedURLException; | 
|---|
|  |  |  | import java.net.URL; | 
|---|
|  |  |  | import java.text.ParseException; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.Map; | 
|---|
|  |  |  | import java.util.UUID; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import javax.sip.message.Response; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Api(tags = "国标设备点播") | 
|---|
|  |  |  | @CrossOrigin | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @author lin | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @Tag(name  = "国标设备点播") | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @RestController | 
|---|
|  |  |  | @RequestMapping("/api/play") | 
|---|
|  |  |  | public class PlayController { | 
|---|
|  |  |  | 
|---|
|  |  |  | private VideoStreamSessionManager streamSession; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IVideoManagerStorager storager; | 
|---|
|  |  |  | private IVideoManagerStorage storager; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IRedisCatchStorage redisCatchStorage; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IInviteStreamService inviteStreamService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ZLMRESTfulUtils zlmresTfulUtils; | 
|---|
|  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IMediaServerService mediaServerService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @ApiOperation("开始点播") | 
|---|
|  |  |  | @ApiImplicitParams({ | 
|---|
|  |  |  | @ApiImplicitParam(name = "deviceId", value = "设备ID", dataTypeClass = String.class), | 
|---|
|  |  |  | @ApiImplicitParam(name = "channelId", value = "通道ID", dataTypeClass = String.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @GetMapping("/start/{deviceId}/{channelId}") | 
|---|
|  |  |  | public DeferredResult<ResponseEntity<String>> play(@PathVariable String deviceId, | 
|---|
|  |  |  | @PathVariable String channelId) { | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private UserSetting userSetting; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Operation(summary = "开始点播", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @Parameter(name = "deviceId", description = "设备国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "channelId", description = "通道国标编号", required = true) | 
|---|
|  |  |  | @GetMapping("/start/{deviceId}/{channelId}") | 
|---|
|  |  |  | public DeferredResult<WVPResult<StreamContent>> play(HttpServletRequest request, @PathVariable String deviceId, | 
|---|
|  |  |  | @PathVariable String channelId) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | logger.info("[开始点播] deviceId:{}, channelId:{}, ", deviceId, channelId); | 
|---|
|  |  |  | // 获取可用的zlm | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | MediaServerItem newMediaServerItem = playService.getNewMediaServerItem(device); | 
|---|
|  |  |  | PlayResult playResult = playService.play(newMediaServerItem, deviceId, channelId, null, null); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return playResult.getResult(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | RequestMessage requestMessage = new RequestMessage(); | 
|---|
|  |  |  | String key = DeferredResultHolder.CALLBACK_CMD_PLAY + deviceId + channelId; | 
|---|
|  |  |  | requestMessage.setKey(key); | 
|---|
|  |  |  | String uuid = UUID.randomUUID().toString(); | 
|---|
|  |  |  | requestMessage.setId(uuid); | 
|---|
|  |  |  | DeferredResult<WVPResult<StreamContent>> result = new DeferredResult<>(userSetting.getPlayTimeout().longValue()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @ApiOperation("停止点播") | 
|---|
|  |  |  | @ApiImplicitParams({ | 
|---|
|  |  |  | @ApiImplicitParam(name = "deviceId", value = "设备ID", dataTypeClass = String.class), | 
|---|
|  |  |  | @ApiImplicitParam(name = "channelId", value = "通道ID", dataTypeClass = String.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @GetMapping("/stop/{deviceId}/{channelId}") | 
|---|
|  |  |  | public DeferredResult<ResponseEntity<String>> playStop(@PathVariable String deviceId, @PathVariable String channelId) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | logger.debug(String.format("设备预览/回放停止API调用,streamId:%s/$s", deviceId, channelId )); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | UUID uuid = UUID.randomUUID(); | 
|---|
|  |  |  | DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(); | 
|---|
|  |  |  | result.onTimeout(()->{ | 
|---|
|  |  |  | logger.info("[点播等待超时] deviceId:{}, channelId:{}, ", deviceId, channelId); | 
|---|
|  |  |  | // 释放rtpserver | 
|---|
|  |  |  | WVPResult<StreamInfo> wvpResult = new WVPResult<>(); | 
|---|
|  |  |  | wvpResult.setCode(ErrorCode.ERROR100.getCode()); | 
|---|
|  |  |  | wvpResult.setMsg("点播超时"); | 
|---|
|  |  |  | requestMessage.setData(wvpResult); | 
|---|
|  |  |  | resultHolder.invokeAllResult(requestMessage); | 
|---|
|  |  |  | inviteStreamService.removeInviteInfoByDeviceAndChannel(InviteSessionType.PLAY, deviceId, channelId); | 
|---|
|  |  |  | storager.stopPlay(deviceId, channelId); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 录像查询以channelId作为deviceId查询 | 
|---|
|  |  |  | resultHolder.put(DeferredResultHolder.CALLBACK_CMD_STOP + uuid, result); | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | cmder.streamByeCmd(deviceId, channelId, (event) -> { | 
|---|
|  |  |  | StreamInfo streamInfo = redisCatchStorage.queryPlayByDevice(deviceId, channelId); | 
|---|
|  |  |  | if (streamInfo == null) { | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_STOP + uuid); | 
|---|
|  |  |  | msg.setData("点播未找到"); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | storager.stopPlay(deviceId, channelId); | 
|---|
|  |  |  | resultHolder.put(key, uuid, result); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | playService.play(newMediaServerItem, deviceId, channelId, null, (code, msg, data) -> { | 
|---|
|  |  |  | WVPResult<StreamContent> wvpResult = new WVPResult<>(); | 
|---|
|  |  |  | if (code == InviteErrorCode.SUCCESS.getCode()) { | 
|---|
|  |  |  | wvpResult.setCode(ErrorCode.SUCCESS.getCode()); | 
|---|
|  |  |  | wvpResult.setMsg(ErrorCode.SUCCESS.getMsg()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (data != null) { | 
|---|
|  |  |  | StreamInfo streamInfo = (StreamInfo)data; | 
|---|
|  |  |  | if (userSetting.getUseSourceIpAsStreamIp()) { | 
|---|
|  |  |  | streamInfo=streamInfo.clone();//深拷贝 | 
|---|
|  |  |  | String host; | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | URL url=new URL(request.getRequestURL().toString()); | 
|---|
|  |  |  | host=url.getHost(); | 
|---|
|  |  |  | } catch (MalformedURLException e) { | 
|---|
|  |  |  | host=request.getLocalAddr(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | streamInfo.channgeStreamIp(host); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | wvpResult.setData(new StreamContent(streamInfo)); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | wvpResult.setCode(code); | 
|---|
|  |  |  | wvpResult.setMsg(msg); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | redisCatchStorage.stopPlay(streamInfo); | 
|---|
|  |  |  | storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId()); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_STOP + uuid); | 
|---|
|  |  |  | //Response response = event.getResponse(); | 
|---|
|  |  |  | msg.setData(String.format("success")); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | wvpResult.setCode(code); | 
|---|
|  |  |  | wvpResult.setMsg(msg); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | mediaServerService.closeRTPServer(device, channelId); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (deviceId != null || channelId != null) { | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("deviceId", deviceId); | 
|---|
|  |  |  | json.put("channelId", channelId); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid); | 
|---|
|  |  |  | msg.setData(json.toString()); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | logger.warn("设备预览/回放停止API调用失败!"); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid); | 
|---|
|  |  |  | msg.setData("streamId null"); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 超时处理 | 
|---|
|  |  |  | result.onTimeout(()->{ | 
|---|
|  |  |  | logger.warn(String.format("设备预览/回放停止超时,deviceId/channelId:%s/$s ", deviceId, channelId)); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_STOP + uuid); | 
|---|
|  |  |  | msg.setData("Timeout"); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | requestMessage.setData(wvpResult); | 
|---|
|  |  |  | resultHolder.invokeResult(requestMessage); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Operation(summary = "停止点播", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @Parameter(name = "deviceId", description = "设备国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "channelId", description = "通道国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "isSubStream", description = "是否子码流(true-子码流,false-主码流),默认为false", required = true) | 
|---|
|  |  |  | @GetMapping("/stop/{deviceId}/{channelId}") | 
|---|
|  |  |  | public JSONObject playStop(@PathVariable String deviceId, @PathVariable String channelId,boolean isSubStream) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | logger.debug(String.format("设备预览/回放停止API调用,streamId:%s_%s", deviceId, channelId )); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (deviceId == null || channelId == null) { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR400); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | if (device == null) { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "设备[" + deviceId + "]不存在"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | InviteInfo inviteInfo = inviteStreamService.getInviteInfoByDeviceAndChannel(InviteSessionType.PLAY, deviceId, channelId); | 
|---|
|  |  |  | if (inviteInfo == null) { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "点播未找到"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (InviteSessionStatus.ok == inviteInfo.getStatus()) { | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | logger.info("[停止点播] {}/{}", device.getDeviceId(), channelId); | 
|---|
|  |  |  | cmder.streamByeCmd(device, channelId, inviteInfo.getStream(), null, null); | 
|---|
|  |  |  | } catch (InvalidArgumentException | SipException | ParseException | SsrcTransactionNotFoundException e) { | 
|---|
|  |  |  | logger.error("[命令发送失败] 停止点播, 发送BYE: {}", e.getMessage()); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "命令发送失败: " + e.getMessage()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | inviteStreamService.removeInviteInfoByDeviceAndChannel(InviteSessionType.PLAY, deviceId, channelId); | 
|---|
|  |  |  | storager.stopPlay(deviceId, channelId); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("deviceId", deviceId); | 
|---|
|  |  |  | json.put("channelId", channelId); | 
|---|
|  |  |  | json.put("isSubStream", isSubStream); | 
|---|
|  |  |  | return json; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 将不是h264的视频通过ffmpeg 转码为h264 + aac | 
|---|
|  |  |  | * @param streamId 流ID | 
|---|
|  |  |  | * @return | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ApiOperation("将不是h264的视频通过ffmpeg 转码为h264 + aac") | 
|---|
|  |  |  | @ApiImplicitParams({ | 
|---|
|  |  |  | @ApiImplicitParam(name = "streamId", value = "视频流ID", dataTypeClass = String.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @Operation(summary = "将不是h264的视频通过ffmpeg 转码为h264 + aac", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @Parameter(name = "streamId", description = "视频流ID", required = true) | 
|---|
|  |  |  | @PostMapping("/convert/{streamId}") | 
|---|
|  |  |  | public ResponseEntity<String> playConvert(@PathVariable String streamId) { | 
|---|
|  |  |  | StreamInfo streamInfo = redisCatchStorage.queryPlayByStreamId(streamId); | 
|---|
|  |  |  | if (streamInfo == null) { | 
|---|
|  |  |  | streamInfo = redisCatchStorage.queryPlaybackByStreamId(streamId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (streamInfo == null) { | 
|---|
|  |  |  | public JSONObject playConvert(@PathVariable String streamId) { | 
|---|
|  |  |  | //      StreamInfo streamInfo = redisCatchStorage.queryPlayByStreamId(streamId); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | InviteInfo inviteInfo = inviteStreamService.getInviteInfoByStream(null, streamId); | 
|---|
|  |  |  | if (inviteInfo == null || inviteInfo.getStreamInfo() == null) { | 
|---|
|  |  |  | logger.warn("视频转码API调用失败!, 视频流已经停止!"); | 
|---|
|  |  |  | return new ResponseEntity<String>("未找到视频流信息, 视频流可能已经停止", HttpStatus.OK); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "未找到视频流信息, 视频流可能已经停止"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(streamInfo.getMediaServerId()); | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(inviteInfo.getStreamInfo().getMediaServerId()); | 
|---|
|  |  |  | JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(mediaInfo, streamId); | 
|---|
|  |  |  | if (!rtpInfo.getBoolean("exist")) { | 
|---|
|  |  |  | logger.warn("视频转码API调用失败!, 视频流已停止推流!"); | 
|---|
|  |  |  | return new ResponseEntity<String>("推流信息在流媒体中不存在, 视频流可能已停止推流", HttpStatus.OK); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "未找到视频流信息, 视频流可能已停止推流"); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | String dstUrl = String.format("rtmp://%s:%s/convert/%s", "127.0.0.1", mediaInfo.getRtmpPort(), | 
|---|
|  |  |  | streamId ); | 
|---|
|  |  |  | String srcUrl = String.format("rtsp://%s:%s/rtp/%s", "127.0.0.1", mediaInfo.getRtspPort(), streamId); | 
|---|
|  |  |  | JSONObject jsonObject = zlmresTfulUtils.addFFmpegSource(mediaInfo, srcUrl, dstUrl, "1000000", true, false, null); | 
|---|
|  |  |  | logger.info(jsonObject.toJSONString()); | 
|---|
|  |  |  | JSONObject result = new JSONObject(); | 
|---|
|  |  |  | if (jsonObject != null && jsonObject.getInteger("code") == 0) { | 
|---|
|  |  |  | result.put("code", 0); | 
|---|
|  |  |  | JSONObject data = jsonObject.getJSONObject("data"); | 
|---|
|  |  |  | if (data != null) { | 
|---|
|  |  |  | result.put("key", data.getString("key")); | 
|---|
|  |  |  | StreamInfo streamInfoResult = mediaService.getStreamInfoByAppAndStreamWithCheck("convert", streamId, mediaInfo.getId()); | 
|---|
|  |  |  | result.put("data", streamInfoResult); | 
|---|
|  |  |  | JSONObject result = new JSONObject(); | 
|---|
|  |  |  | result.put("key", data.getString("key")); | 
|---|
|  |  |  | StreamInfo streamInfoResult = mediaService.getStreamInfoByAppAndStreamWithCheck("convert", streamId, mediaInfo.getId(), false); | 
|---|
|  |  |  | result.put("StreamInfo", streamInfoResult); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "转码失败"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | result.put("code", 1); | 
|---|
|  |  |  | result.put("msg", "cover fail"); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "转码失败"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return new ResponseEntity<String>( result.toJSONString(), HttpStatus.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 结束转码 | 
|---|
|  |  |  | * @param key | 
|---|
|  |  |  | * @return | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ApiOperation("结束转码") | 
|---|
|  |  |  | @ApiImplicitParams({ | 
|---|
|  |  |  | @ApiImplicitParam(name = "key", value = "视频流key", dataTypeClass = String.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @Operation(summary = "结束转码", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @Parameter(name = "key", description = "视频流key", required = true) | 
|---|
|  |  |  | @Parameter(name = "mediaServerId", description = "流媒体服务ID", required = true) | 
|---|
|  |  |  | @PostMapping("/convertStop/{key}") | 
|---|
|  |  |  | public ResponseEntity<String> playConvertStop(@PathVariable String key, String mediaServerId) { | 
|---|
|  |  |  | JSONObject result = new JSONObject(); | 
|---|
|  |  |  | public void playConvertStop(@PathVariable String key, String mediaServerId) { | 
|---|
|  |  |  | if (mediaServerId == null) { | 
|---|
|  |  |  | result.put("code", 400); | 
|---|
|  |  |  | result.put("msg", "mediaServerId is null"); | 
|---|
|  |  |  | return new ResponseEntity<String>( result.toJSONString(), HttpStatus.BAD_REQUEST); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR400.getCode(), "流媒体:" + mediaServerId + "不存在" ); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  | if (mediaInfo == null) { | 
|---|
|  |  |  | result.put("code", 0); | 
|---|
|  |  |  | result.put("msg", "使用的流媒体已经停止运行"); | 
|---|
|  |  |  | return new ResponseEntity<String>( result.toJSONString(), HttpStatus.OK); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100.getCode(), "使用的流媒体已经停止运行" ); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | JSONObject jsonObject = zlmresTfulUtils.delFFmpegSource(mediaInfo, key); | 
|---|
|  |  |  | logger.info(jsonObject.toJSONString()); | 
|---|
|  |  |  | if (jsonObject != null && jsonObject.getInteger("code") == 0) { | 
|---|
|  |  |  | result.put("code", 0); | 
|---|
|  |  |  | JSONObject data = jsonObject.getJSONObject("data"); | 
|---|
|  |  |  | if (data != null && data.getBoolean("flag")) { | 
|---|
|  |  |  | result.put("code", "0"); | 
|---|
|  |  |  | result.put("msg", "success"); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (data == null || data.getBoolean("flag") == null || !data.getBoolean("flag")) { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100 ); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | result.put("code", 1); | 
|---|
|  |  |  | result.put("msg", "delFFmpegSource fail"); | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR100 ); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return new ResponseEntity<String>( result.toJSONString(), HttpStatus.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Operation(summary = "语音广播命令", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @Parameter(name = "deviceId", description = "设备国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "deviceId", description = "通道国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "timeout", description = "推流超时时间(秒)", required = true) | 
|---|
|  |  |  | @GetMapping("/broadcast/{deviceId}/{channelId}") | 
|---|
|  |  |  | @PostMapping("/broadcast/{deviceId}/{channelId}") | 
|---|
|  |  |  | public AudioBroadcastResult broadcastApi(@PathVariable String deviceId, @PathVariable String channelId, Integer timeout, Boolean broadcastMode) { | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("语音广播API调用"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | if (device == null) { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR400.getCode(), "未找到设备: " + deviceId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (channelId == null) { | 
|---|
|  |  |  | throw new ControllerException(ErrorCode.ERROR400.getCode(), "未找到通道: " + channelId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return playService.audioBroadcast(device, channelId, broadcastMode); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @ApiOperation("语音广播命令") | 
|---|
|  |  |  | @ApiImplicitParams({ | 
|---|
|  |  |  | @ApiImplicitParam(name = "deviceId", value = "设备Id", dataTypeClass = String.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @GetMapping("/broadcast/{deviceId}") | 
|---|
|  |  |  | @PostMapping("/broadcast/{deviceId}") | 
|---|
|  |  |  | public DeferredResult<ResponseEntity<String>> broadcastApi(@PathVariable String deviceId) { | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("语音广播API调用"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(3 * 1000L); | 
|---|
|  |  |  | if (device == null) { | 
|---|
|  |  |  | resultHolder.put(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId, result); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId); | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("DeviceID", deviceId); | 
|---|
|  |  |  | json.put("CmdType", "Broadcast"); | 
|---|
|  |  |  | json.put("Result", "Failed"); | 
|---|
|  |  |  | json.put("Description", "Device 不存在"); | 
|---|
|  |  |  | msg.setData(json); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | @Operation(summary = "停止语音广播") | 
|---|
|  |  |  | @Parameter(name = "deviceId", description = "设备Id", required = true) | 
|---|
|  |  |  | @Parameter(name = "channelId", description = "通道Id", required = true) | 
|---|
|  |  |  | @GetMapping("/broadcast/stop/{deviceId}/{channelId}") | 
|---|
|  |  |  | @PostMapping("/broadcast/stop/{deviceId}/{channelId}") | 
|---|
|  |  |  | public void stopBroadcast(@PathVariable String deviceId, @PathVariable String channelId) { | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("停止语音广播API调用"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | cmder.audioBroadcastCmd(device, (event) -> { | 
|---|
|  |  |  | Response response = event.getResponse(); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId); | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("DeviceID", deviceId); | 
|---|
|  |  |  | json.put("CmdType", "Broadcast"); | 
|---|
|  |  |  | json.put("Result", "Failed"); | 
|---|
|  |  |  | json.put("Description", String.format("语音广播操作失败,错误码: %s, %s", response.getStatusCode(), response.getReasonPhrase())); | 
|---|
|  |  |  | msg.setData(json); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | result.onTimeout(() -> { | 
|---|
|  |  |  | logger.warn(String.format("语音广播操作超时, 设备未返回应答指令")); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId); | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("DeviceID", deviceId); | 
|---|
|  |  |  | json.put("CmdType", "Broadcast"); | 
|---|
|  |  |  | json.put("Result", "Failed"); | 
|---|
|  |  |  | json.put("Error", "Timeout. Device did not response to broadcast command."); | 
|---|
|  |  |  | msg.setData(json); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | resultHolder.put(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId, result); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | //      try { | 
|---|
|  |  |  | //         playService.stopAudioBroadcast(deviceId, channelId); | 
|---|
|  |  |  | //      } catch (InvalidArgumentException | ParseException  | SipException e) { | 
|---|
|  |  |  | //         logger.error("[命令发送失败] 停止语音: {}", e.getMessage()); | 
|---|
|  |  |  | //         throw new ControllerException(ErrorCode.ERROR100.getCode(), "命令发送失败: " +  e.getMessage()); | 
|---|
|  |  |  | //      } | 
|---|
|  |  |  | playService.stopAudioBroadcast(deviceId, channelId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @ApiOperation("获取所有的ssrc") | 
|---|
|  |  |  | @Operation(summary = "获取所有的ssrc", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @GetMapping("/ssrc") | 
|---|
|  |  |  | public WVPResult<JSONObject> getSSRC() { | 
|---|
|  |  |  | public JSONObject getSSRC() { | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("获取所有的ssrc"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | jsonObject.put("deviceId", transaction.getDeviceId()); | 
|---|
|  |  |  | jsonObject.put("channelId", transaction.getChannelId()); | 
|---|
|  |  |  | jsonObject.put("ssrc", transaction.getSsrc()); | 
|---|
|  |  |  | jsonObject.put("streamId", transaction.getStreamId()); | 
|---|
|  |  |  | jsonObject.put("streamId", transaction.getStream()); | 
|---|
|  |  |  | objects.add(jsonObject); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | WVPResult<JSONObject> result = new WVPResult<>(); | 
|---|
|  |  |  | result.setCode(0); | 
|---|
|  |  |  | result.setMsg("success"); | 
|---|
|  |  |  | JSONObject jsonObject = new JSONObject(); | 
|---|
|  |  |  | jsonObject.put("data", objects); | 
|---|
|  |  |  | jsonObject.put("count", objects.size()); | 
|---|
|  |  |  | result.setData(jsonObject); | 
|---|
|  |  |  | return jsonObject; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Operation(summary = "获取截图", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
|---|
|  |  |  | @Parameter(name = "deviceId", description = "设备国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "channelId", description = "通道国标编号", required = true) | 
|---|
|  |  |  | @Parameter(name = "isSubStream", description = "是否子码流(true-子码流,false-主码流),默认为false", required = true) | 
|---|
|  |  |  | @GetMapping("/snap") | 
|---|
|  |  |  | public DeferredResult<String> getSnap(String deviceId, String channelId,boolean isSubStream) { | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("获取截图: {}/{}", deviceId, channelId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | DeferredResult<String> result = new DeferredResult<>(3 * 1000L); | 
|---|
|  |  |  | String key  = DeferredResultHolder.CALLBACK_CMD_SNAP + deviceId; | 
|---|
|  |  |  | String uuid  = UUID.randomUUID().toString(); | 
|---|
|  |  |  | resultHolder.put(key, uuid,  result); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | RequestMessage message = new RequestMessage(); | 
|---|
|  |  |  | message.setKey(key); | 
|---|
|  |  |  | message.setId(uuid); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | String fileName = deviceId + "_" + channelId + "_" + DateUtil.getNowForUrl() + ".jpg"; | 
|---|
|  |  |  | playService.getSnap(deviceId, channelId, fileName, (code, msg, data) -> { | 
|---|
|  |  |  | if (code == InviteErrorCode.SUCCESS.getCode()) { | 
|---|
|  |  |  | message.setData(data); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | message.setData(WVPResult.fail(code, msg)); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | resultHolder.invokeResult(message); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|