jshqhxm
2022-10-09 c18dd55c742a9d01bfa55217708cc9d67039dee6
src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMHttpHookListener.java
@@ -1,5 +1,6 @@
package com.genersoft.iot.vmp.media.zlm;
import java.text.ParseException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -7,6 +8,7 @@
import com.alibaba.fastjson.JSON;
import com.genersoft.iot.vmp.common.StreamInfo;
import com.genersoft.iot.vmp.conf.UserSetting;
import com.genersoft.iot.vmp.conf.exception.SsrcTransactionNotFoundException;
import com.genersoft.iot.vmp.gb28181.bean.*;
import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
@@ -19,8 +21,6 @@
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.PostMapping;
@@ -33,6 +33,8 @@
import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
import javax.servlet.http.HttpServletRequest;
import javax.sip.InvalidArgumentException;
import javax.sip.SipException;
/**    
 * @description:针对 ZLMediaServer的hook事件监听
@@ -59,6 +61,9 @@
   @Autowired
   private IRedisCatchStorage redisCatchStorage;
   @Autowired
   private IDeviceService deviceService;
   @Autowired
   private IMediaServerService mediaServerService;
@@ -105,7 +110,7 @@
   @PostMapping(value = "/on_server_keepalive", produces = "application/json;charset=UTF-8")
   public JSONObject onServerKeepalive(@RequestBody JSONObject json){
      logger.info("[ ZLM HOOK ] on_server_keepalive API调用,参数:" + json.toString());
      logger.info("[ ZLM HOOK ]on_server_keepalive API调用,参数:" + json.toString());
      String mediaServerId = json.getString("mediaServerId");
      List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_keepalive);
      if (subscribes != null  && subscribes.size() > 0) {
@@ -419,10 +424,11 @@
      String schema = item.getSchema();
      List<MediaItem.MediaTrack> tracks = item.getTracks();
      boolean regist = item.isRegist();
      if (item.getOriginType() == OriginType.RTMP_PUSH.ordinal()
            || item.getOriginType() == OriginType.RTSP_PUSH.ordinal()
            || item.getOriginType() == OriginType.RTC_PUSH.ordinal()) {
         if (regist) {
      if (regist) {
         if (item.getOriginType() == OriginType.RTMP_PUSH.ordinal()
               || item.getOriginType() == OriginType.RTSP_PUSH.ordinal()
               || item.getOriginType() == OriginType.RTC_PUSH.ordinal()) {
            StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(app, stream);
            if (streamAuthorityInfo == null) {
               streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(item);
@@ -431,9 +437,9 @@
               streamAuthorityInfo.setOriginTypeStr(item.getOriginTypeStr());
            }
            redisCatchStorage.updateStreamAuthorityInfo(app, stream, streamAuthorityInfo);
         }else {
            redisCatchStorage.removeStreamAuthorityInfo(app, stream);
         }
      }else {
         redisCatchStorage.removeStreamAuthorityInfo(app, stream);
      }
      if ("rtsp".equals(schema)){
@@ -453,15 +459,12 @@
            if (streamInfo!=null){
               redisCatchStorage.stopPlay(streamInfo);
               storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId());
               // 如果正在给上级推送,则发送bye
            }else{
               streamInfo = redisCatchStorage.queryPlayback(null, null, stream, null);
               if (streamInfo != null) {
                  redisCatchStorage.stopPlayback(streamInfo.getDeviceID(), streamInfo.getChannelId(),
                        streamInfo.getStream(), null);
               }
               // 如果正在给上级推送,则发送bye
            }
         }else {
            if (!"rtp".equals(app)){
@@ -511,6 +514,28 @@
               }
            }
         }
         if (!regist) {
            List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(stream);
            if (sendRtpItems.size() > 0) {
               for (SendRtpItem sendRtpItem : sendRtpItems) {
                  if (sendRtpItem.getApp().equals(app)) {
                     String platformId = sendRtpItem.getPlatformId();
                     ParentPlatform platform = storager.queryParentPlatByServerGBId(platformId);
                     Device device = deviceService.queryDevice(platformId);
                     try {
                        if (platform != null) {
                           commanderFroPlatform.streamByeCmd(platform, sendRtpItem);
                        }else {
                           cmder.streamByeCmd(device, sendRtpItem.getChannelId(), stream, sendRtpItem.getCallId());
                        }
                     } catch (SipException | InvalidArgumentException | ParseException | SsrcTransactionNotFoundException e) {
                        logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
                     }
                  }
               }
            }
         }
      }
      JSONObject ret = new JSONObject();
@@ -543,21 +568,48 @@
               if (sendRtpItems.size() > 0) {
                  for (SendRtpItem sendRtpItem : sendRtpItems) {
                     ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId());
                     commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
                     try {
                        commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
                     } catch (SipException | InvalidArgumentException | ParseException e) {
                        logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
                     }
                     redisCatchStorage.deleteSendRTPServer(parentPlatform.getServerGBId(), sendRtpItem.getChannelId(),
                           sendRtpItem.getCallId(), sendRtpItem.getStreamId());
                  }
               }
            }
            cmder.streamByeCmd(streamInfoForPlayCatch.getDeviceID(), streamInfoForPlayCatch.getChannelId(),
                  streamInfoForPlayCatch.getStream(), null);
            Device device = deviceService.queryDevice(streamInfoForPlayCatch.getDeviceID());
            if (device != null) {
               try {
                  cmder.streamByeCmd(device, streamInfoForPlayCatch.getChannelId(),
                        streamInfoForPlayCatch.getStream(), null);
               } catch (InvalidArgumentException | ParseException | SipException | SsrcTransactionNotFoundException e) {
                  logger.error("[无人观看]点播, 发送BYE失败 {}", e.getMessage());
               }
            }
            redisCatchStorage.stopPlay(streamInfoForPlayCatch);
            storager.stopPlay(streamInfoForPlayCatch.getDeviceID(), streamInfoForPlayCatch.getChannelId());
         }else{
            StreamInfo streamInfoForPlayBackCatch = redisCatchStorage.queryPlayback(null, null, streamId, null);
            if (streamInfoForPlayBackCatch != null) {
               cmder.streamByeCmd(streamInfoForPlayBackCatch.getDeviceID(),
                     streamInfoForPlayBackCatch.getChannelId(), streamInfoForPlayBackCatch.getStream(), null);
               redisCatchStorage.stopPlayback(streamInfoForPlayBackCatch.getDeviceID(),
                     streamInfoForPlayBackCatch.getChannelId(), streamInfoForPlayBackCatch.getStream(), null);
            if (streamInfoForPlayBackCatch != null ) {
               if (streamInfoForPlayBackCatch.isPause()) {
                  ret.put("close", false);
               }else {
                  Device device = deviceService.queryDevice(streamInfoForPlayBackCatch.getDeviceID());
                  if (device != null) {
                     try {
                        cmder.streamByeCmd(device,streamInfoForPlayBackCatch.getChannelId(),
                              streamInfoForPlayBackCatch.getStream(), null);
                     } catch (InvalidArgumentException | ParseException | SipException |
                            SsrcTransactionNotFoundException e) {
                        logger.error("[无人观看]回放, 发送BYE失败 {}", e.getMessage());
                     }
                  }
                  redisCatchStorage.stopPlayback(streamInfoForPlayBackCatch.getDeviceID(),
                        streamInfoForPlayBackCatch.getChannelId(), streamInfoForPlayBackCatch.getStream(), null);
               }
            }else {
               StreamInfo streamInfoForDownload = redisCatchStorage.queryDownload(null, null, streamId, null);
               // 进行录像下载时无人观看不断流
@@ -573,13 +625,19 @@
         return ret;
      }else {
         StreamProxyItem streamProxyItem = streamProxyService.getStreamProxyByAppAndStream(app, streamId);
         if (streamProxyItem != null && streamProxyItem.isEnable_remove_none_reader()) {
            ret.put("close", true);
            streamProxyService.del(app, streamId);
            String url = streamProxyItem.getUrl() != null?streamProxyItem.getUrl():streamProxyItem.getSrc_url();
            logger.info("[{}/{}]<-[{}] 拉流代理无人观看已经移除",  app, streamId, url);
         }else {
            ret.put("close", false);
         if (streamProxyItem != null ) {
            if (streamProxyItem.isEnable_remove_none_reader()) {
               // 无人观看自动移除
               ret.put("close", true);
               streamProxyService.del(app, streamId);
               String url = streamProxyItem.getUrl() != null?streamProxyItem.getUrl():streamProxyItem.getSrc_url();
               logger.info("[{}/{}]<-[{}] 拉流代理无人观看已经移除",  app, streamId, url);
            }else if (streamProxyItem.isEnable_disable_none_reader()) {
               // 无人观看停用
               ret.put("close", true);
            }else {
               ret.put("close", false);
            }
         }
         return ret;
      }
@@ -626,7 +684,7 @@
   @ResponseBody
   @PostMapping(value = "/on_server_started", produces = "application/json;charset=UTF-8")
   public JSONObject onServerStarted(HttpServletRequest request, @RequestBody JSONObject jsonObject){
      if (logger.isDebugEnabled()) {
         logger.debug("[ ZLM HOOK ]on_server_started API调用,参数:" + jsonObject.toString());
      }
@@ -649,6 +707,43 @@
      return ret;
   }
   /**
    * 发送rtp(startSendRtp)被动关闭时回调
    */
   @ResponseBody
   @PostMapping(value = "/on_send_rtp_stopped", produces = "application/json;charset=UTF-8")
   public JSONObject onSendRtpStopped(HttpServletRequest request, @RequestBody JSONObject jsonObject){
      logger.info("[ ZLM HOOK ]on_send_rtp_stopped API调用,参数:" + jsonObject);
      JSONObject ret = new JSONObject();
      ret.put("code", 0);
      ret.put("msg", "success");
      // 查找对应的上级推流,发送停止
      String app = jsonObject.getString("app");
      if (!"rtp".equals(app)) {
         return ret;
      }
      String stream = jsonObject.getString("stream");
      List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(stream);
      if (sendRtpItems.size() > 0) {
         for (SendRtpItem sendRtpItem : sendRtpItems) {
            ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId());
            try {
               commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
            } catch (SipException | InvalidArgumentException | ParseException e) {
               logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
            }
            redisCatchStorage.deleteSendRTPServer(parentPlatform.getServerGBId(), sendRtpItem.getChannelId(),
                  sendRtpItem.getCallId(), sendRtpItem.getStreamId());
         }
      }
      return ret;
   }
   private Map<String, String> urlParamToMap(String params) {
      HashMap<String, String> map = new HashMap<>();
      if (ObjectUtils.isEmpty(params)) {