|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import java.util.ArrayList; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.UUID; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.alibaba.fastjson.JSON; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.StreamInfo; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.MediaConfig; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.UserSetup; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.UserSetting; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.Device; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.GbStream; | 
|---|
|  |  |  | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.dto.*; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.*; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.service.bean.SSRCInfo; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorager; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; | 
|---|
|  |  |  | import org.slf4j.Logger; | 
|---|
|  |  |  | import org.slf4j.LoggerFactory; | 
|---|
|  |  |  | import org.springframework.beans.factory.annotation.Autowired; | 
|---|
|  |  |  | 
|---|
|  |  |  | private IPlayService playService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IVideoManagerStorager storager; | 
|---|
|  |  |  | private IVideoManagerStorage storager; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IRedisCatchStorage redisCatchStorage; | 
|---|
|  |  |  | 
|---|
|  |  |  | private ZLMHttpHookSubscribe subscribe; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private UserSetup userSetup; | 
|---|
|  |  |  | private UserSetting userSetting; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private VideoStreamSessionManager sessionManager; | 
|---|
|  |  |  | 
|---|
|  |  |  | JSONObject ret = new JSONObject(); | 
|---|
|  |  |  | ret.put("code", 0); | 
|---|
|  |  |  | ret.put("msg", "success"); | 
|---|
|  |  |  | ret.put("enableHls", true); | 
|---|
|  |  |  | ret.put("enable_hls", true); | 
|---|
|  |  |  | if (json.getInteger("originType") == 1 | 
|---|
|  |  |  | || json.getInteger("originType") == 2 | 
|---|
|  |  |  | || json.getInteger("originType") == 3) { | 
|---|
|  |  |  | ret.put("enable_audio", true); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | String mediaServerId = json.getString("mediaServerId"); | 
|---|
|  |  |  | ZLMHttpHookSubscribe.Event subscribe = this.subscribe.getSubscribe(ZLMHttpHookSubscribe.HookType.on_publish, json); | 
|---|
|  |  |  | if (subscribe != null) { | 
|---|
|  |  |  | 
|---|
|  |  |  | String app = json.getString("app"); | 
|---|
|  |  |  | String stream = json.getString("stream"); | 
|---|
|  |  |  | if ("rtp".equals(app)) { | 
|---|
|  |  |  | ret.put("enableMP4", userSetup.getRecordSip()); | 
|---|
|  |  |  | ret.put("enable_mp4", userSetting.getRecordSip()); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | ret.put("enableMP4", userSetup.isRecordPushLive()); | 
|---|
|  |  |  | ret.put("enable_mp4", userSetting.isRecordPushLive()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | List<SsrcTransaction> ssrcTransactionForAll = sessionManager.getSsrcTransactionForAll(null, null, null, stream); | 
|---|
|  |  |  | if (ssrcTransactionForAll != null && ssrcTransactionForAll.size() == 1) { | 
|---|
|  |  |  | 
|---|
|  |  |  | if (deviceChannel != null) { | 
|---|
|  |  |  | ret.put("enable_audio", deviceChannel.isHasAudio()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 如果是录像下载就设置视频间隔十秒 | 
|---|
|  |  |  | if (ssrcTransactionForAll.get(0).getType() == VideoStreamSessionManager.SessionType.download) { | 
|---|
|  |  |  | ret.put("mp4_max_second", 10); | 
|---|
|  |  |  | ret.put("enable_mp4", true); | 
|---|
|  |  |  | ret.put("enable_audio", true); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | return new ResponseEntity<String>(ret.toString(), HttpStatus.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | if (mediaInfo != null) { | 
|---|
|  |  |  | subscribe.response(mediaInfo, json); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 流消失移除redis play | 
|---|
|  |  |  | String app = item.getApp(); | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (gbStreams.size() > 0) { | 
|---|
|  |  |  | eventPublisher.catalogEventPublishForStream(null, gbStreams, CatalogEvent.ON); | 
|---|
|  |  |  | //                        eventPublisher.catalogEventPublishForStream(null, gbStreams, CatalogEvent.ON); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | GbStream gbStream = storager.getGbStream(app, streamId); | 
|---|
|  |  |  | if (gbStream != null) { | 
|---|
|  |  |  | eventPublisher.catalogEventPublishForStream(null, gbStream, CatalogEvent.OFF); | 
|---|
|  |  |  | //                        eventPublisher.catalogEventPublishForStream(null, gbStream, CatalogEvent.OFF); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | zlmMediaListManager.removeMedia(app, streamId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (type != null) { | 
|---|
|  |  |  | // 发送流变化redis消息 | 
|---|
|  |  |  | JSONObject jsonObject = new JSONObject(); | 
|---|
|  |  |  | jsonObject.put("serverId", userSetup.getServerId()); | 
|---|
|  |  |  | jsonObject.put("serverId", userSetting.getServerId()); | 
|---|
|  |  |  | jsonObject.put("app", app); | 
|---|
|  |  |  | jsonObject.put("stream", streamId); | 
|---|
|  |  |  | jsonObject.put("register", regist); | 
|---|
|  |  |  | 
|---|
|  |  |  | if ("rtp".equals(app)){ | 
|---|
|  |  |  | ret.put("close", true); | 
|---|
|  |  |  | StreamInfo streamInfoForPlayCatch = redisCatchStorage.queryPlayByStreamId(streamId); | 
|---|
|  |  |  | SsrcTransaction ssrcTransaction = sessionManager.getSsrcTransaction(null, null, null, streamId); | 
|---|
|  |  |  | if (streamInfoForPlayCatch != null) { | 
|---|
|  |  |  | // 如果在给上级推流,也不停止。 | 
|---|
|  |  |  | if (redisCatchStorage.isChannelSendingRTP(streamInfoForPlayCatch.getChannelId())) { | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | String mediaServerId = json.getString("mediaServerId"); | 
|---|
|  |  |  | MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId); | 
|---|
|  |  |  | if (userSetup.isAutoApplyPlay() && mediaInfo != null && mediaInfo.isRtpEnable()) { | 
|---|
|  |  |  | if (userSetting.isAutoApplyPlay() && mediaInfo != null && mediaInfo.isRtpEnable()) { | 
|---|
|  |  |  | String app = json.getString("app"); | 
|---|
|  |  |  | String streamId = json.getString("stream"); | 
|---|
|  |  |  | if ("rtp".equals(app)) { | 
|---|