old mode 100644
new mode 100755
| | |
| | | |
| | | import com.alibaba.fastjson2.JSONArray; |
| | | import com.alibaba.fastjson2.JSONObject; |
| | | import com.fasterxml.jackson.databind.ObjectMapper; |
| | | import com.genersoft.iot.vmp.common.InviteInfo; |
| | | 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.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.bean.MediaServer; |
| | | import com.genersoft.iot.vmp.media.service.IMediaServerService; |
| | | import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils; |
| | | import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; |
| | | import com.genersoft.iot.vmp.service.IMediaServerService; |
| | | import com.genersoft.iot.vmp.service.IMediaService; |
| | | import com.genersoft.iot.vmp.service.IInviteStreamService; |
| | | import com.genersoft.iot.vmp.service.IPlayService; |
| | | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; |
| | | import com.genersoft.iot.vmp.service.bean.InviteErrorCode; |
| | | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; |
| | | import com.genersoft.iot.vmp.vmanager.bean.*; |
| | | import com.genersoft.iot.vmp.utils.DateUtil; |
| | | import com.genersoft.iot.vmp.utils.IdUtils; |
| | | import com.genersoft.iot.vmp.vmanager.bean.AudioBroadcastResult; |
| | | 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 okhttp3.HttpUrl; |
| | | import okhttp3.OkHttpClient; |
| | | import okhttp3.Request; |
| | | import okhttp3.Response; |
| | | import org.bytedeco.javacv.*; |
| | | import org.bytedeco.opencv.global.opencv_imgcodecs; |
| | | import org.bytedeco.opencv.opencv_core.Mat; |
| | | import org.slf4j.Logger; |
| | | import org.slf4j.LoggerFactory; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.beans.factory.annotation.Value; |
| | | import org.springframework.util.ObjectUtils; |
| | | import org.springframework.util.StringUtils; |
| | | import org.springframework.web.bind.annotation.*; |
| | | import org.springframework.web.context.request.async.DeferredResult; |
| | | |
| | | import javax.imageio.ImageIO; |
| | | import javax.servlet.http.HttpServletRequest; |
| | | import javax.sip.InvalidArgumentException; |
| | | import javax.sip.SipException; |
| | | import java.text.ParseException; |
| | | import java.awt.image.BufferedImage; |
| | | import java.io.BufferedReader; |
| | | import java.io.File; |
| | | import java.io.IOException; |
| | | import java.io.InputStreamReader; |
| | | import java.net.*; |
| | | import java.text.SimpleDateFormat; |
| | | import java.util.Date; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.UUID; |
| | | |
| | | |
| | | /** |
| | | * @author lin |
| | | */ |
| | | @Tag(name = "国标设备点播") |
| | | @Tag(name = "国标设备点播") |
| | | |
| | | @RestController |
| | | @RequestMapping("/api/play") |
| | | public class PlayController { |
| | | |
| | | private final static Logger logger = LoggerFactory.getLogger(PlayController.class); |
| | | private final static Logger logger = LoggerFactory.getLogger(PlayController.class); |
| | | @Value("${platform.upload}") |
| | | public String upload; |
| | | @Autowired |
| | | private SIPCommander cmder; |
| | | @Autowired |
| | | private ZLMRESTfulUtils zlmresTfulUtils; |
| | | @Autowired |
| | | private VideoStreamSessionManager streamSession; |
| | | |
| | | @Autowired |
| | | private SIPCommander cmder; |
| | | @Autowired |
| | | private IVideoManagerStorage storager; |
| | | |
| | | @Autowired |
| | | private VideoStreamSessionManager streamSession; |
| | | @Autowired |
| | | private IInviteStreamService inviteStreamService; |
| | | |
| | | @Autowired |
| | | private IVideoManagerStorage storager; |
| | | @Autowired |
| | | private DeferredResultHolder resultHolder; |
| | | |
| | | @Autowired |
| | | private IRedisCatchStorage redisCatchStorage; |
| | | @Autowired |
| | | private IPlayService playService; |
| | | |
| | | @Autowired |
| | | private ZLMRESTfulUtils zlmresTfulUtils; |
| | | @Autowired |
| | | private IMediaServerService mediaServerService; |
| | | |
| | | @Autowired |
| | | private DeferredResultHolder resultHolder; |
| | | @Autowired |
| | | private UserSetting userSetting; |
| | | |
| | | @Autowired |
| | | private IPlayService playService; |
| | | private final SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd"); |
| | | |
| | | @Autowired |
| | | private IMediaService mediaService; |
| | | @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) { |
| | | |
| | | @Autowired |
| | | private IMediaServerService mediaServerService; |
| | | logger.info("[开始点播] deviceId:{}, channelId:{}, ", deviceId, channelId); |
| | | // 获取可用的zlm |
| | | Device device = storager.queryVideoDevice(deviceId); |
| | | MediaServer newMediaServerItem = playService.getNewMediaServerItem(device); |
| | | |
| | | @Autowired |
| | | private UserSetting userSetting; |
| | | 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()); |
| | | |
| | | @Operation(summary = "开始点播") |
| | | @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) { |
| | | 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); |
| | | }); |
| | | |
| | | // 获取可用的zlm |
| | | Device device = storager.queryVideoDevice(deviceId); |
| | | MediaServerItem newMediaServerItem = playService.getNewMediaServerItem(device); |
| | | // 录像查询以channelId作为deviceId查询 |
| | | resultHolder.put(key, uuid, result); |
| | | |
| | | RequestMessage msg = new RequestMessage(); |
| | | String key = DeferredResultHolder.CALLBACK_CMD_PLAY + deviceId + channelId; |
| | | boolean exist = resultHolder.exist(key, null); |
| | | msg.setKey(key); |
| | | String uuid = UUID.randomUUID().toString(); |
| | | msg.setId(uuid); |
| | | DeferredResult<WVPResult<StreamContent>> result = new DeferredResult<>(userSetting.getPlayTimeout().longValue()); |
| | | DeferredResultEx<WVPResult<StreamContent>> deferredResultEx = new DeferredResultEx<>(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()); |
| | | |
| | | result.onTimeout(()->{ |
| | | logger.info("点播接口等待超时"); |
| | | // 释放rtpserver |
| | | WVPResult<StreamInfo> wvpResult = new WVPResult<>(); |
| | | wvpResult.setCode(ErrorCode.ERROR100.getCode()); |
| | | wvpResult.setMsg("点播超时"); |
| | | msg.setData(wvpResult); |
| | | resultHolder.invokeResult(msg); |
| | | }); |
| | | // TODO 在点播未成功的情况下在此调用接口点播会导致返回的流地址ip错误 |
| | | deferredResultEx.setFilter(result1 -> { |
| | | WVPResult<StreamInfo> wvpResult1 = (WVPResult<StreamInfo>)result1; |
| | | WVPResult<StreamContent> resultStream = new WVPResult<>(); |
| | | resultStream.setCode(wvpResult1.getCode()); |
| | | resultStream.setMsg(wvpResult1.getMsg()); |
| | | if (wvpResult1.getCode() == ErrorCode.SUCCESS.getCode()) { |
| | | StreamInfo data = wvpResult1.getData().clone(); |
| | | if (userSetting.getUseSourceIpAsStreamIp()) { |
| | | data.channgeStreamIp(request.getLocalName()); |
| | | } |
| | | resultStream.setData(new StreamContent(wvpResult1.getData())); |
| | | } |
| | | return resultStream; |
| | | }); |
| | | 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); |
| | | } |
| | | if (!ObjectUtils.isEmpty(newMediaServerItem.getTranscodeSuffix()) && !"null".equalsIgnoreCase(newMediaServerItem.getTranscodeSuffix())) { |
| | | streamInfo.setStream(streamInfo.getStream() + "_" + newMediaServerItem.getTranscodeSuffix()); |
| | | } |
| | | wvpResult.setData(new StreamContent(streamInfo)); |
| | | } else { |
| | | wvpResult.setCode(code); |
| | | wvpResult.setMsg(msg); |
| | | } |
| | | } else { |
| | | wvpResult.setCode(code); |
| | | wvpResult.setMsg(msg); |
| | | } |
| | | requestMessage.setData(wvpResult); |
| | | // 此处必须释放所有请求 |
| | | resultHolder.invokeAllResult(requestMessage); |
| | | }); |
| | | return result; |
| | | } |
| | | |
| | | |
| | | // 录像查询以channelId作为deviceId查询 |
| | | resultHolder.put(key, uuid, deferredResultEx); |
| | | @Operation(summary = "开始点播并返回图片", security = @SecurityRequirement(name = JwtUtils.HEADER)) |
| | | @Parameter(name = "deviceId", description = "设备国标编号", required = true) |
| | | @Parameter(name = "channelId", description = "通道国标编号", required = true) |
| | | @GetMapping("/start/img/{deviceId}/{channelId}") |
| | | public DeferredResult<WVPResult<String>> playReturnImg(HttpServletRequest request, @PathVariable String deviceId, |
| | | @PathVariable String channelId) throws IOException { |
| | | logger.info("[开始点播] deviceId:{}, channelId:{}, ", deviceId, channelId); |
| | | // 获取可用的zlm |
| | | Device device = storager.queryVideoDevice(deviceId); |
| | | MediaServer newMediaServerItem = playService.getNewMediaServerItem(device); |
| | | RequestMessage requestMessage = new RequestMessage(); |
| | | String key = DeferredResultHolder.CALLBACK_CMD_PLAY + deviceId + channelId; |
| | | requestMessage.setKey(key); |
| | | String uuid = UUID.randomUUID().toString(); |
| | | requestMessage.setId(uuid); |
| | | |
| | | if (!exist) { |
| | | playService.play(newMediaServerItem, deviceId, channelId, null, null, null); |
| | | } |
| | | return result; |
| | | } |
| | | //存活时间 |
| | | DeferredResult<WVPResult<String>> result = new DeferredResult<>(120 * 1000L); |
| | | |
| | | @Operation(summary = "停止点播") |
| | | @Parameter(name = "deviceId", description = "设备国标编号", required = true) |
| | | @Parameter(name = "channelId", description = "通道国标编号", required = true) |
| | | @GetMapping("/stop/{deviceId}/{channelId}") |
| | | public JSONObject playStop(@PathVariable String deviceId, @PathVariable String channelId) { |
| | | //120秒存活 |
| | | 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); |
| | | }); |
| | | |
| | | logger.debug(String.format("设备预览/回放停止API调用,streamId:%s_%s", deviceId, channelId )); |
| | | // 录像查询以channelId作为deviceId查询 |
| | | resultHolder.put(key, uuid, result); |
| | | |
| | | if (deviceId == null || channelId == null) { |
| | | throw new ControllerException(ErrorCode.ERROR400); |
| | | } |
| | | playService.play(newMediaServerItem, deviceId, channelId, null, (code, msg, data) -> { |
| | | WVPResult<String> wvpResult = new WVPResult<>(); |
| | | if (code == InviteErrorCode.SUCCESS.getCode()) { |
| | | wvpResult.setCode(ErrorCode.SUCCESS.getCode()); |
| | | wvpResult.setMsg(ErrorCode.SUCCESS.getMsg()); |
| | | |
| | | Device device = storager.queryVideoDevice(deviceId); |
| | | if (device == null) { |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "设备[" + deviceId + "]不存在"); |
| | | } |
| | | 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); |
| | | } |
| | | if (!ObjectUtils.isEmpty(newMediaServerItem.getTranscodeSuffix()) && !"null".equalsIgnoreCase(newMediaServerItem.getTranscodeSuffix())) { |
| | | streamInfo.setStream(streamInfo.getStream() + "_" + newMediaServerItem.getTranscodeSuffix()); |
| | | } |
| | | StreamContent streamContent = new StreamContent(streamInfo); |
| | | String rtspUrl = streamContent.getFmp4(); // 取mp4地址 |
| | | String img = getImg(rtspUrl); |
| | | wvpResult.setData(img); |
| | | } else { |
| | | wvpResult.setCode(code); |
| | | wvpResult.setMsg(msg); |
| | | } |
| | | } else { |
| | | wvpResult.setCode(code); |
| | | wvpResult.setMsg(msg); |
| | | } |
| | | requestMessage.setData(wvpResult); |
| | | // 此处必须释放所有请求 |
| | | resultHolder.invokeAllResult(requestMessage); |
| | | }); |
| | | return result; |
| | | } |
| | | |
| | | StreamInfo streamInfo = redisCatchStorage.queryPlayByDevice(deviceId, channelId); |
| | | if (streamInfo == null) { |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "点播未找到"); |
| | | } |
| | | private String getImg(String rtspUrl) { |
| | | String imgUrl = null; |
| | | if (StringUtils.hasText(rtspUrl)) { |
| | | System.out.println("目标地址:" + rtspUrl); |
| | | FFmpegFrameGrabber grabber = null; |
| | | try { |
| | | grabber = new FFmpegFrameGrabber(rtspUrl); |
| | | // grabber.setOption("rtsp_transport", "tcp"); // 使用tcp的方式,不然会丢包很严重 |
| | | // grabber.setVideoOption("probesize", "10000"); // 设置捕获分析的最大字节 |
| | | // grabber.setFrameRate(25); |
| | | grabber.startUnsafe(); |
| | | Frame frame = grabber.grabImage(); // 直接捕获一帧 |
| | | // //转换图像 |
| | | // Java2DFrameConverter converter = new Java2DFrameConverter(); |
| | | // BufferedImage srcImage = converter.getBufferedImage(frame); |
| | | // if (srcImage!=null) { |
| | | // //创建文件 |
| | | // File file = new File(picPath); |
| | | // //输出文件 |
| | | // ImageIO.write(srcImage, "png", file); |
| | | // } |
| | | |
| | | try { |
| | | logger.warn("[停止点播] {}/{}", device.getDeviceId(), channelId); |
| | | cmder.streamByeCmd(device, channelId, streamInfo.getStream(), null, null); |
| | | } catch (InvalidArgumentException | SipException | ParseException | SsrcTransactionNotFoundException e) { |
| | | logger.error("[命令发送失败] 停止点播, 发送BYE: {}", e.getMessage()); |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "命令发送失败: " + e.getMessage()); |
| | | } |
| | | redisCatchStorage.stopPlay(streamInfo); |
| | | if (frame != null) { |
| | | System.out.println("成功捕获一帧"); |
| | | // 将Frame转换为Mat |
| | | OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat(); |
| | | Mat mat = converter.convertToMat(frame); |
| | | |
| | | storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId()); |
| | | JSONObject json = new JSONObject(); |
| | | json.put("deviceId", deviceId); |
| | | json.put("channelId", channelId); |
| | | return json; |
| | | } |
| | | imgUrl = format.format(new Date()) + "_" + IdUtils.randomUUID() + ".png"; |
| | | // 生成图片路径 |
| | | String imgPath = upload + imgUrl; |
| | | System.out.println("图片保存地址:" + imgPath); |
| | | imgUrl = "/profile/" + imgUrl; |
| | | // 保存图片 |
| | | opencv_imgcodecs.imwrite(imgPath, mat); |
| | | } else { |
| | | System.out.println("未捕获到帧"); |
| | | } |
| | | } catch (FrameGrabber.Exception e) { |
| | | e.printStackTrace(); |
| | | } finally { |
| | | if (grabber != null) { |
| | | try { |
| | | grabber.stop(); // 停止捕获 |
| | | } catch (FrameGrabber.Exception e) { |
| | | e.printStackTrace(); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | return imgUrl; |
| | | } |
| | | |
| | | /** |
| | | * 将不是h264的视频通过ffmpeg 转码为h264 + aac |
| | | * @param streamId 流ID |
| | | */ |
| | | @Operation(summary = "将不是h264的视频通过ffmpeg 转码为h264 + aac") |
| | | @Parameter(name = "streamId", description = "视频流ID", required = true) |
| | | @PostMapping("/convert/{streamId}") |
| | | public JSONObject playConvert(@PathVariable String streamId) { |
| | | StreamInfo streamInfo = redisCatchStorage.queryPlayByStreamId(streamId); |
| | | if (streamInfo == null) { |
| | | streamInfo = redisCatchStorage.queryPlayback(null, null, streamId, null); |
| | | } |
| | | if (streamInfo == null) { |
| | | logger.warn("视频转码API调用失败!, 视频流已经停止!"); |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "未找到视频流信息, 视频流可能已经停止"); |
| | | } |
| | | MediaServerItem mediaInfo = mediaServerService.getOne(streamInfo.getMediaServerId()); |
| | | JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(mediaInfo, streamId); |
| | | if (!rtpInfo.getBoolean("exist")) { |
| | | logger.warn("视频转码API调用失败!, 视频流已停止推流!"); |
| | | 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()); |
| | | if (jsonObject != null && jsonObject.getInteger("code") == 0) { |
| | | JSONObject data = jsonObject.getJSONObject("data"); |
| | | if (data != null) { |
| | | 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 { |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "转码失败"); |
| | | } |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 结束转码 |
| | | */ |
| | | @Operation(summary = "结束转码") |
| | | @Parameter(name = "key", description = "视频流key", required = true) |
| | | @Parameter(name = "mediaServerId", description = "流媒体服务ID", required = true) |
| | | @PostMapping("/convertStop/{key}") |
| | | public void playConvertStop(@PathVariable String key, String mediaServerId) { |
| | | if (mediaServerId == null) { |
| | | throw new ControllerException(ErrorCode.ERROR400.getCode(), "流媒体:" + mediaServerId + "不存在" ); |
| | | } |
| | | MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); |
| | | if (mediaInfo == null) { |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "使用的流媒体已经停止运行" ); |
| | | }else { |
| | | JSONObject jsonObject = zlmresTfulUtils.delFFmpegSource(mediaInfo, key); |
| | | logger.info(jsonObject.toJSONString()); |
| | | if (jsonObject != null && jsonObject.getInteger("code") == 0) { |
| | | JSONObject data = jsonObject.getJSONObject("data"); |
| | | if (data == null || data.getBoolean("flag") == null || !data.getBoolean("flag")) { |
| | | throw new ControllerException(ErrorCode.ERROR100 ); |
| | | } |
| | | }else { |
| | | throw new ControllerException(ErrorCode.ERROR100 ); |
| | | } |
| | | } |
| | | } |
| | | @Operation(summary = "停止点播", security = @SecurityRequirement(name = JwtUtils.HEADER)) |
| | | @Parameter(name = "deviceId", description = "设备国标编号", required = true) |
| | | @Parameter(name = "channelId", description = "通道国标编号", required = true) |
| | | @GetMapping("/stop/{deviceId}/{channelId}") |
| | | public JSONObject playStop(@PathVariable String deviceId, @PathVariable String channelId) { |
| | | |
| | | @Operation(summary = "语音广播命令") |
| | | @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}") |
| | | 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 + "]不存在"); |
| | | } |
| | | |
| | | playService.stopPlay(device, channelId); |
| | | JSONObject json = new JSONObject(); |
| | | json.put("deviceId", deviceId); |
| | | json.put("channelId", channelId); |
| | | return json; |
| | | } |
| | | |
| | | /** |
| | | * 结束转码 |
| | | */ |
| | | @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 void playConvertStop(@PathVariable String key, String mediaServerId) { |
| | | if (mediaServerId == null) { |
| | | throw new ControllerException(ErrorCode.ERROR400.getCode(), "流媒体:" + mediaServerId + "不存在"); |
| | | } |
| | | MediaServer mediaInfo = mediaServerService.getOne(mediaServerId); |
| | | if (mediaInfo == null) { |
| | | throw new ControllerException(ErrorCode.ERROR100.getCode(), "使用的流媒体已经停止运行"); |
| | | } else { |
| | | Boolean deleted = mediaServerService.delFFmpegSource(mediaInfo, key); |
| | | if (!deleted) { |
| | | throw new ControllerException(ErrorCode.ERROR100); |
| | | } |
| | | } |
| | | } |
| | | |
| | | @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); |
| | | } |
| | | 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); |
| | | return playService.audioBroadcast(device, channelId, broadcastMode); |
| | | |
| | | } |
| | | } |
| | | |
| | | @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调用"); |
| | | } |
| | | @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调用"); |
| | | } |
| | | // 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); |
| | | } |
| | | playService.stopAudioBroadcast(deviceId, channelId); |
| | | } |
| | | |
| | | @Operation(summary = "获取所有的ssrc") |
| | | @GetMapping("/ssrc") |
| | | public JSONObject getSSRC() { |
| | | if (logger.isDebugEnabled()) { |
| | | logger.debug("获取所有的ssrc"); |
| | | } |
| | | JSONArray objects = new JSONArray(); |
| | | List<SsrcTransaction> allSsrc = streamSession.getAllSsrc(); |
| | | for (SsrcTransaction transaction : allSsrc) { |
| | | JSONObject jsonObject = new JSONObject(); |
| | | jsonObject.put("deviceId", transaction.getDeviceId()); |
| | | jsonObject.put("channelId", transaction.getChannelId()); |
| | | jsonObject.put("ssrc", transaction.getSsrc()); |
| | | jsonObject.put("streamId", transaction.getStream()); |
| | | objects.add(jsonObject); |
| | | } |
| | | @Operation(summary = "获取所有的ssrc", security = @SecurityRequirement(name = JwtUtils.HEADER)) |
| | | @GetMapping("/ssrc") |
| | | public JSONObject getSSRC() { |
| | | if (logger.isDebugEnabled()) { |
| | | logger.debug("获取所有的ssrc"); |
| | | } |
| | | JSONArray objects = new JSONArray(); |
| | | List<SsrcTransaction> allSsrc = streamSession.getAllSsrc(); |
| | | for (SsrcTransaction transaction : allSsrc) { |
| | | JSONObject jsonObject = new JSONObject(); |
| | | jsonObject.put("deviceId", transaction.getDeviceId()); |
| | | jsonObject.put("channelId", transaction.getChannelId()); |
| | | jsonObject.put("ssrc", transaction.getSsrc()); |
| | | jsonObject.put("streamId", transaction.getStream()); |
| | | objects.add(jsonObject); |
| | | } |
| | | |
| | | JSONObject jsonObject = new JSONObject(); |
| | | jsonObject.put("data", objects); |
| | | jsonObject.put("count", objects.size()); |
| | | return jsonObject; |
| | | } |
| | | JSONObject jsonObject = new JSONObject(); |
| | | jsonObject.put("data", objects); |
| | | jsonObject.put("count", objects.size()); |
| | | 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; |
| | | } |
| | | |
| | | } |
| | | |