|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.alibaba.fastjson.JSONArray; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.StreamInfo; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.SsrcTransaction; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig; | 
|---|
|  |  |  | 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.media.zlm.ZLMRESTfulUtils; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.IMediaServerItem; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.IMediaServerService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
|---|
|  |  |  | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorager; | 
|---|
|  |  |  | import org.springframework.web.context.request.async.DeferredResult; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import java.util.Enumeration; | 
|---|
|  |  |  | import java.util.Map; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.UUID; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import javax.sip.message.Response; | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 获取可用的zlm | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | IMediaServerItem newMediaServerItem = playService.getNewMediaServerItem(device); | 
|---|
|  |  |  | MediaServerItem newMediaServerItem = playService.getNewMediaServerItem(device); | 
|---|
|  |  |  | PlayResult playResult = playService.play(newMediaServerItem, deviceId, channelId, null, null); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return playResult.getResult(); | 
|---|
|  |  |  | 
|---|
|  |  |  | @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 )); | 
|---|
|  |  |  | logger.debug(String.format("设备预览/回放停止API调用,streamId:%s_%s", deviceId, channelId )); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | UUID uuid = UUID.randomUUID(); | 
|---|
|  |  |  | String uuid = UUID.randomUUID().toString(); | 
|---|
|  |  |  | DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 录像查询以channelId作为deviceId查询 | 
|---|
|  |  |  | resultHolder.put(DeferredResultHolder.CALLBACK_CMD_STOP + uuid, result); | 
|---|
|  |  |  | String key = DeferredResultHolder.CALLBACK_CMD_STOP + deviceId + channelId; | 
|---|
|  |  |  | resultHolder.put(key, 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.setId(uuid); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setData("点播未找到"); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | storager.stopPlay(deviceId, channelId); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | redisCatchStorage.stopPlay(streamInfo); | 
|---|
|  |  |  | storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId()); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_STOP + uuid); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | //Response response = event.getResponse(); | 
|---|
|  |  |  | msg.setData(String.format("success")); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | mediaServerService.closeRTPServer(device, channelId); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | 
|---|
|  |  |  | json.put("deviceId", deviceId); | 
|---|
|  |  |  | json.put("channelId", channelId); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setData(json.toString()); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | logger.warn("设备预览/回放停止API调用失败!"); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setData("streamId null"); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 超时处理 | 
|---|
|  |  |  | result.onTimeout(()->{ | 
|---|
|  |  |  | logger.warn(String.format("设备预览/回放停止超时,deviceId/channelId:%s/$s ", deviceId, channelId)); | 
|---|
|  |  |  | logger.warn(String.format("设备预览/回放停止超时,deviceId/channelId:%s_%s ", deviceId, channelId)); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_STOP + uuid); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setData("Timeout"); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @PostMapping("/convert/{streamId}") | 
|---|
|  |  |  | public ResponseEntity<String> playConvert(@PathVariable String streamId) { | 
|---|
|  |  |  | StreamInfo streamInfo = streamId.startsWith("gb_play_") ? redisCatchStorage.queryPlayByStreamId(streamId) : redisCatchStorage.queryPlaybackByStreamId(streamId); | 
|---|
|  |  |  | StreamInfo streamInfo = redisCatchStorage.queryPlayByStreamId(streamId); | 
|---|
|  |  |  | if (streamInfo == null) { | 
|---|
|  |  |  | streamInfo = redisCatchStorage.queryPlaybackByStreamId(streamId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (streamInfo == null) { | 
|---|
|  |  |  | logger.warn("视频转码API调用失败!, 视频流已经停止!"); | 
|---|
|  |  |  | return new ResponseEntity<String>("未找到视频流信息, 视频流可能已经停止", HttpStatus.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | IMediaServerItem mediaInfo = mediaServerService.getOne(streamInfo.getMediaServerId()); | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(streamInfo.getMediaServerId()); | 
|---|
|  |  |  | JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(mediaInfo, streamId); | 
|---|
|  |  |  | if (!rtpInfo.getBoolean("exist")) { | 
|---|
|  |  |  | logger.warn("视频转码API调用失败!, 视频流已停止推流!"); | 
|---|
|  |  |  | 
|---|
|  |  |  | result.put("msg", "mediaServerId is null"); | 
|---|
|  |  |  | return new ResponseEntity<String>( result.toJSONString(), HttpStatus.BAD_REQUEST); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | IMediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  | if (mediaInfo == null) { | 
|---|
|  |  |  | result.put("code", 0); | 
|---|
|  |  |  | result.put("msg", "使用的流媒体已经停止运行"); | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(3 * 1000L); | 
|---|
|  |  |  | String key  = DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId; | 
|---|
|  |  |  | if (resultHolder.exist(key, null)) { | 
|---|
|  |  |  | result.setResult(new ResponseEntity<>("设备使用中",HttpStatus.OK)); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | String uuid  = UUID.randomUUID().toString(); | 
|---|
|  |  |  | if (device == null) { | 
|---|
|  |  |  | resultHolder.put(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId, result); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | resultHolder.put(key, key,  result); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("DeviceID", deviceId); | 
|---|
|  |  |  | json.put("CmdType", "Broadcast"); | 
|---|
|  |  |  | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | cmder.audioBroadcastCmd(device, (event) -> { | 
|---|
|  |  |  | Response response = event.getResponse(); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | 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())); | 
|---|
|  |  |  | json.put("Description", String.format("语音广播操作失败,错误码: %s, %s", event.statusCode, event.msg)); | 
|---|
|  |  |  | msg.setData(json); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | 
|---|
|  |  |  | result.onTimeout(() -> { | 
|---|
|  |  |  | logger.warn(String.format("语音广播操作超时, 设备未返回应答指令")); | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setId(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("DeviceID", deviceId); | 
|---|
|  |  |  | json.put("CmdType", "Broadcast"); | 
|---|
|  |  |  | 
|---|
|  |  |  | msg.setData(json); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | resultHolder.put(DeferredResultHolder.CALLBACK_CMD_BROADCAST + deviceId, result); | 
|---|
|  |  |  | resultHolder.put(key, uuid, result); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | logger.debug("获取所有的ssrc"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | JSONArray objects = new JSONArray(); | 
|---|
|  |  |  | for(Map.Entry<String, String> entry: streamSession.getSsrcMap().entrySet()) { | 
|---|
|  |  |  | System.out.println("Key = " + entry.getKey() + ", Value = " + entry.getValue()); | 
|---|
|  |  |  | List<SsrcTransaction> allSsrc = streamSession.getAllSsrc(); | 
|---|
|  |  |  | for (SsrcTransaction transaction : allSsrc) { | 
|---|
|  |  |  | JSONObject jsonObject = new JSONObject(); | 
|---|
|  |  |  | String[] keyArray = entry.getKey().split("_"); | 
|---|
|  |  |  | jsonObject.put("deviceId", keyArray[0]); | 
|---|
|  |  |  | jsonObject.put("channelId", keyArray[1]); | 
|---|
|  |  |  | jsonObject.put("ssrc", entry.getValue()); | 
|---|
|  |  |  | jsonObject.put("streamId", streamSession.getStreamIdMap().get(entry.getKey())); | 
|---|
|  |  |  | jsonObject.put("deviceId", transaction.getDeviceId()); | 
|---|
|  |  |  | jsonObject.put("channelId", transaction.getChannelId()); | 
|---|
|  |  |  | jsonObject.put("ssrc", transaction.getSsrc()); | 
|---|
|  |  |  | jsonObject.put("streamId", transaction.getStreamId()); | 
|---|
|  |  |  | objects.add(jsonObject); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | WVPResult<JSONObject> result = new WVPResult<>(); | 
|---|
|  |  |  | result.setCode(0); | 
|---|
|  |  |  | result.setMsg("success"); | 
|---|