old mode 100644
new mode 100755
 |  |  | 
 |  |  | import com.genersoft.iot.vmp.conf.DynamicTask; | 
 |  |  | import com.genersoft.iot.vmp.conf.UserSetting; | 
 |  |  | import com.genersoft.iot.vmp.gb28181.bean.SendRtpItem; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMServerFactory; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.ZlmHttpHookSubscribe; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.dto.HookSubscribeFactory; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.dto.HookSubscribeForStreamChange; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; | 
 |  |  | import com.genersoft.iot.vmp.service.IMediaServerService; | 
 |  |  | import com.genersoft.iot.vmp.service.bean.*; | 
 |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
 |  |  | import com.genersoft.iot.vmp.utils.redis.RedisUtil; | 
 |  |  | import com.genersoft.iot.vmp.vmanager.bean.WVPResult; | 
 |  |  | import org.slf4j.Logger; | 
 |  |  | import org.slf4j.LoggerFactory; | 
 |  |  | 
 |  |  |  | 
 |  |  | import java.text.ParseException; | 
 |  |  | import java.util.HashMap; | 
 |  |  | import java.util.List; | 
 |  |  | import java.util.Map; | 
 |  |  | import java.util.UUID; | 
 |  |  | import java.util.concurrent.ConcurrentHashMap; | 
 |  |  | 
 |  |  |     private RedisTemplate<Object, Object> redisTemplate; | 
 |  |  |  | 
 |  |  |     @Autowired | 
 |  |  |     private ZLMRTPServerFactory zlmrtpServerFactory; | 
 |  |  |     private ZLMServerFactory zlmServerFactory; | 
 |  |  |  | 
 |  |  |     @Autowired | 
 |  |  |     private IMediaServerService mediaServerService; | 
 |  |  |  | 
 |  |  |     @Autowired | 
 |  |  |     private IRedisCatchStorage redisCatchStorage; | 
 |  |  |  | 
 |  |  |  | 
 |  |  |     @Autowired | 
 |  |  | 
 |  |  |                 while (!taskQueue.isEmpty()) { | 
 |  |  |                     Message msg = taskQueue.poll(); | 
 |  |  |                     try { | 
 |  |  |                         JSONObject msgJSON = JSON.parseObject(msg.getBody(), JSONObject.class); | 
 |  |  |                         WvpRedisMsg wvpRedisMsg = JSON.to(WvpRedisMsg.class, msgJSON); | 
 |  |  |                         WvpRedisMsg wvpRedisMsg = JSON.parseObject(msg.getBody(), WvpRedisMsg.class); | 
 |  |  |                         logger.info("[收到REDIS通知] 消息: {}", JSON.toJSONString(wvpRedisMsg)); | 
 |  |  |                         if (!userSetting.getServerId().equals(wvpRedisMsg.getToId())) { | 
 |  |  |                             continue; | 
 |  |  |                         } | 
 |  |  | 
 |  |  |  | 
 |  |  |                             switch (wvpRedisMsg.getCmd()){ | 
 |  |  |                                 case WvpRedisMsgCmd.GET_SEND_ITEM: | 
 |  |  |                                     RequestSendItemMsg content = JSON.to(RequestSendItemMsg.class, wvpRedisMsg.getContent()); | 
 |  |  |                                     RequestSendItemMsg content = JSON.parseObject(wvpRedisMsg.getContent(), RequestSendItemMsg.class); | 
 |  |  |                                     requestSendItemMsgHand(content, wvpRedisMsg.getFromId(), wvpRedisMsg.getSerial()); | 
 |  |  |                                     break; | 
 |  |  |                                 case WvpRedisMsgCmd.REQUEST_PUSH_STREAM: | 
 |  |  |                                     RequestPushStreamMsg param = JSON.to(RequestPushStreamMsg.class, wvpRedisMsg.getContent()); | 
 |  |  |                                     requestPushStreamMsgHand(param, wvpRedisMsg.getFromId(), wvpRedisMsg.getSerial()); | 
 |  |  |  | 
 |  |  |                                     break; | 
 |  |  |                                 default: | 
 |  |  |                                     break; | 
 |  |  | 
 |  |  |         param.put("pt", requestPushStreamMsg.getPt()); | 
 |  |  |         param.put("use_ps", requestPushStreamMsg.isPs() ? "1" : "0"); | 
 |  |  |         param.put("only_audio", requestPushStreamMsg.isOnlyAudio() ? "1" : "0"); | 
 |  |  |         JSONObject jsonObject = zlmrtpServerFactory.startSendRtpStream(mediaInfo, param); | 
 |  |  |         JSONObject jsonObject = zlmServerFactory.startSendRtpStream(mediaInfo, param); | 
 |  |  |         // 回复消息 | 
 |  |  |         responsePushStream(jsonObject, fromId, serial); | 
 |  |  |     } | 
 |  |  | 
 |  |  |         result.setData(content); | 
 |  |  |  | 
 |  |  |         WvpRedisMsg response = WvpRedisMsg.getResponseInstance(userSetting.getServerId(), toId, | 
 |  |  |                 WvpRedisMsgCmd.REQUEST_PUSH_STREAM, serial, result); | 
 |  |  |                 WvpRedisMsgCmd.REQUEST_PUSH_STREAM, serial, JSON.toJSONString(result)); | 
 |  |  |         JSONObject jsonObject = (JSONObject)JSON.toJSON(response); | 
 |  |  |         redisTemplate.convertAndSend(WVP_PUSH_STREAM_KEY, jsonObject); | 
 |  |  |     } | 
 |  |  | 
 |  |  |             result.setMsg("流媒体不存在"); | 
 |  |  |  | 
 |  |  |             WvpRedisMsg response = WvpRedisMsg.getResponseInstance(userSetting.getServerId(), toId, | 
 |  |  |                     WvpRedisMsgCmd.GET_SEND_ITEM, serial, result); | 
 |  |  |                     WvpRedisMsgCmd.GET_SEND_ITEM, serial, JSON.toJSONString(result)); | 
 |  |  |  | 
 |  |  |             JSONObject jsonObject = (JSONObject)JSON.toJSON(response); | 
 |  |  |             redisTemplate.convertAndSend(WVP_PUSH_STREAM_KEY, jsonObject); | 
 |  |  |             return; | 
 |  |  |         } | 
 |  |  |         // 确定流是否在线 | 
 |  |  |         boolean streamReady = zlmrtpServerFactory.isStreamReady(mediaServerItem, content.getApp(), content.getStream()); | 
 |  |  |         if (streamReady) { | 
 |  |  |         Boolean streamReady = zlmServerFactory.isStreamReady(mediaServerItem, content.getApp(), content.getStream()); | 
 |  |  |         if (streamReady != null && streamReady) { | 
 |  |  |             logger.info("[回复推流信息]  {}/{}", content.getApp(), content.getStream()); | 
 |  |  |             responseSendItem(mediaServerItem, content, toId, serial); | 
 |  |  |         }else { | 
 |  |  | 
 |  |  |                 WVPResult<SendRtpItem> result = new WVPResult<>(); | 
 |  |  |                 result.setCode(ERROR_CODE_TIMEOUT); | 
 |  |  |                 WvpRedisMsg response = WvpRedisMsg.getResponseInstance( | 
 |  |  |                         userSetting.getServerId(), toId, WvpRedisMsgCmd.GET_SEND_ITEM, serial, result | 
 |  |  |                         userSetting.getServerId(), toId, WvpRedisMsgCmd.GET_SEND_ITEM, serial, JSON.toJSONString(result) | 
 |  |  |                 ); | 
 |  |  |                 JSONObject jsonObject = (JSONObject)JSON.toJSON(response); | 
 |  |  |                 redisTemplate.convertAndSend(WVP_PUSH_STREAM_KEY, jsonObject); | 
 |  |  | 
 |  |  |             // 添加订阅 | 
 |  |  |             HookSubscribeForStreamChange hookSubscribe = HookSubscribeFactory.on_stream_changed(content.getApp(), content.getStream(), true, "rtsp", mediaServerItem.getId()); | 
 |  |  |  | 
 |  |  |             subscribe.addSubscribe(hookSubscribe, (MediaServerItem mediaServerItemInUse, JSONObject json)->{ | 
 |  |  |             subscribe.addSubscribe(hookSubscribe, (mediaServerItemInUse, hookParam)->{ | 
 |  |  |                         dynamicTask.stop(taskKey); | 
 |  |  |                         responseSendItem(mediaServerItem, content, toId, serial); | 
 |  |  |                     }); | 
 |  |  | 
 |  |  |             String key = VideoManagerConstants.VM_MSG_STREAM_PUSH_REQUESTED; | 
 |  |  |             logger.info("[redis发送通知] 推流被请求 {}: {}/{}", key, messageForPushChannel.getApp(), messageForPushChannel.getStream()); | 
 |  |  |             redisTemplate.convertAndSend(key, JSON.toJSON(messageForPushChannel)); | 
 |  |  |  | 
 |  |  | //            redisCatchStorage.sendStreamPushRequestedMsg(messageForPushChannel); | 
 |  |  |  | 
 |  |  |         } | 
 |  |  |     } | 
 |  |  |  | 
 |  |  | 
 |  |  |      * 将获取到的sendItem发送出去 | 
 |  |  |      */ | 
 |  |  |     private void responseSendItem(MediaServerItem mediaServerItem, RequestSendItemMsg content, String toId, String serial) { | 
 |  |  |         SendRtpItem sendRtpItem = zlmrtpServerFactory.createSendRtpItem(mediaServerItem, content.getIp(), | 
 |  |  |         SendRtpItem sendRtpItem = zlmServerFactory.createSendRtpItem(mediaServerItem, content.getIp(), | 
 |  |  |                 content.getPort(), content.getSsrc(), content.getPlatformId(), | 
 |  |  |                 content.getApp(), content.getStream(), content.getChannelId(), | 
 |  |  |                 content.getTcp(), content.getRtcp()); | 
 |  |  | 
 |  |  |         responseSendItemMsg.setSendRtpItem(sendRtpItem); | 
 |  |  |         responseSendItemMsg.setMediaServerItem(mediaServerItem); | 
 |  |  |         result.setData(responseSendItemMsg); | 
 |  |  |         redisCatchStorage.updateSendRTPSever(sendRtpItem); | 
 |  |  |  | 
 |  |  |         WvpRedisMsg response = WvpRedisMsg.getResponseInstance( | 
 |  |  |                 userSetting.getServerId(), toId, WvpRedisMsgCmd.GET_SEND_ITEM, serial, result | 
 |  |  |                 userSetting.getServerId(), toId, WvpRedisMsgCmd.GET_SEND_ITEM, serial, JSON.toJSONString(result) | 
 |  |  |         ); | 
 |  |  |         JSONObject jsonObject = (JSONObject)JSON.toJSON(response); | 
 |  |  |         redisTemplate.convertAndSend(WVP_PUSH_STREAM_KEY, jsonObject); | 
 |  |  | 
 |  |  |         requestSendItemMsg.setServerId(serverId); | 
 |  |  |         String key = UUID.randomUUID().toString(); | 
 |  |  |         WvpRedisMsg redisMsg = WvpRedisMsg.getRequestInstance(userSetting.getServerId(), serverId, WvpRedisMsgCmd.GET_SEND_ITEM, | 
 |  |  |                 key, requestSendItemMsg); | 
 |  |  |                 key, JSON.toJSONString(requestSendItemMsg)); | 
 |  |  |  | 
 |  |  |         JSONObject jsonObject = (JSONObject)JSON.toJSON(redisMsg); | 
 |  |  |         logger.info("[请求推流SendItem] {}: {}", serverId, jsonObject); | 
 |  |  | 
 |  |  |     public void sendMsgForStartSendRtpStream(String serverId, RequestPushStreamMsg param, PlayMsgCallbackForStartSendRtpStream callback) { | 
 |  |  |         String key = UUID.randomUUID().toString(); | 
 |  |  |         WvpRedisMsg redisMsg = WvpRedisMsg.getRequestInstance(userSetting.getServerId(), serverId, | 
 |  |  |                 WvpRedisMsgCmd.REQUEST_PUSH_STREAM, key, param); | 
 |  |  |                 WvpRedisMsgCmd.REQUEST_PUSH_STREAM, key, JSON.toJSONString(param)); | 
 |  |  |  | 
 |  |  |         JSONObject jsonObject = (JSONObject)JSON.toJSON(redisMsg); | 
 |  |  |         logger.info("[REDIS 请求其他平台推流] {}: {}", serverId, jsonObject); | 
 |  |  | 
 |  |  |         }); | 
 |  |  |         redisTemplate.convertAndSend(WVP_PUSH_STREAM_KEY, jsonObject); | 
 |  |  |     } | 
 |  |  |  | 
 |  |  |     private SendRtpItem querySendRTPServer(String platformGbId, String channelId, String streamId, String callId) { | 
 |  |  |         if (platformGbId == null) { | 
 |  |  |             platformGbId = "*"; | 
 |  |  |         } | 
 |  |  |         if (channelId == null) { | 
 |  |  |             channelId = "*"; | 
 |  |  |         } | 
 |  |  |         if (streamId == null) { | 
 |  |  |             streamId = "*"; | 
 |  |  |         } | 
 |  |  |         if (callId == null) { | 
 |  |  |             callId = "*"; | 
 |  |  |         } | 
 |  |  |         String key = VideoManagerConstants.PLATFORM_SEND_RTP_INFO_PREFIX | 
 |  |  |                 + userSetting.getServerId() + "_*_" | 
 |  |  |                 + platformGbId + "_" | 
 |  |  |                 + channelId + "_" | 
 |  |  |                 + streamId + "_" | 
 |  |  |                 + callId; | 
 |  |  |         List<Object> scan = RedisUtil.scan(redisTemplate, key); | 
 |  |  |         if (scan.size() > 0) { | 
 |  |  |             return (SendRtpItem)redisTemplate.opsForValue().get(scan.get(0)); | 
 |  |  |         }else { | 
 |  |  |             return null; | 
 |  |  |         } | 
 |  |  |     } | 
 |  |  | } |