|  |  |  | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSON; | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSONArray; | 
|---|
|  |  |  | import com.alibaba.fastjson2.JSONObject; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.CommonCallback; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.common.VideoManagerConstants; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.DynamicTask; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.SipConfig; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.UserSetting; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.conf.exception.ControllerException; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.event.EventPublisher; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.SsrcConfig; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.session.SSRCFactory; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.AssistRESTfulUtils; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory; | 
|---|
|  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private SipConfig sipConfig; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private SSRCFactory ssrcFactory; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Value("${server.ssl.enabled:false}") | 
|---|
|  |  |  | private boolean sslEnabled; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private RedisTemplate<Object, Object> redisTemplate; | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 初始化 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | 
|---|
|  |  |  | continue; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 更新 | 
|---|
|  |  |  | if (mediaServerItem.getSsrcConfig() == null) { | 
|---|
|  |  |  | SsrcConfig ssrcConfig = new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain()); | 
|---|
|  |  |  | mediaServerItem.setSsrcConfig(ssrcConfig); | 
|---|
|  |  |  | redisTemplate.opsForValue().set(VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId(), mediaServerItem); | 
|---|
|  |  |  | if (ssrcFactory.hasMediaServerSSRC(mediaServerItem.getId())) { | 
|---|
|  |  |  | ssrcFactory.initMediaServerSSRC(mediaServerItem.getId(), null); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 查询redis是否存在此mediaServer | 
|---|
|  |  |  | String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId(); | 
|---|
|  |  |  | 
|---|
|  |  |  | return null; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 获取mediaServer可用的ssrc | 
|---|
|  |  |  | String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId(); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | SsrcConfig ssrcConfig = mediaServerItem.getSsrcConfig(); | 
|---|
|  |  |  | if (ssrcConfig == null) { | 
|---|
|  |  |  | logger.info("media server [ {} ] ssrcConfig is null", mediaServerItem.getId()); | 
|---|
|  |  |  | return null; | 
|---|
|  |  |  | String ssrc; | 
|---|
|  |  |  | if (presetSsrc != null) { | 
|---|
|  |  |  | ssrc = presetSsrc; | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | String ssrc; | 
|---|
|  |  |  | if (presetSsrc != null) { | 
|---|
|  |  |  | ssrc = presetSsrc; | 
|---|
|  |  |  | if (isPlayback) { | 
|---|
|  |  |  | ssrc = ssrcFactory.getPlayBackSsrc(mediaServerItem.getId()); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | if (isPlayback) { | 
|---|
|  |  |  | ssrc = ssrcConfig.getPlayBackSsrc(); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | ssrc = ssrcConfig.getPlaySsrc(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | ssrc = ssrcFactory.getPlaySsrc(mediaServerItem.getId()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (streamId == null) { | 
|---|
|  |  |  | streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | int rtpServerPort; | 
|---|
|  |  |  | if (mediaServerItem.isRtpEnable()) { | 
|---|
|  |  |  | rtpServerPort = zlmrtpServerFactory.createRTPServer(mediaServerItem, streamId, ssrcCheck?Integer.parseInt(ssrc):0, port); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | rtpServerPort = mediaServerItem.getRtpProxyPort(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | redisTemplate.opsForValue().set(key, mediaServerItem); | 
|---|
|  |  |  | return new SSRCInfo(rtpServerPort, ssrc, streamId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (streamId == null) { | 
|---|
|  |  |  | streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | int rtpServerPort; | 
|---|
|  |  |  | if (mediaServerItem.isRtpEnable()) { | 
|---|
|  |  |  | rtpServerPort = zlmrtpServerFactory.createRTPServer(mediaServerItem, streamId, ssrcCheck?Integer.parseInt(ssrc):0, port); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | rtpServerPort = mediaServerItem.getRtpProxyPort(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return new SSRCInfo(rtpServerPort, ssrc, streamId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public void closeRTPServer(MediaServerItem mediaServerItem, String streamId, CommonCallback<Boolean> callback) { | 
|---|
|  |  |  | if (mediaServerItem == null) { | 
|---|
|  |  |  | callback.run(false); | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | zlmrtpServerFactory.closeRtpServer(mediaServerItem, streamId, callback); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public void closeRTPServer(String mediaServerId, String streamId) { | 
|---|
|  |  |  | MediaServerItem mediaServerItem = this.getOne(mediaServerId); | 
|---|
|  |  |  | closeRTPServer(mediaServerItem, streamId); | 
|---|
|  |  |  | 
|---|
|  |  |  | if (mediaServerItem == null || ssrc == null) { | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | SsrcConfig ssrcConfig = mediaServerItem.getSsrcConfig(); | 
|---|
|  |  |  | ssrcConfig.releaseSsrc(ssrc); | 
|---|
|  |  |  | mediaServerItem.setSsrcConfig(ssrcConfig); | 
|---|
|  |  |  | String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId(); | 
|---|
|  |  |  | redisTemplate.opsForValue().set(key, mediaServerItem); | 
|---|
|  |  |  | ssrcFactory.releaseSsrc(mediaServerItemId, ssrc); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public void clearRTPServer(MediaServerItem mediaServerItem) { | 
|---|
|  |  |  | mediaServerItem.setSsrcConfig(new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain())); | 
|---|
|  |  |  | redisTemplate.opsForZSet().add(VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId(), mediaServerItem.getId(), 0); | 
|---|
|  |  |  | ssrcFactory.reset(mediaServerItem.getId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | mediaServerMapper.update(mediaSerItem); | 
|---|
|  |  |  | MediaServerItem mediaServerItemInRedis = getOne(mediaSerItem.getId()); | 
|---|
|  |  |  | MediaServerItem mediaServerItemInDataBase = mediaServerMapper.queryOne(mediaSerItem.getId()); | 
|---|
|  |  |  | if (mediaServerItemInRedis != null && mediaServerItemInRedis.getSsrcConfig() != null) { | 
|---|
|  |  |  | mediaServerItemInDataBase.setSsrcConfig(mediaServerItemInRedis.getSsrcConfig()); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | mediaServerItemInDataBase.setSsrcConfig( | 
|---|
|  |  |  | new SsrcConfig( | 
|---|
|  |  |  | mediaServerItemInDataBase.getId(), | 
|---|
|  |  |  | null, | 
|---|
|  |  |  | sipConfig.getDomain() | 
|---|
|  |  |  | ) | 
|---|
|  |  |  | ); | 
|---|
|  |  |  | if (mediaServerItemInRedis == null || ssrcFactory.hasMediaServerSSRC(mediaSerItem.getId())) { | 
|---|
|  |  |  | ssrcFactory.initMediaServerSSRC(mediaServerItemInDataBase.getId(),null); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItemInDataBase.getId(); | 
|---|
|  |  |  | redisTemplate.opsForValue().set(key, mediaServerItemInDataBase); | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | mediaServerMapper.update(serverItem); | 
|---|
|  |  |  | String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + zlmServerConfig.getGeneralMediaServerId(); | 
|---|
|  |  |  | if (redisTemplate.opsForValue().get(key) == null) { | 
|---|
|  |  |  | SsrcConfig ssrcConfig = new SsrcConfig(zlmServerConfig.getGeneralMediaServerId(), null, sipConfig.getDomain()); | 
|---|
|  |  |  | serverItem.setSsrcConfig(ssrcConfig); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | MediaServerItem mediaServerItemInRedis = JsonUtil.redisJsonToObject(redisTemplate, key, MediaServerItem.class); | 
|---|
|  |  |  | if (Objects.nonNull(mediaServerItemInRedis)) { | 
|---|
|  |  |  | serverItem.setSsrcConfig(mediaServerItemInRedis.getSsrcConfig()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (ssrcFactory.hasMediaServerSSRC(serverItem.getId())) { | 
|---|
|  |  |  | ssrcFactory.initMediaServerSSRC(zlmServerConfig.getGeneralMediaServerId(), null); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | redisTemplate.opsForValue().set(key, serverItem); | 
|---|
|  |  |  | resetOnlineServerItem(serverItem); | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // zlm连接重试 | 
|---|
|  |  |  | logger.warn("[更新ZLM 保活信息]尝试链接zml id {}", mediaServerId); | 
|---|
|  |  |  | SsrcConfig ssrcConfig = new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain()); | 
|---|
|  |  |  | mediaServerItem.setSsrcConfig(ssrcConfig); | 
|---|
|  |  |  | ssrcFactory.initMediaServerSSRC(mediaServerItem.getId(), null); | 
|---|
|  |  |  | String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId(); | 
|---|
|  |  |  | redisTemplate.opsForValue().set(key, mediaServerItem); | 
|---|
|  |  |  | clearRTPServer(mediaServerItem); | 
|---|