old mode 100644
new mode 100755
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSON; | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSONObject; | 
|---|
|  |  |  | 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.SsrcTransactionNotFoundException; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.*; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.event.EventPublisher; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.AudioBroadcastManager; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.event.subscribe.catalog.CatalogEvent; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.SSRCFactory; | 
|---|
|  |  |  | 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.ISIPCommanderForPlatform; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.HookType; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.StreamAuthorityInfo; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.bean.MediaServer; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.bean.ResultForOnPublish; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.event.hook.HookSubscribe; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.event.media.*; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.event.mediaServer.MediaSendRtpStoppedEvent; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.service.IMediaServerService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.ZLMServerConfig; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.hook.*; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.event.HookZlmServerKeepaliveEvent; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.event.HookZlmServerStartEvent; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.*; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.bean.MessageForPushChannel; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.redisMsg.IRedisRpcService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.vmanager.bean.ErrorCode; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.vmanager.bean.StreamContent; | 
|---|
|  |  |  | import org.slf4j.Logger; | 
|---|
|  |  |  | import org.slf4j.LoggerFactory; | 
|---|
|  |  |  | import org.springframework.beans.factory.annotation.Autowired; | 
|---|
|  |  |  | import org.springframework.beans.factory.annotation.Qualifier; | 
|---|
|  |  |  | import org.springframework.context.ApplicationEventPublisher; | 
|---|
|  |  |  | import org.springframework.data.redis.core.RedisTemplate; | 
|---|
|  |  |  | import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; | 
|---|
|  |  |  | import org.springframework.util.ObjectUtils; | 
|---|
|  |  |  | 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.text.ParseException; | 
|---|
|  |  |  | import java.util.HashMap; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.Map; | 
|---|
|  |  |  | import java.util.UUID; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @description:针对 ZLMediaServer的hook事件监听 | 
|---|
|  |  |  | 
|---|
|  |  |  | private AudioBroadcastManager audioBroadcastManager; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ZLMRTPServerFactory zlmrtpServerFactory; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IPlayService playService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IRedisCatchStorage redisCatchStorage; | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IRedisRpcService redisRpcService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IInviteStreamService inviteStreamService; | 
|---|
|  |  |  | 
|---|
|  |  |  | private EventPublisher eventPublisher; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ZLMMediaListManager zlmMediaListManager; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ZlmHttpHookSubscribe subscribe; | 
|---|
|  |  |  | private HookSubscribe subscribe; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private UserSetting userSetting; | 
|---|
|  |  |  | 
|---|
|  |  |  | private IUserService userService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private VideoStreamSessionManager sessionManager; | 
|---|
|  |  |  | private ICloudRecordService cloudRecordService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private AssistRESTfulUtils assistRESTfulUtils; | 
|---|
|  |  |  | private VideoStreamSessionManager sessionManager; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private SSRCFactory ssrcFactory; | 
|---|
|  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ThreadPoolTaskExecutor taskExecutor; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private RedisTemplate<Object, Object> redisTemplate; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ApplicationEventPublisher applicationEventPublisher; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IStreamPushService streamPushService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 服务器定时上报时间,上报间隔可配置,默认10s上报一次 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ResponseBody | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @PostMapping(value = "/on_server_keepalive", produces = "application/json;charset=UTF-8") | 
|---|
|  |  |  | public HookResult onServerKeepalive(@RequestBody OnServerKeepaliveHookParam param) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_keepalive); | 
|---|
|  |  |  | if (subscribes != null && subscribes.size() > 0) { | 
|---|
|  |  |  | for (ZlmHttpHookSubscribe.Event subscribe : subscribes) { | 
|---|
|  |  |  | subscribe.response(null, param); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | HookZlmServerKeepaliveEvent event = new HookZlmServerKeepaliveEvent(this); | 
|---|
|  |  |  | MediaServer mediaServerItem = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (mediaServerItem != null) { | 
|---|
|  |  |  | event.setMediaServerItem(mediaServerItem); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(event); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | mediaServerService.updateMediaServerKeepalive(param.getMediaServerId(), param.getData()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | }catch (Exception e) { | 
|---|
|  |  |  | logger.info("[ZLM-HOOK-心跳] 发送通知失败 ", e); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | * 播放器鉴权事件,rtsp/rtmp/http-flv/ws-flv/hls的播放都将触发此鉴权事件。 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ResponseBody | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @PostMapping(value = "/on_play", produces = "application/json;charset=UTF-8") | 
|---|
|  |  |  | public HookResult onPlay(@RequestBody OnPlayHookParam param) { | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("[ZLM HOOK] 播放鉴权:{}->{}" + param.getMediaServerId(), param); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | String mediaServerId = param.getMediaServerId(); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | JSONObject json = (JSONObject) JSON.toJSON(param); | 
|---|
|  |  |  | ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_play, json); | 
|---|
|  |  |  | if (subscribe != null) { | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  | if (mediaInfo != null) { | 
|---|
|  |  |  | subscribe.response(mediaInfo, param); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | if (!"rtp".equals(param.getApp())) { | 
|---|
|  |  |  | Map<String, String> paramMap = urlParamToMap(param.getParams()); | 
|---|
|  |  |  | StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (streamAuthorityInfo != null && streamAuthorityInfo.getCallId() != null && !streamAuthorityInfo.getCallId().equals(paramMap.get("callId"))) { | 
|---|
|  |  |  | return new HookResult(401, "Unauthorized"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Map<String, String> paramMap = urlParamToMap(param.getParams()); | 
|---|
|  |  |  | // 对于播放流进行鉴权 | 
|---|
|  |  |  | boolean authenticateResult = mediaService.authenticatePlay(param.getApp(), param.getStream(), paramMap.get("callId")); | 
|---|
|  |  |  | if (!authenticateResult) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 播放鉴权 失败:{}->{}", param.getMediaServerId(), param); | 
|---|
|  |  |  | return new HookResult(401, "Unauthorized"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 播放鉴权成功:{}->{}", param.getMediaServerId(), param); | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | JSONObject json = (JSONObject) JSON.toJSON(param); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | logger.info("[ZLM HOOK]推流鉴权:{}->{}", param.getMediaServerId(), param); | 
|---|
|  |  |  | // TODO 加快处理速度 | 
|---|
|  |  |  |  | 
|---|
|  |  |  | String mediaServerId = json.getString("mediaServerId"); | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (!"rtp".equals(param.getApp())) { | 
|---|
|  |  |  | if (userSetting.getPushAuthority()) { | 
|---|
|  |  |  | // 推流鉴权 | 
|---|
|  |  |  | if (param.getParams() == null) { | 
|---|
|  |  |  | logger.info("推流鉴权失败: 缺少必要参数:sign=md5(user表的pushKey)"); | 
|---|
|  |  |  | return new HookResultForOnPublish(401, "Unauthorized"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Map<String, String> paramMap = urlParamToMap(param.getParams()); | 
|---|
|  |  |  | String sign = paramMap.get("sign"); | 
|---|
|  |  |  | if (sign == null) { | 
|---|
|  |  |  | logger.info("推流鉴权失败: 缺少必要参数:sign=md5(user表的pushKey)"); | 
|---|
|  |  |  | return new HookResultForOnPublish(401, "Unauthorized"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 推流自定义播放鉴权码 | 
|---|
|  |  |  | String callId = paramMap.get("callId"); | 
|---|
|  |  |  | // 鉴权配置 | 
|---|
|  |  |  | boolean hasAuthority = userService.checkPushAuthority(callId, sign); | 
|---|
|  |  |  | if (!hasAuthority) { | 
|---|
|  |  |  | logger.info("推流鉴权失败: sign 无权限: callId={}. sign={}", callId, sign); | 
|---|
|  |  |  | return new HookResultForOnPublish(401, "Unauthorized"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | StreamAuthorityInfo streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(param); | 
|---|
|  |  |  | streamAuthorityInfo.setCallId(callId); | 
|---|
|  |  |  | streamAuthorityInfo.setSign(sign); | 
|---|
|  |  |  | // 鉴权通过 | 
|---|
|  |  |  | redisCatchStorage.updateStreamAuthorityInfo(param.getApp(), param.getStream(), streamAuthorityInfo); | 
|---|
|  |  |  | // 通知assist新的callId | 
|---|
|  |  |  | if (mediaInfo != null && mediaInfo.getRecordAssistPort() > 0) { | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | assistRESTfulUtils.addStreamCallInfo(mediaInfo, param.getApp(), param.getStream(), callId, null); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | zlmMediaListManager.sendStreamEvent(param.getApp(), param.getStream(), param.getMediaServerId()); | 
|---|
|  |  |  | MediaServer mediaServer = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  | if (mediaServer == null) { | 
|---|
|  |  |  | return new HookResultForOnPublish(0, "success"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | HookResultForOnPublish result = HookResultForOnPublish.SUCCESS(); | 
|---|
|  |  |  | if (!"rtp".equals(param.getApp())) { | 
|---|
|  |  |  | result.setEnable_audio(true); | 
|---|
|  |  |  | ResultForOnPublish resultForOnPublish = mediaService.authenticatePublish(mediaServer, param.getApp(), param.getStream(), param.getParams()); | 
|---|
|  |  |  | if (resultForOnPublish != null) { | 
|---|
|  |  |  | HookResultForOnPublish successResult = HookResultForOnPublish.getInstance(resultForOnPublish); | 
|---|
|  |  |  | logger.info("[ZLM HOOK]推流鉴权 响应:{}->{}->>>>{}", param.getMediaServerId(), param, successResult); | 
|---|
|  |  |  | return successResult; | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | HookResultForOnPublish fail = HookResultForOnPublish.Fail(); | 
|---|
|  |  |  | logger.info("[ZLM HOOK]推流鉴权 响应:{}->{}->>>>{}", param.getMediaServerId(), param, fail); | 
|---|
|  |  |  | return fail; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_publish, json); | 
|---|
|  |  |  | if (subscribe != null) { | 
|---|
|  |  |  | if (mediaInfo != null) { | 
|---|
|  |  |  | subscribe.response(mediaInfo, param); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | new HookResultForOnPublish(1, "zlm not register"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if ("rtp".equals(param.getApp())) { | 
|---|
|  |  |  | result.setEnable_mp4(userSetting.getRecordSip()); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | result.setEnable_mp4(userSetting.isRecordPushLive()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | List<SsrcTransaction> ssrcTransactionForAll = sessionManager.getSsrcTransactionForAll(null, null, null, param.getStream()); | 
|---|
|  |  |  | if (ssrcTransactionForAll != null && ssrcTransactionForAll.size() == 1) { | 
|---|
|  |  |  | String deviceId = ssrcTransactionForAll.get(0).getDeviceId(); | 
|---|
|  |  |  | String channelId = ssrcTransactionForAll.get(0).getChannelId(); | 
|---|
|  |  |  | DeviceChannel deviceChannel = storager.queryChannel(deviceId, channelId); | 
|---|
|  |  |  | if (deviceChannel != null) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | result.setEnable_audio(deviceChannel.isHasAudio()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 如果是录像下载就设置视频间隔十秒 | 
|---|
|  |  |  | if (ssrcTransactionForAll.get(0).getType() == InviteSessionType.DOWNLOAD) { | 
|---|
|  |  |  | result.setMp4_max_second(10); | 
|---|
|  |  |  | result.setEnable_audio(true); | 
|---|
|  |  |  | result.setEnable_mp4(true); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 如果是talk对讲,则默认获取声音 | 
|---|
|  |  |  | if (ssrcTransactionForAll.get(0).getType() == InviteSessionType.TALK) { | 
|---|
|  |  |  | result.setEnable_audio(true); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (mediaInfo.getRecordAssistPort() > 0 && userSetting.getRecordPath() == null) { | 
|---|
|  |  |  | logger.info("推流时发现尚未设置录像路径,从assist服务中读取"); | 
|---|
|  |  |  | JSONObject info = assistRESTfulUtils.getInfo(mediaInfo, null); | 
|---|
|  |  |  | if (info != null && info.getInteger("code") != null && info.getInteger("code") == 0) { | 
|---|
|  |  |  | JSONObject dataJson = info.getJSONObject("data"); | 
|---|
|  |  |  | if (dataJson != null) { | 
|---|
|  |  |  | String recordPath = dataJson.getString("record"); | 
|---|
|  |  |  | userSetting.setRecordPath(recordPath); | 
|---|
|  |  |  | result.setMp4_save_path(recordPath); | 
|---|
|  |  |  | // 修改zlm中的录像路径 | 
|---|
|  |  |  | if (mediaInfo.isAutoConfig()) { | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | mediaServerService.setZLMConfig(mediaInfo, false); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | @PostMapping(value = "/on_stream_changed", produces = "application/json;charset=UTF-8") | 
|---|
|  |  |  | public HookResult onStreamChanged(@RequestBody OnStreamChangedHookParam param) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流注册, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream()); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流注销, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream()); | 
|---|
|  |  |  | MediaServer mediaServer = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (mediaServer == null) { | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | JSONObject ret = new JSONObject(); | 
|---|
|  |  |  | ret.put("code", 0); | 
|---|
|  |  |  | ret.put("msg", "success"); | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | JSONObject json = (JSONObject) JSON.toJSON(param); | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_stream_changed, json); | 
|---|
|  |  |  | if (mediaInfo == null) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流变化未找到ZLM, {}", param.getMediaServerId()); | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (subscribe != null) { | 
|---|
|  |  |  | subscribe.response(mediaInfo, param); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流注册, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream()); | 
|---|
|  |  |  | MediaArrivalEvent mediaArrivalEvent = MediaArrivalEvent.getInstance(this, param, mediaServer); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(mediaArrivalEvent); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流注销, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream()); | 
|---|
|  |  |  | MediaDepartureEvent mediaDepartureEvent = MediaDepartureEvent.getInstance(this, param, mediaServer); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(mediaDepartureEvent); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | List<OnStreamChangedHookParam.MediaTrack> tracks = param.getTracks(); | 
|---|
|  |  |  | // TODO 重构此处逻辑 | 
|---|
|  |  |  | boolean isPush = false; | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | // 处理流注册的鉴权信息 | 
|---|
|  |  |  | if (param.getOriginType() == OriginType.RTMP_PUSH.ordinal() | 
|---|
|  |  |  | || param.getOriginType() == OriginType.RTSP_PUSH.ordinal() | 
|---|
|  |  |  | || param.getOriginType() == OriginType.RTC_PUSH.ordinal()) { | 
|---|
|  |  |  | isPush = true; | 
|---|
|  |  |  | StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (streamAuthorityInfo == null) { | 
|---|
|  |  |  | streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(param); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | streamAuthorityInfo.setOriginType(param.getOriginType()); | 
|---|
|  |  |  | streamAuthorityInfo.setOriginTypeStr(param.getOriginTypeStr()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | redisCatchStorage.updateStreamAuthorityInfo(param.getApp(), param.getStream(), streamAuthorityInfo); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | redisCatchStorage.removeStreamAuthorityInfo(param.getApp(), param.getStream()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if ("rtsp".equals(param.getSchema())) { | 
|---|
|  |  |  | logger.info("流变化:注册->{}, app->{}, stream->{}", param.isRegist(), param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | mediaServerService.addCount(param.getMediaServerId()); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | mediaServerService.removeCount(param.getMediaServerId()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | int updateStatusResult = streamProxyService.updateStatus(param.isRegist(), param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (updateStatusResult > 0) { | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if ("rtp".equals(param.getApp()) && !param.isRegist()) { | 
|---|
|  |  |  | InviteInfo inviteInfo = inviteStreamService.getInviteInfoByStream(null, param.getStream()); | 
|---|
|  |  |  | if (inviteInfo != null && (inviteInfo.getType() == InviteSessionType.PLAY || inviteInfo.getType() == InviteSessionType.PLAYBACK)) { | 
|---|
|  |  |  | inviteStreamService.removeInviteInfo(inviteInfo); | 
|---|
|  |  |  | storager.stopPlay(inviteInfo.getDeviceId(), inviteInfo.getChannelId()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else if ("broadcast".equals(param.getApp())) { | 
|---|
|  |  |  | // 语音对讲推流  stream需要满足格式deviceId_channelId | 
|---|
|  |  |  | if (param.getStream().indexOf("_") > 0) { | 
|---|
|  |  |  | String[] streamArray = param.getStream().split("_"); | 
|---|
|  |  |  | if (streamArray.length == 2) { | 
|---|
|  |  |  | String deviceId = streamArray[0]; | 
|---|
|  |  |  | String channelId = streamArray[1]; | 
|---|
|  |  |  | Device device = deviceService.getDevice(deviceId); | 
|---|
|  |  |  | if (device != null) { | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | if (audioBroadcastManager.exit(deviceId, channelId)) { | 
|---|
|  |  |  | playService.stopAudioBroadcast(deviceId, channelId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 开启语音对讲通道 | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | playService.audioBroadcastCmd(device, channelId, mediaInfo, param.getApp(), param.getStream(), 60, false, (msg) -> { | 
|---|
|  |  |  | logger.info("[语音对讲] 通道建立成功, device: {}, channel: {}", deviceId, channelId); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } catch (InvalidArgumentException | ParseException | SipException e) { | 
|---|
|  |  |  | logger.error("[命令发送失败] 语音对讲: {}", e.getMessage()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | // 流注销 | 
|---|
|  |  |  | playService.stopAudioBroadcast(deviceId, channelId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | logger.info("[语音对讲] 未找到设备:{}", deviceId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else if ("talk".equals(param.getApp())) { | 
|---|
|  |  |  | // 语音对讲推流  stream需要满足格式deviceId_channelId | 
|---|
|  |  |  | if (param.getStream().indexOf("_") > 0) { | 
|---|
|  |  |  | String[] streamArray = param.getStream().split("_"); | 
|---|
|  |  |  | if (streamArray.length == 2) { | 
|---|
|  |  |  | String deviceId = streamArray[0]; | 
|---|
|  |  |  | String channelId = streamArray[1]; | 
|---|
|  |  |  | Device device = deviceService.getDevice(deviceId); | 
|---|
|  |  |  | if (device != null) { | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | if (audioBroadcastManager.exit(deviceId, channelId)) { | 
|---|
|  |  |  | playService.stopAudioBroadcast(deviceId, channelId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 开启语音对讲通道 | 
|---|
|  |  |  | playService.talkCmd(device, channelId, mediaInfo, param.getStream(), (msg) -> { | 
|---|
|  |  |  | logger.info("[语音对讲] 通道建立成功, device: {}, channel: {}", deviceId, channelId); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | // 流注销 | 
|---|
|  |  |  | playService.stopTalk(device, channelId, param.isRegist()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | logger.info("[语音对讲] 未找到设备:{}", deviceId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | if (!"rtp".equals(param.getApp())) { | 
|---|
|  |  |  | String type = OriginType.values()[param.getOriginType()].getType(); | 
|---|
|  |  |  | MediaServerItem mediaServerItem = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (mediaServerItem != null) { | 
|---|
|  |  |  | if (param.isRegist()) { | 
|---|
|  |  |  | StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream()); | 
|---|
|  |  |  | String callId = null; | 
|---|
|  |  |  | if (streamAuthorityInfo != null) { | 
|---|
|  |  |  | callId = streamAuthorityInfo.getCallId(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | StreamInfo streamInfoByAppAndStream = mediaService.getStreamInfoByAppAndStream(mediaServerItem, | 
|---|
|  |  |  | param.getApp(), param.getStream(), param.getTracks(), callId); | 
|---|
|  |  |  | param.setStreamInfo(new StreamContent(streamInfoByAppAndStream)); | 
|---|
|  |  |  | redisCatchStorage.addStream(mediaServerItem, type, param.getApp(), param.getStream(), param); | 
|---|
|  |  |  | if (param.getOriginType() == OriginType.RTSP_PUSH.ordinal() | 
|---|
|  |  |  | || param.getOriginType() == OriginType.RTMP_PUSH.ordinal() | 
|---|
|  |  |  | || param.getOriginType() == OriginType.RTC_PUSH.ordinal()) { | 
|---|
|  |  |  | param.setSeverId(userSetting.getServerId()); | 
|---|
|  |  |  | zlmMediaListManager.addPush(param); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | // 兼容流注销时类型从redis记录获取 | 
|---|
|  |  |  | OnStreamChangedHookParam onStreamChangedHookParam = redisCatchStorage.getStreamInfo( | 
|---|
|  |  |  | param.getApp(), param.getStream(), param.getMediaServerId()); | 
|---|
|  |  |  | if (onStreamChangedHookParam != null) { | 
|---|
|  |  |  | type = OriginType.values()[onStreamChangedHookParam.getOriginType()].getType(); | 
|---|
|  |  |  | redisCatchStorage.removeStream(mediaServerItem.getId(), type, param.getApp(), param.getStream()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | GbStream gbStream = storager.getGbStream(param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (gbStream != null) { | 
|---|
|  |  |  | //                           eventPublisher.catalogEventPublishForStream(null, gbStream, CatalogEvent.OFF); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | zlmMediaListManager.removeMedia(param.getApp(), param.getStream()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | GbStream gbStream = storager.getGbStream(param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (gbStream != null) { | 
|---|
|  |  |  | if (userSetting.isUsePushingAsStatus()) { | 
|---|
|  |  |  | eventPublisher.catalogEventPublishForStream(null, gbStream, param.isRegist()?CatalogEvent.ON:CatalogEvent.OFF); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (type != null) { | 
|---|
|  |  |  | // 发送流变化redis消息 | 
|---|
|  |  |  | JSONObject jsonObject = new JSONObject(); | 
|---|
|  |  |  | jsonObject.put("serverId", userSetting.getServerId()); | 
|---|
|  |  |  | jsonObject.put("app", param.getApp()); | 
|---|
|  |  |  | jsonObject.put("stream", param.getStream()); | 
|---|
|  |  |  | jsonObject.put("register", param.isRegist()); | 
|---|
|  |  |  | jsonObject.put("mediaServerId", param.getMediaServerId()); | 
|---|
|  |  |  | redisCatchStorage.sendStreamChangeMsg(type, jsonObject); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (!param.isRegist()) { | 
|---|
|  |  |  | List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(param.getStream()); | 
|---|
|  |  |  | if (sendRtpItems.size() > 0) { | 
|---|
|  |  |  | for (SendRtpItem sendRtpItem : sendRtpItems) { | 
|---|
|  |  |  | if (sendRtpItem != null && sendRtpItem.getApp().equals(param.getApp())) { | 
|---|
|  |  |  | String platformId = sendRtpItem.getPlatformId(); | 
|---|
|  |  |  | ParentPlatform platform = storager.queryParentPlatByServerGBId(platformId); | 
|---|
|  |  |  | Device device = deviceService.getDevice(platformId); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | if (platform != null) { | 
|---|
|  |  |  | commanderFroPlatform.streamByeCmd(platform, sendRtpItem); | 
|---|
|  |  |  | redisCatchStorage.deleteSendRTPServer(platformId, sendRtpItem.getChannelId(), | 
|---|
|  |  |  | sendRtpItem.getCallId(), sendRtpItem.getStream()); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | cmder.streamByeCmd(device, sendRtpItem.getChannelId(), param.getStream(), sendRtpItem.getCallId()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } catch (SipException | InvalidArgumentException | ParseException | | 
|---|
|  |  |  | SsrcTransactionNotFoundException e) { | 
|---|
|  |  |  | logger.error("[命令发送失败] 发送BYE: {}", e.getMessage()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | logger.info("[ZLM HOOK]流无人观看:{}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), | 
|---|
|  |  |  | param.getApp(), param.getStream()); | 
|---|
|  |  |  | JSONObject ret = new JSONObject(); | 
|---|
|  |  |  | boolean close = mediaService.closeStreamOnNoneReader(param.getMediaServerId(), param.getApp(), param.getStream(), param.getSchema()); | 
|---|
|  |  |  | ret.put("code", 0); | 
|---|
|  |  |  | // 国标类型的流 | 
|---|
|  |  |  | if ("rtp".equals(param.getApp())) { | 
|---|
|  |  |  | ret.put("close", userSetting.getStreamOnDemand()); | 
|---|
|  |  |  | // 国标流, 点播/录像回放/录像下载 | 
|---|
|  |  |  | //            StreamInfo streamInfoForPlayCatch = redisCatchStorage.queryPlayByStreamId(param.getStream()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | InviteInfo inviteInfo = inviteStreamService.getInviteInfoByStream(null, param.getStream()); | 
|---|
|  |  |  | // 点播 | 
|---|
|  |  |  | if (inviteInfo != null) { | 
|---|
|  |  |  | // 录像下载 | 
|---|
|  |  |  | if (inviteInfo.getType() == InviteSessionType.DOWNLOAD) { | 
|---|
|  |  |  | ret.put("close", false); | 
|---|
|  |  |  | return ret; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 收到无人观看说明流也没有在往上级推送 | 
|---|
|  |  |  | if (redisCatchStorage.isChannelSendingRTP(inviteInfo.getChannelId())) { | 
|---|
|  |  |  | List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByChnnelId( | 
|---|
|  |  |  | inviteInfo.getChannelId()); | 
|---|
|  |  |  | 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.getStream()); | 
|---|
|  |  |  | if (InviteStreamType.PUSH == sendRtpItem.getPlayType()) { | 
|---|
|  |  |  | MessageForPushChannel messageForPushChannel = MessageForPushChannel.getInstance(0, | 
|---|
|  |  |  | sendRtpItem.getApp(), sendRtpItem.getStream(), sendRtpItem.getChannelId(), | 
|---|
|  |  |  | sendRtpItem.getPlatformId(), parentPlatform.getName(), userSetting.getServerId(), sendRtpItem.getMediaServerId()); | 
|---|
|  |  |  | messageForPushChannel.setPlatFormIndex(parentPlatform.getId()); | 
|---|
|  |  |  | redisCatchStorage.sendPlatformStopPlayMsg(messageForPushChannel); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Device device = deviceService.getDevice(inviteInfo.getDeviceId()); | 
|---|
|  |  |  | if (device != null) { | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | InviteInfo info = inviteStreamService.getInviteInfo(inviteInfo.getType(), | 
|---|
|  |  |  | inviteInfo.getDeviceId(), inviteInfo.getChannelId(), inviteInfo.getStream()); | 
|---|
|  |  |  | if (info != null) { | 
|---|
|  |  |  | cmder.streamByeCmd(device, inviteInfo.getChannelId(), | 
|---|
|  |  |  | inviteInfo.getStream(), null); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } catch (InvalidArgumentException | ParseException | SipException | | 
|---|
|  |  |  | SsrcTransactionNotFoundException e) { | 
|---|
|  |  |  | logger.error("[无人观看]点播, 发送BYE失败 {}", e.getMessage()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | inviteStreamService.removeInviteInfo(inviteInfo.getType(), inviteInfo.getDeviceId(), | 
|---|
|  |  |  | inviteInfo.getChannelId(), inviteInfo.getStream()); | 
|---|
|  |  |  | storager.stopPlay(inviteInfo.getDeviceId(), inviteInfo.getChannelId()); | 
|---|
|  |  |  | return ret; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | SendRtpItem sendRtpItem = redisCatchStorage.querySendRTPServer(null, null, param.getStream(), null); | 
|---|
|  |  |  | if ("talk".equals(sendRtpItem.getApp())) { | 
|---|
|  |  |  | ret.put("close", false); | 
|---|
|  |  |  | return ret; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } else if ("talk".equals(param.getApp()) || "broadcast".equals(param.getApp())) { | 
|---|
|  |  |  | ret.put("close", false); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | // 非国标流 推流/拉流代理 | 
|---|
|  |  |  | // 拉流代理 | 
|---|
|  |  |  | StreamProxyItem streamProxyItem = streamProxyService.getStreamProxyByAppAndStream(param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (streamProxyItem != null) { | 
|---|
|  |  |  | if (streamProxyItem.isEnableDisableNoneReader()) { | 
|---|
|  |  |  | // 无人观看自动移除 | 
|---|
|  |  |  | ret.put("close", true); | 
|---|
|  |  |  | streamProxyService.del(param.getApp(), param.getStream()); | 
|---|
|  |  |  | String url = streamProxyItem.getUrl() != null ? streamProxyItem.getUrl() : streamProxyItem.getSrcUrl(); | 
|---|
|  |  |  | logger.info("[{}/{}]<-[{}] 拉流代理无人观看已经移除", param.getApp(), param.getStream(), url); | 
|---|
|  |  |  | } else if (streamProxyItem.isEnableDisableNoneReader()) { | 
|---|
|  |  |  | // 无人观看停用 | 
|---|
|  |  |  | ret.put("close", true); | 
|---|
|  |  |  | // 修改数据 | 
|---|
|  |  |  | streamProxyService.stop(param.getApp(), param.getStream()); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | // 无人观看不做处理 | 
|---|
|  |  |  | ret.put("close", false); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return ret; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // TODO 推流具有主动性,暂时不做处理 | 
|---|
|  |  |  | //         StreamPushItem streamPushItem = streamPushService.getPush(app, streamId); | 
|---|
|  |  |  | //         if (streamPushItem != null) { | 
|---|
|  |  |  | //            // TODO 发送停止 | 
|---|
|  |  |  | // | 
|---|
|  |  |  | //         } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | ret.put("close", close); | 
|---|
|  |  |  | return ret; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ResponseBody | 
|---|
|  |  |  | @PostMapping(value = "/on_stream_not_found", produces = "application/json;charset=UTF-8") | 
|---|
|  |  |  | public DeferredResult<HookResult> onStreamNotFound(@RequestBody OnStreamNotFoundHookParam param) { | 
|---|
|  |  |  | public HookResult onStreamNotFound(@RequestBody OnStreamNotFoundHookParam param) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流未找到:{}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | DeferredResult<HookResult> defaultResult = new DeferredResult<>(); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (!userSetting.isAutoApplyPlay() || mediaInfo == null) { | 
|---|
|  |  |  | defaultResult.setResult(new HookResult(ErrorCode.ERROR404.getCode(), ErrorCode.ERROR404.getMsg())); | 
|---|
|  |  |  | return defaultResult; | 
|---|
|  |  |  | MediaServer mediaServer = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (!userSetting.isAutoApplyPlay() || mediaServer == null) { | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if ("rtp".equals(param.getApp())) { | 
|---|
|  |  |  | String[] s = param.getStream().split("_"); | 
|---|
|  |  |  | if (!mediaInfo.isRtpEnable() || s.length != 2) { | 
|---|
|  |  |  | defaultResult.setResult(HookResult.SUCCESS()); | 
|---|
|  |  |  | return defaultResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | String deviceId = s[0]; | 
|---|
|  |  |  | String channelId = s[1]; | 
|---|
|  |  |  | Device device = redisCatchStorage.getDevice(deviceId); | 
|---|
|  |  |  | if (device == null) { | 
|---|
|  |  |  | defaultResult.setResult(new HookResult(ErrorCode.ERROR404.getCode(), ErrorCode.ERROR404.getMsg())); | 
|---|
|  |  |  | return defaultResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | DeviceChannel deviceChannel = storager.queryChannel(deviceId, channelId); | 
|---|
|  |  |  | if (deviceChannel == null) { | 
|---|
|  |  |  | defaultResult.setResult(new HookResult(ErrorCode.ERROR404.getCode(), ErrorCode.ERROR404.getMsg())); | 
|---|
|  |  |  | return defaultResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 流未找到, 发起自动点播:{}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 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<HookResult> result = new DeferredResult<>(userSetting.getPlayTimeout().longValue()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | result.onTimeout(() -> { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 自动点播, 等待超时"); | 
|---|
|  |  |  | // 释放rtpserver | 
|---|
|  |  |  | msg.setData(new HookResult(ErrorCode.ERROR100.getCode(), "点播超时")); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 录像查询以channelId作为deviceId查询 | 
|---|
|  |  |  | resultHolder.put(key, uuid, result); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (!exist) { | 
|---|
|  |  |  | playService.play(mediaInfo, deviceId, channelId, (code, message, data) -> { | 
|---|
|  |  |  | msg.setData(new HookResult(code, message)); | 
|---|
|  |  |  | resultHolder.invokeResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | // 拉流代理 | 
|---|
|  |  |  | StreamProxyItem streamProxyByAppAndStream = streamProxyService.getStreamProxyByAppAndStream(param.getApp(), param.getStream()); | 
|---|
|  |  |  | if (streamProxyByAppAndStream != null && streamProxyByAppAndStream.isEnableDisableNoneReader()) { | 
|---|
|  |  |  | streamProxyService.start(param.getApp(), param.getStream()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | DeferredResult<HookResult> result = new DeferredResult<>(); | 
|---|
|  |  |  | result.setResult(HookResult.SUCCESS()); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | MediaNotFoundEvent mediaNotFoundEvent = MediaNotFoundEvent.getInstance(this, param, mediaServer); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(mediaNotFoundEvent); | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | 
|---|
|  |  |  | ZLMServerConfig zlmServerConfig = JSON.to(ZLMServerConfig.class, jsonObject); | 
|---|
|  |  |  | zlmServerConfig.setIp(request.getRemoteAddr()); | 
|---|
|  |  |  | logger.info("[ZLM HOOK] zlm 启动 " + zlmServerConfig.getGeneralMediaServerId()); | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_started); | 
|---|
|  |  |  | if (subscribes != null && subscribes.size() > 0) { | 
|---|
|  |  |  | for (ZlmHttpHookSubscribe.Event subscribe : subscribes) { | 
|---|
|  |  |  | subscribe.response(null, zlmServerConfig); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | HookZlmServerStartEvent event = new HookZlmServerStartEvent(this); | 
|---|
|  |  |  | MediaServer mediaServerItem = mediaServerService.getOne(zlmServerConfig.getMediaServerId()); | 
|---|
|  |  |  | if (mediaServerItem != null) { | 
|---|
|  |  |  | event.setMediaServerItem(mediaServerItem); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(event); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | mediaServerService.zlmServerOnline(zlmServerConfig); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | }catch (Exception e) { | 
|---|
|  |  |  | logger.info("[ZLM-HOOK-ZLM启动] 发送通知失败 ", e); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | if (!"rtp".equals(param.getApp())) { | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(param.getStream()); | 
|---|
|  |  |  | if (sendRtpItems.size() > 0) { | 
|---|
|  |  |  | for (SendRtpItem sendRtpItem : sendRtpItems) { | 
|---|
|  |  |  | ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId()); | 
|---|
|  |  |  | ssrcFactory.releaseSsrc(sendRtpItem.getMediaServerId(), sendRtpItem.getSsrc()); | 
|---|
|  |  |  | 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.getStream()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | MediaSendRtpStoppedEvent event = new MediaSendRtpStoppedEvent(this); | 
|---|
|  |  |  | MediaServer mediaServerItem = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (mediaServerItem != null) { | 
|---|
|  |  |  | event.setMediaServer(mediaServerItem); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(event); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | }catch (Exception e) { | 
|---|
|  |  |  | logger.info("[ZLM-HOOK-rtp发送关闭] 发送通知失败 ", e); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ResponseBody | 
|---|
|  |  |  | @PostMapping(value = "/on_rtp_server_timeout", produces = "application/json;charset=UTF-8") | 
|---|
|  |  |  | public HookResult onRtpServerTimeout(HttpServletRequest request, @RequestBody OnRtpServerTimeoutHookParam | 
|---|
|  |  |  | public HookResult onRtpServerTimeout(@RequestBody OnRtpServerTimeoutHookParam | 
|---|
|  |  |  | param) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] rtpServer收流超时:{}->{}({})", param.getMediaServerId(), param.getStream_id(), param.getSsrc()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | taskExecutor.execute(() -> { | 
|---|
|  |  |  | JSONObject json = (JSONObject) JSON.toJSON(param); | 
|---|
|  |  |  | List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_rtp_server_timeout); | 
|---|
|  |  |  | if (subscribes != null && subscribes.size() > 0) { | 
|---|
|  |  |  | for (ZlmHttpHookSubscribe.Event subscribe : subscribes) { | 
|---|
|  |  |  | subscribe.response(null, param); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | MediaRtpServerTimeoutEvent event = new MediaRtpServerTimeoutEvent(this); | 
|---|
|  |  |  | MediaServer mediaServerItem = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (mediaServerItem != null) { | 
|---|
|  |  |  | event.setMediaServer(mediaServerItem); | 
|---|
|  |  |  | event.setApp("rtp"); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(event); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | }catch (Exception e) { | 
|---|
|  |  |  | logger.info("[ZLM-HOOK-rtpServer收流超时] 发送通知失败 ", e); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 录像完成事件 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @ResponseBody | 
|---|
|  |  |  | @PostMapping(value = "/on_record_mp4", produces = "application/json;charset=UTF-8") | 
|---|
|  |  |  | public HookResult onRecordMp4(HttpServletRequest request, @RequestBody OnRecordMp4HookParam param) { | 
|---|
|  |  |  | logger.info("[ZLM HOOK] 录像完成事件:{}->{}", param.getMediaServerId(), param.getFile_path()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | MediaServer mediaServerItem = mediaServerService.getOne(param.getMediaServerId()); | 
|---|
|  |  |  | if (mediaServerItem != null) { | 
|---|
|  |  |  | MediaRecordMp4Event event = MediaRecordMp4Event.getInstance(this, param, mediaServerItem); | 
|---|
|  |  |  | event.setMediaServer(mediaServerItem); | 
|---|
|  |  |  | applicationEventPublisher.publishEvent(event); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }catch (Exception e) { | 
|---|
|  |  |  | logger.info("[ZLM-HOOK-rtpServer收流超时] 发送通知失败 ", e); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return HookResult.SUCCESS(); | 
|---|
|  |  |  | } | 
|---|