MediaServerServiceImpl.java 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658
  1. package com.genersoft.iot.vmp.service.impl;
  2. import com.alibaba.fastjson.JSON;
  3. import com.alibaba.fastjson.JSONArray;
  4. import com.alibaba.fastjson.JSONObject;
  5. import com.genersoft.iot.vmp.common.VideoManagerConstants;
  6. import com.genersoft.iot.vmp.conf.SipConfig;
  7. import com.genersoft.iot.vmp.conf.UserSetting;
  8. import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
  9. import com.genersoft.iot.vmp.gb28181.session.SsrcConfig;
  10. import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
  11. import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
  12. import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
  13. import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig;
  14. import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
  15. import com.genersoft.iot.vmp.service.IMediaServerService;
  16. import com.genersoft.iot.vmp.service.IStreamProxyService;
  17. import com.genersoft.iot.vmp.service.bean.SSRCInfo;
  18. import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
  19. import com.genersoft.iot.vmp.storager.dao.MediaServerMapper;
  20. import com.genersoft.iot.vmp.utils.DateUtil;
  21. import com.genersoft.iot.vmp.utils.redis.JedisUtil;
  22. import com.genersoft.iot.vmp.utils.redis.RedisUtil;
  23. import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
  24. import okhttp3.*;
  25. import org.slf4j.Logger;
  26. import org.slf4j.LoggerFactory;
  27. import org.springframework.beans.factory.annotation.Autowired;
  28. import org.springframework.beans.factory.annotation.Value;
  29. import org.springframework.jdbc.datasource.DataSourceTransactionManager;
  30. import org.springframework.stereotype.Service;
  31. import org.springframework.transaction.TransactionDefinition;
  32. import org.springframework.transaction.TransactionStatus;
  33. import org.springframework.util.StringUtils;
  34. import java.text.ParseException;
  35. import java.text.SimpleDateFormat;
  36. import java.util.*;
  37. /**
  38. * 媒体服务器节点管理
  39. */
  40. @Service
  41. public class MediaServerServiceImpl implements IMediaServerService {
  42. private final static Logger logger = LoggerFactory.getLogger(MediaServerServiceImpl.class);
  43. @Autowired
  44. private SipConfig sipConfig;
  45. @Value("${server.ssl.enabled:false}")
  46. private boolean sslEnabled;
  47. @Value("${server.port}")
  48. private Integer serverPort;
  49. @Autowired
  50. private UserSetting userSetting;
  51. @Autowired
  52. private ZLMRESTfulUtils zlmresTfulUtils;
  53. @Autowired
  54. private MediaServerMapper mediaServerMapper;
  55. @Autowired
  56. DataSourceTransactionManager dataSourceTransactionManager;
  57. @Autowired
  58. TransactionDefinition transactionDefinition;
  59. @Autowired
  60. private VideoStreamSessionManager streamSession;
  61. @Autowired
  62. private ZLMRTPServerFactory zlmrtpServerFactory;
  63. @Autowired
  64. private RedisUtil redisUtil;
  65. @Autowired
  66. private IVideoManagerStorage storager;
  67. @Autowired
  68. private IStreamProxyService streamProxyService;
  69. @Autowired
  70. private EventPublisher publisher;
  71. @Autowired
  72. JedisUtil jedisUtil;
  73. /**
  74. * 初始化
  75. */
  76. @Override
  77. public void updateVmServer(List<MediaServerItem> mediaServerItemList) {
  78. logger.info("[zlm] 缓存初始化 ");
  79. for (MediaServerItem mediaServerItem : mediaServerItemList) {
  80. if (StringUtils.isEmpty(mediaServerItem.getId())) {
  81. continue;
  82. }
  83. // 更新
  84. if (mediaServerItem.getSsrcConfig() == null) {
  85. SsrcConfig ssrcConfig = new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain());
  86. mediaServerItem.setSsrcConfig(ssrcConfig);
  87. redisUtil.set(VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId(), mediaServerItem);
  88. }
  89. // 查询redis是否存在此mediaServer
  90. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId();
  91. if (!redisUtil.hasKey(key)) {
  92. redisUtil.set(key, mediaServerItem);
  93. }
  94. }
  95. }
  96. @Override
  97. public SSRCInfo openRTPServer(MediaServerItem mediaServerItem, String streamId, boolean ssrcCheck, boolean isPlayback) {
  98. return openRTPServer(mediaServerItem, streamId, null, ssrcCheck,isPlayback);
  99. }
  100. @Override
  101. public SSRCInfo openRTPServer(MediaServerItem mediaServerItem, String streamId, String presetSsrc, boolean ssrcCheck, boolean isPlayback) {
  102. if (mediaServerItem == null || mediaServerItem.getId() == null) {
  103. return null;
  104. }
  105. // 获取mediaServer可用的ssrc
  106. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId();
  107. SsrcConfig ssrcConfig = mediaServerItem.getSsrcConfig();
  108. if (ssrcConfig == null) {
  109. logger.info("media server [ {} ] ssrcConfig is null", mediaServerItem.getId());
  110. return null;
  111. }else {
  112. String ssrc = null;
  113. if (presetSsrc != null) {
  114. ssrc = presetSsrc;
  115. }else {
  116. if (isPlayback) {
  117. ssrc = ssrcConfig.getPlayBackSsrc();
  118. }else {
  119. ssrc = ssrcConfig.getPlaySsrc();
  120. }
  121. }
  122. if (streamId == null) {
  123. streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase();
  124. }
  125. int rtpServerPort = mediaServerItem.getRtpProxyPort();
  126. if (mediaServerItem.isRtpEnable()) {
  127. rtpServerPort = zlmrtpServerFactory.createRTPServer(mediaServerItem, streamId, ssrcCheck?Integer.parseInt(ssrc):0);
  128. }
  129. redisUtil.set(key, mediaServerItem);
  130. return new SSRCInfo(rtpServerPort, ssrc, streamId);
  131. }
  132. }
  133. @Override
  134. public void closeRTPServer(String deviceId, String channelId, String stream) {
  135. String mediaServerId = streamSession.getMediaServerId(deviceId, channelId, stream);
  136. String ssrc = streamSession.getSSRC(deviceId, channelId, stream);
  137. MediaServerItem mediaServerItem = this.getOne(mediaServerId);
  138. if (mediaServerItem != null) {
  139. String streamId = String.format("%s_%s", deviceId, channelId);
  140. zlmrtpServerFactory.closeRTPServer(mediaServerItem, streamId);
  141. releaseSsrc(mediaServerItem.getId(), ssrc);
  142. }
  143. streamSession.remove(deviceId, channelId, stream);
  144. }
  145. @Override
  146. public void releaseSsrc(String mediaServerItemId, String ssrc) {
  147. MediaServerItem mediaServerItem = getOne(mediaServerItemId);
  148. if (mediaServerItem == null || ssrc == null) {
  149. return;
  150. }
  151. SsrcConfig ssrcConfig = mediaServerItem.getSsrcConfig();
  152. ssrcConfig.releaseSsrc(ssrc);
  153. mediaServerItem.setSsrcConfig(ssrcConfig);
  154. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId();
  155. redisUtil.set(key, mediaServerItem);
  156. }
  157. /**
  158. * zlm 重启后重置他的推流信息, TODO 给正在使用的设备发送停止命令
  159. */
  160. @Override
  161. public void clearRTPServer(MediaServerItem mediaServerItem) {
  162. mediaServerItem.setSsrcConfig(new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain()));
  163. redisUtil.zAdd(VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId(), mediaServerItem.getId(), 0);
  164. }
  165. @Override
  166. public void update(MediaServerItem mediaSerItem) {
  167. mediaServerMapper.update(mediaSerItem);
  168. MediaServerItem mediaServerItemInRedis = getOne(mediaSerItem.getId());
  169. MediaServerItem mediaServerItemInDataBase = mediaServerMapper.queryOne(mediaSerItem.getId());
  170. if (mediaServerItemInRedis != null && mediaServerItemInRedis.getSsrcConfig() != null) {
  171. mediaServerItemInDataBase.setSsrcConfig(mediaServerItemInRedis.getSsrcConfig());
  172. }else {
  173. mediaServerItemInDataBase.setSsrcConfig(
  174. new SsrcConfig(
  175. mediaServerItemInDataBase.getId(),
  176. null,
  177. sipConfig.getDomain()
  178. )
  179. );
  180. }
  181. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItemInDataBase.getId();
  182. redisUtil.set(key, mediaServerItemInDataBase);
  183. }
  184. @Override
  185. public List<MediaServerItem> getAll() {
  186. List<MediaServerItem> result = new ArrayList<>();
  187. List<Object> mediaServerKeys = redisUtil.scan(String.format("%S*", VideoManagerConstants.MEDIA_SERVER_PREFIX+ userSetting.getServerId() + "_" ));
  188. String onlineKey = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  189. for (Object mediaServerKey : mediaServerKeys) {
  190. String key = (String) mediaServerKey;
  191. MediaServerItem mediaServerItem = (MediaServerItem) redisUtil.get(key);
  192. // 检查状态
  193. Double aDouble = redisUtil.zScore(onlineKey, mediaServerItem.getId());
  194. if (aDouble != null) {
  195. mediaServerItem.setStatus(true);
  196. }
  197. result.add(mediaServerItem);
  198. }
  199. result.sort((serverItem1, serverItem2)->{
  200. int sortResult = 0;
  201. try {
  202. sortResult = DateUtil.format.parse(serverItem1.getCreateTime()).compareTo(DateUtil.format.parse(serverItem2.getCreateTime()));
  203. } catch (ParseException e) {
  204. e.printStackTrace();
  205. }
  206. return sortResult;
  207. });
  208. return result;
  209. }
  210. @Override
  211. public List<MediaServerItem> getAllFromDatabase() {
  212. return mediaServerMapper.queryAll();
  213. }
  214. @Override
  215. public List<MediaServerItem> getAllOnline() {
  216. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  217. Set<String> mediaServerIdSet = redisUtil.zRevRange(key, 0, -1);
  218. List<MediaServerItem> result = new ArrayList<>();
  219. if (mediaServerIdSet != null && mediaServerIdSet.size() > 0) {
  220. for (String mediaServerId : mediaServerIdSet) {
  221. String serverKey = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerId;
  222. result.add((MediaServerItem) redisUtil.get(serverKey));
  223. }
  224. }
  225. Collections.reverse(result);
  226. return result;
  227. }
  228. /**
  229. * 获取单个zlm服务器
  230. * @param mediaServerId 服务id
  231. * @return MediaServerItem
  232. */
  233. @Override
  234. public MediaServerItem getOne(String mediaServerId) {
  235. if (mediaServerId == null) {
  236. return null;
  237. }
  238. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerId;
  239. return (MediaServerItem)redisUtil.get(key);
  240. }
  241. @Override
  242. public MediaServerItem getDefaultMediaServer() {
  243. return mediaServerMapper.queryDefault();
  244. }
  245. @Override
  246. public void clearMediaServerForOnline() {
  247. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  248. redisUtil.del(key);
  249. }
  250. @Override
  251. public WVPResult<String> add(MediaServerItem mediaServerItem) {
  252. WVPResult<String> result = new WVPResult<>();
  253. mediaServerItem.setCreateTime(DateUtil.getNow());
  254. mediaServerItem.setUpdateTime(DateUtil.getNow());
  255. mediaServerItem.setHookAliveInterval(120);
  256. JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig(mediaServerItem);
  257. if (responseJSON != null) {
  258. JSONArray data = responseJSON.getJSONArray("data");
  259. if (data != null && data.size() > 0) {
  260. ZLMServerConfig zlmServerConfig= JSON.parseObject(JSON.toJSONString(data.get(0)), ZLMServerConfig.class);
  261. if (mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId()) != null) {
  262. result.setCode(-1);
  263. result.setMsg("保存失败,媒体服务ID [ " + zlmServerConfig.getGeneralMediaServerId() + " ] 已存在,请修改媒体服务器配置");
  264. return result;
  265. }
  266. mediaServerItem.setId(zlmServerConfig.getGeneralMediaServerId());
  267. zlmServerConfig.setIp(mediaServerItem.getIp());
  268. mediaServerMapper.add(mediaServerItem);
  269. zlmServerOnline(zlmServerConfig);
  270. result.setCode(0);
  271. result.setMsg("success");
  272. }else {
  273. result.setCode(-1);
  274. result.setMsg("连接失败");
  275. }
  276. }else {
  277. result.setCode(-1);
  278. result.setMsg("连接失败");
  279. }
  280. return result;
  281. }
  282. @Override
  283. public int addToDatabase(MediaServerItem mediaSerItem) {
  284. return mediaServerMapper.add(mediaSerItem);
  285. }
  286. @Override
  287. public int updateToDatabase(MediaServerItem mediaSerItem) {
  288. int result = 0;
  289. if (mediaSerItem.isDefaultServer()) {
  290. TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
  291. int delResult = mediaServerMapper.delDefault();
  292. if (delResult == 0) {
  293. logger.error("移除数据库默认zlm节点失败");
  294. //事务回滚
  295. dataSourceTransactionManager.rollback(transactionStatus);
  296. return 0;
  297. }
  298. result = mediaServerMapper.add(mediaSerItem);
  299. dataSourceTransactionManager.commit(transactionStatus); //手动提交
  300. }else {
  301. result = mediaServerMapper.update(mediaSerItem);
  302. }
  303. return result;
  304. }
  305. /**
  306. * 处理zlm上线
  307. * @param zlmServerConfig zlm上线携带的参数
  308. */
  309. @Override
  310. public void zlmServerOnline(ZLMServerConfig zlmServerConfig) {
  311. logger.info("[ZLM] 正在连接 : {} -> {}:{}",
  312. zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(), zlmServerConfig.getHttpPort());
  313. MediaServerItem serverItem = mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId());
  314. if (serverItem == null) {
  315. logger.warn("[未注册的zlm] 拒接接入:{}来自{}:{}", zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(),zlmServerConfig.getHttpPort() );
  316. logger.warn("请检查ZLM的<general.mediaServerId>配置是否与WVP的<media.id>一致");
  317. return;
  318. }
  319. serverItem.setHookAliveInterval(zlmServerConfig.getHookAliveInterval());
  320. if (serverItem.getHttpPort() == 0) {
  321. serverItem.setHttpPort(zlmServerConfig.getHttpPort());
  322. }
  323. if (serverItem.getHttpSSlPort() == 0) {
  324. serverItem.setHttpSSlPort(zlmServerConfig.getHttpSSLport());
  325. }
  326. if (serverItem.getRtmpPort() == 0) {
  327. serverItem.setRtmpPort(zlmServerConfig.getRtmpPort());
  328. }
  329. if (serverItem.getRtmpSSlPort() == 0) {
  330. serverItem.setRtmpSSlPort(zlmServerConfig.getRtmpSslPort());
  331. }
  332. if (serverItem.getRtspPort() == 0) {
  333. serverItem.setRtspPort(zlmServerConfig.getRtspPort());
  334. }
  335. if (serverItem.getRtspSSLPort() == 0) {
  336. serverItem.setRtspSSLPort(zlmServerConfig.getRtspSSlport());
  337. }
  338. if (serverItem.getRtpProxyPort() == 0) {
  339. serverItem.setRtpProxyPort(zlmServerConfig.getRtpProxyPort());
  340. }
  341. serverItem.setStatus(true);
  342. if (StringUtils.isEmpty(serverItem.getId())) {
  343. logger.warn("[未注册的zlm] serverItem缺少ID, 无法接入:{}:{}", zlmServerConfig.getIp(),zlmServerConfig.getHttpPort() );
  344. return;
  345. }
  346. mediaServerMapper.update(serverItem);
  347. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + zlmServerConfig.getGeneralMediaServerId();
  348. if (redisUtil.get(key) == null) {
  349. SsrcConfig ssrcConfig = new SsrcConfig(zlmServerConfig.getGeneralMediaServerId(), null, sipConfig.getDomain());
  350. serverItem.setSsrcConfig(ssrcConfig);
  351. }else {
  352. MediaServerItem mediaServerItemInRedis = (MediaServerItem)redisUtil.get(key);
  353. serverItem.setSsrcConfig(mediaServerItemInRedis.getSsrcConfig());
  354. }
  355. redisUtil.set(key, serverItem);
  356. resetOnlineServerItem(serverItem);
  357. updateMediaServerKeepalive(serverItem.getId(), null);
  358. if (serverItem.isAutoConfig()) {
  359. setZLMConfig(serverItem, "0".equals(zlmServerConfig.getHookEnable()));
  360. }
  361. publisher.zlmOnlineEventPublish(serverItem.getId());
  362. logger.info("[ZLM] 连接成功 {} - {}:{} ",
  363. zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(), zlmServerConfig.getHttpPort());
  364. }
  365. @Override
  366. public void zlmServerOffline(String mediaServerId) {
  367. delete(mediaServerId);
  368. }
  369. @Override
  370. public void resetOnlineServerItem(MediaServerItem serverItem) {
  371. // 更新缓存
  372. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  373. // 使用zset的分数作为当前并发量, 默认值设置为0
  374. if (redisUtil.zScore(key, serverItem.getId()) == null) { // 不存在则设置默认值 已存在则重置
  375. redisUtil.zAdd(key, serverItem.getId(), 0L);
  376. // 查询服务流数量
  377. zlmresTfulUtils.getMediaList(serverItem, null, null, "rtmp",(mediaList ->{
  378. Integer code = mediaList.getInteger("code");
  379. if (code == 0) {
  380. JSONArray data = mediaList.getJSONArray("data");
  381. if (data != null) {
  382. redisUtil.zAdd(key, serverItem.getId(), data.size());
  383. }
  384. }
  385. }));
  386. }else {
  387. clearRTPServer(serverItem);
  388. }
  389. }
  390. @Override
  391. public void addCount(String mediaServerId) {
  392. if (mediaServerId == null) {
  393. return;
  394. }
  395. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  396. redisUtil.zIncrScore(key, mediaServerId, 1);
  397. }
  398. @Override
  399. public void removeCount(String mediaServerId) {
  400. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  401. redisUtil.zIncrScore(key, mediaServerId, - 1);
  402. }
  403. /**
  404. * 获取负载最低的节点
  405. * @return MediaServerItem
  406. */
  407. @Override
  408. public MediaServerItem getMediaServerForMinimumLoad() {
  409. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  410. if (redisUtil.zSize(key) == null || redisUtil.zSize(key) == 0) {
  411. logger.info("获取负载最低的节点时无在线节点");
  412. return null;
  413. }
  414. // 获取分数最低的,及并发最低的
  415. Set<Object> objects = redisUtil.ZRange(key, 0, -1);
  416. ArrayList<Object> mediaServerObjectS = new ArrayList<>(objects);
  417. String mediaServerId = (String)mediaServerObjectS.get(0);
  418. return getOne(mediaServerId);
  419. }
  420. /**
  421. * 对zlm服务器进行基础配置
  422. * @param mediaServerItem 服务ID
  423. * @param restart 是否重启zlm
  424. */
  425. @Override
  426. public void setZLMConfig(MediaServerItem mediaServerItem, boolean restart) {
  427. logger.info("[ZLM] 正在设置 :{} -> {}:{}",
  428. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  429. String protocol = sslEnabled ? "https" : "http";
  430. String hookPrex = String.format("%s://%s:%s/index/hook", protocol, mediaServerItem.getHookIp(), serverPort);
  431. String recordHookPrex = null;
  432. if (mediaServerItem.getRecordAssistPort() != 0) {
  433. recordHookPrex = String.format("http://127.0.0.1:%s/api/record", mediaServerItem.getRecordAssistPort());
  434. }
  435. Map<String, Object> param = new HashMap<>();
  436. param.put("api.secret",mediaServerItem.getSecret()); // -profile:v Baseline
  437. param.put("ffmpeg.cmd","%s -fflags nobuffer -i %s -c:a aac -strict -2 -ar 44100 -ab 48k -c:v libx264 -f flv %s");
  438. param.put("hook.enable","1");
  439. param.put("hook.on_flow_report","");
  440. param.put("hook.on_play",String.format("%s/on_play", hookPrex));
  441. param.put("hook.on_http_access","");
  442. param.put("hook.on_publish", String.format("%s/on_publish", hookPrex));
  443. param.put("hook.on_record_mp4",recordHookPrex != null? String.format("%s/on_record_mp4", recordHookPrex): "");
  444. param.put("hook.on_record_ts","");
  445. param.put("hook.on_rtsp_auth","");
  446. param.put("hook.on_rtsp_realm","");
  447. param.put("hook.on_server_started",String.format("%s/on_server_started", hookPrex));
  448. param.put("hook.on_shell_login",String.format("%s/on_shell_login", hookPrex));
  449. param.put("hook.on_stream_changed",String.format("%s/on_stream_changed", hookPrex));
  450. param.put("hook.on_stream_none_reader",String.format("%s/on_stream_none_reader", hookPrex));
  451. param.put("hook.on_stream_not_found",String.format("%s/on_stream_not_found", hookPrex));
  452. param.put("hook.on_server_keepalive",String.format("%s/on_server_keepalive", hookPrex));
  453. param.put("hook.timeoutSec","20");
  454. param.put("general.streamNoneReaderDelayMS",mediaServerItem.getStreamNoneReaderDelayMS()==-1?"3600000":mediaServerItem.getStreamNoneReaderDelayMS() );
  455. // 推流断开后可以在超时时间内重新连接上继续推流,这样播放器会接着播放。
  456. // 置0关闭此特性(推流断开会导致立即断开播放器)
  457. // 此参数不应大于播放器超时时间
  458. // 优化此消息以更快的收到流注销事件
  459. param.put("general.continue_push_ms", "3000" );
  460. // 最多等待未初始化的Track时间,单位毫秒,超时之后会忽略未初始化的Track, 设置此选项优化那些音频错误的不规范流,
  461. // 等zlm支持给每个rtpServer设置关闭音频的时候可以不设置此选项
  462. param.put("general.wait_track_ready_ms", "3000" );
  463. if (mediaServerItem.isRtpEnable() && !StringUtils.isEmpty(mediaServerItem.getRtpPortRange())) {
  464. param.put("rtp_proxy.port_range", mediaServerItem.getRtpPortRange().replace(",", "-"));
  465. }
  466. JSONObject responseJSON = zlmresTfulUtils.setServerConfig(mediaServerItem, param);
  467. if (responseJSON != null && responseJSON.getInteger("code") == 0) {
  468. if (restart) {
  469. logger.info("[ZLM] 设置成功,开始重启以保证配置生效 {} -> {}:{}",
  470. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  471. zlmresTfulUtils.restartServer(mediaServerItem);
  472. }else {
  473. logger.info("[ZLM] 设置成功 {} -> {}:{}",
  474. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  475. }
  476. }else {
  477. logger.info("[ZLM] 设置zlm失败 {} -> {}:{}",
  478. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  479. }
  480. }
  481. @Override
  482. public WVPResult<MediaServerItem> checkMediaServer(String ip, int port, String secret) {
  483. WVPResult<MediaServerItem> result = new WVPResult<>();
  484. if (mediaServerMapper.queryOneByHostAndPort(ip, port) != null) {
  485. result.setCode(-1);
  486. result.setMsg("此连接已存在");
  487. return result;
  488. }
  489. MediaServerItem mediaServerItem = new MediaServerItem();
  490. mediaServerItem.setIp(ip);
  491. mediaServerItem.setHttpPort(port);
  492. mediaServerItem.setSecret(secret);
  493. JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig(mediaServerItem);
  494. if (responseJSON == null) {
  495. result.setCode(-1);
  496. result.setMsg("连接失败");
  497. return result;
  498. }
  499. JSONArray data = responseJSON.getJSONArray("data");
  500. ZLMServerConfig zlmServerConfig = JSON.parseObject(JSON.toJSONString(data.get(0)), ZLMServerConfig.class);
  501. if (zlmServerConfig == null) {
  502. result.setCode(-1);
  503. result.setMsg("读取配置失败");
  504. return result;
  505. }
  506. if (mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId()) != null) {
  507. result.setCode(-1);
  508. result.setMsg("媒体服务ID [" + zlmServerConfig.getGeneralMediaServerId() + " ] 已存在,请修改媒体服务器配置");
  509. return result;
  510. }
  511. mediaServerItem.setHttpSSlPort(zlmServerConfig.getHttpPort());
  512. mediaServerItem.setRtmpPort(zlmServerConfig.getRtmpPort());
  513. mediaServerItem.setRtmpSSlPort(zlmServerConfig.getRtmpSslPort());
  514. mediaServerItem.setRtspPort(zlmServerConfig.getRtspPort());
  515. mediaServerItem.setRtspSSLPort(zlmServerConfig.getRtspSSlport());
  516. mediaServerItem.setRtpProxyPort(zlmServerConfig.getRtpProxyPort());
  517. mediaServerItem.setStreamIp(ip);
  518. mediaServerItem.setHookIp(sipConfig.getIp());
  519. mediaServerItem.setSdpIp(ip);
  520. mediaServerItem.setStreamNoneReaderDelayMS(zlmServerConfig.getGeneralStreamNoneReaderDelayMS());
  521. result.setCode(0);
  522. result.setMsg("成功");
  523. result.setData(mediaServerItem);
  524. return result;
  525. }
  526. @Override
  527. public boolean checkMediaRecordServer(String ip, int port) {
  528. boolean result = false;
  529. OkHttpClient client = new OkHttpClient();
  530. String url = String.format("http://%s:%s/index/api/record", ip, port);
  531. FormBody.Builder builder = new FormBody.Builder();
  532. Request request = new Request.Builder()
  533. .get()
  534. .url(url)
  535. .build();
  536. try {
  537. Response response = client.newCall(request).execute();
  538. if (response != null) {
  539. result = true;
  540. }
  541. } catch (Exception e) {}
  542. return result;
  543. }
  544. @Override
  545. public void delete(String id) {
  546. redisUtil.zRemove(VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId(), id);
  547. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + id;
  548. redisUtil.del(key);
  549. }
  550. @Override
  551. public void deleteDb(String id){
  552. //同步删除数据库中的数据
  553. mediaServerMapper.delOne(id);
  554. }
  555. @Override
  556. public void updateMediaServerKeepalive(String mediaServerId, JSONObject data) {
  557. MediaServerItem mediaServerItem = getOne(mediaServerId);
  558. if (mediaServerItem == null) {
  559. // zlm连接重试
  560. logger.warn("[更新ZLM 保活信息]失败,未找到流媒体信息");
  561. return;
  562. }
  563. String key = VideoManagerConstants.MEDIA_SERVER_KEEPALIVE_PREFIX + userSetting.getServerId() + "_" + mediaServerId;
  564. int hookAliveInterval = mediaServerItem.getHookAliveInterval() + 2;
  565. redisUtil.set(key, data, hookAliveInterval);
  566. }
  567. @Override
  568. public void syncCatchFromDatabase() {
  569. List<MediaServerItem> allInCatch = getAll();
  570. List<MediaServerItem> allInDatabase = mediaServerMapper.queryAll();
  571. Map<String, MediaServerItem> mediaServerItemMap = new HashMap<>();
  572. for (MediaServerItem mediaServerItem : allInDatabase) {
  573. mediaServerItemMap.put(mediaServerItem.getId(), mediaServerItem);
  574. }
  575. for (MediaServerItem mediaServerItem : allInCatch) {
  576. if (mediaServerItemMap.get(mediaServerItem) == null) {
  577. delete(mediaServerItem.getId());
  578. }
  579. }
  580. }
  581. }