MediaServerServiceImpl.java 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657
  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.MediaConfig;
  7. import com.genersoft.iot.vmp.conf.SipConfig;
  8. import com.genersoft.iot.vmp.conf.UserSetup;
  9. import com.genersoft.iot.vmp.gb28181.bean.Device;
  10. import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
  11. import com.genersoft.iot.vmp.gb28181.session.SsrcConfig;
  12. import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
  13. import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
  14. import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
  15. import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig;
  16. import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
  17. import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem;
  18. import com.genersoft.iot.vmp.service.IMediaServerService;
  19. import com.genersoft.iot.vmp.service.IStreamProxyService;
  20. import com.genersoft.iot.vmp.service.bean.SSRCInfo;
  21. import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
  22. import com.genersoft.iot.vmp.storager.dao.MediaServerMapper;
  23. import com.genersoft.iot.vmp.utils.redis.JedisUtil;
  24. import com.genersoft.iot.vmp.utils.redis.RedisUtil;
  25. import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
  26. import okhttp3.*;
  27. import org.slf4j.Logger;
  28. import org.slf4j.LoggerFactory;
  29. import org.springframework.beans.factory.annotation.Autowired;
  30. import org.springframework.beans.factory.annotation.Value;
  31. import org.springframework.boot.CommandLineRunner;
  32. import org.springframework.core.annotation.Order;
  33. import org.springframework.jdbc.datasource.DataSourceTransactionManager;
  34. import org.springframework.security.core.parameters.P;
  35. import org.springframework.stereotype.Service;
  36. import org.springframework.transaction.TransactionDefinition;
  37. import org.springframework.transaction.TransactionStatus;
  38. import org.springframework.util.StringUtils;
  39. import java.text.ParseException;
  40. import java.text.SimpleDateFormat;
  41. import java.util.*;
  42. /**
  43. * 媒体服务器节点管理
  44. */
  45. @Service
  46. public class MediaServerServiceImpl implements IMediaServerService {
  47. private final static Logger logger = LoggerFactory.getLogger(MediaServerServiceImpl.class);
  48. @Autowired
  49. private SipConfig sipConfig;
  50. @Value("${server.ssl.enabled:false}")
  51. private boolean sslEnabled;
  52. @Value("${server.port}")
  53. private Integer serverPort;
  54. @Autowired
  55. private UserSetup userSetup;
  56. @Autowired
  57. private ZLMRESTfulUtils zlmresTfulUtils;
  58. @Autowired
  59. private MediaServerMapper mediaServerMapper;
  60. @Autowired
  61. DataSourceTransactionManager dataSourceTransactionManager;
  62. @Autowired
  63. TransactionDefinition transactionDefinition;
  64. @Autowired
  65. private VideoStreamSessionManager streamSession;
  66. @Autowired
  67. private ZLMRTPServerFactory zlmrtpServerFactory;
  68. @Autowired
  69. private RedisUtil redisUtil;
  70. @Autowired
  71. private IVideoManagerStorager storager;
  72. @Autowired
  73. private IStreamProxyService streamProxyService;
  74. @Autowired
  75. private EventPublisher publisher;
  76. @Autowired
  77. JedisUtil jedisUtil;
  78. private final SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
  79. /**
  80. * 初始化
  81. */
  82. @Override
  83. public void updateVmServer(List<MediaServerItem> mediaServerItemList) {
  84. logger.info("[缓存初始化] Media Server ");
  85. for (MediaServerItem mediaServerItem : mediaServerItemList) {
  86. if (StringUtils.isEmpty(mediaServerItem.getId())) {
  87. continue;
  88. }
  89. // 更新
  90. if (mediaServerItem.getSsrcConfig() == null) {
  91. SsrcConfig ssrcConfig = new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain());
  92. mediaServerItem.setSsrcConfig(ssrcConfig);
  93. redisUtil.set(VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerItem.getId(), mediaServerItem);
  94. }
  95. // 查询redis是否存在此mediaServer
  96. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerItem.getId();
  97. if (!redisUtil.hasKey(key)) {
  98. redisUtil.set(key, mediaServerItem);
  99. }
  100. }
  101. }
  102. @Override
  103. public SSRCInfo openRTPServer(MediaServerItem mediaServerItem, String streamId) {
  104. return openRTPServer(mediaServerItem, streamId, false);
  105. }
  106. @Override
  107. public SSRCInfo openRTPServer(MediaServerItem mediaServerItem, String streamId, boolean isPlayback) {
  108. if (mediaServerItem == null || mediaServerItem.getId() == null) {
  109. return null;
  110. }
  111. // 获取mediaServer可用的ssrc
  112. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerItem.getId();
  113. SsrcConfig ssrcConfig = mediaServerItem.getSsrcConfig();
  114. if (ssrcConfig == null) {
  115. logger.info("media server [ {} ] ssrcConfig is null", mediaServerItem.getId());
  116. return null;
  117. }else {
  118. String ssrc = null;
  119. if (isPlayback) {
  120. ssrc = ssrcConfig.getPlayBackSsrc();
  121. }else {
  122. ssrc = ssrcConfig.getPlaySsrc();
  123. }
  124. if (streamId == null) {
  125. streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase();
  126. }
  127. int rtpServerPort = mediaServerItem.getRtpProxyPort();
  128. if (mediaServerItem.isRtpEnable()) {
  129. rtpServerPort = zlmrtpServerFactory.createRTPServer(mediaServerItem, streamId);
  130. }
  131. redisUtil.set(key, mediaServerItem);
  132. return new SSRCInfo(rtpServerPort, ssrc, streamId);
  133. }
  134. }
  135. @Override
  136. public void closeRTPServer(String deviceId, String channelId, String stream) {
  137. String mediaServerId = streamSession.getMediaServerId(deviceId, channelId, stream);
  138. String ssrc = streamSession.getSSRC(deviceId, channelId, stream);
  139. MediaServerItem mediaServerItem = this.getOne(mediaServerId);
  140. if (mediaServerItem != null) {
  141. String streamId = String.format("%s_%s", deviceId, channelId);
  142. zlmrtpServerFactory.closeRTPServer(mediaServerItem, streamId);
  143. releaseSsrc(mediaServerItem.getId(), ssrc);
  144. }
  145. streamSession.remove(deviceId, channelId, stream);
  146. }
  147. @Override
  148. public void releaseSsrc(String mediaServerItemId, String ssrc) {
  149. MediaServerItem mediaServerItem = getOne(mediaServerItemId);
  150. if (mediaServerItem == null || ssrc == null) {
  151. return;
  152. }
  153. SsrcConfig ssrcConfig = mediaServerItem.getSsrcConfig();
  154. ssrcConfig.releaseSsrc(ssrc);
  155. mediaServerItem.setSsrcConfig(ssrcConfig);
  156. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerItem.getId();
  157. redisUtil.set(key, mediaServerItem);
  158. }
  159. /**
  160. * zlm 重启后重置他的推流信息, TODO 给正在使用的设备发送停止命令
  161. */
  162. @Override
  163. public void clearRTPServer(MediaServerItem mediaServerItem) {
  164. mediaServerItem.setSsrcConfig(new SsrcConfig(mediaServerItem.getId(), null, sipConfig.getDomain()));
  165. redisUtil.zAdd(VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetup.getServerId(), mediaServerItem.getId(), 0);
  166. }
  167. @Override
  168. public void update(MediaServerItem mediaSerItem) {
  169. mediaServerMapper.update(mediaSerItem);
  170. MediaServerItem mediaServerItemInRedis = getOne(mediaSerItem.getId());
  171. MediaServerItem mediaServerItemInDataBase = mediaServerMapper.queryOne(mediaSerItem.getId());
  172. if (mediaServerItemInRedis != null && mediaServerItemInRedis.getSsrcConfig() != null) {
  173. mediaServerItemInDataBase.setSsrcConfig(mediaServerItemInRedis.getSsrcConfig());
  174. }else {
  175. mediaServerItemInDataBase.setSsrcConfig(
  176. new SsrcConfig(
  177. mediaServerItemInDataBase.getId(),
  178. null,
  179. sipConfig.getDomain()
  180. )
  181. );
  182. }
  183. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerItemInDataBase.getId();
  184. redisUtil.set(key, mediaServerItemInDataBase);
  185. }
  186. @Override
  187. public List<MediaServerItem> getAll() {
  188. List<MediaServerItem> result = new ArrayList<>();
  189. List<Object> mediaServerKeys = redisUtil.scan(String.format("%S*", VideoManagerConstants.MEDIA_SERVER_PREFIX+ userSetup.getServerId() + "_" ));
  190. String onlineKey = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetup.getServerId();
  191. for (Object mediaServerKey : mediaServerKeys) {
  192. String key = (String) mediaServerKey;
  193. MediaServerItem mediaServerItem = (MediaServerItem) redisUtil.get(key);
  194. // 检查状态
  195. Double aDouble = redisUtil.zScore(onlineKey, mediaServerItem.getId());
  196. if (aDouble != null) {
  197. mediaServerItem.setStatus(true);
  198. }
  199. result.add(mediaServerItem);
  200. }
  201. result.sort((serverItem1, serverItem2)->{
  202. int sortResult = 0;
  203. try {
  204. sortResult = format.parse(serverItem1.getCreateTime()).compareTo(format.parse(serverItem2.getCreateTime()));
  205. } catch (ParseException e) {
  206. e.printStackTrace();
  207. }
  208. return sortResult;
  209. });
  210. return result;
  211. }
  212. @Override
  213. public List<MediaServerItem> getAllFromDatabase() {
  214. return mediaServerMapper.queryAll();
  215. }
  216. @Override
  217. public List<MediaServerItem> getAllOnline() {
  218. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetup.getServerId();
  219. Set<String> mediaServerIdSet = redisUtil.zRevRange(key, 0, -1);
  220. List<MediaServerItem> result = new ArrayList<>();
  221. if (mediaServerIdSet != null && mediaServerIdSet.size() > 0) {
  222. for (String mediaServerId : mediaServerIdSet) {
  223. String serverKey = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerId;
  224. result.add((MediaServerItem) redisUtil.get(serverKey));
  225. }
  226. }
  227. Collections.reverse(result);
  228. return result;
  229. }
  230. /**
  231. * 获取单个zlm服务器
  232. * @param mediaServerId 服务id
  233. * @return MediaServerItem
  234. */
  235. @Override
  236. public MediaServerItem getOne(String mediaServerId) {
  237. if (mediaServerId == null) {
  238. return null;
  239. }
  240. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + mediaServerId;
  241. return (MediaServerItem)redisUtil.get(key);
  242. }
  243. @Override
  244. public MediaServerItem getDefaultMediaServer() {
  245. return mediaServerMapper.queryDefault();
  246. }
  247. @Override
  248. public void clearMediaServerForOnline() {
  249. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetup.getServerId();
  250. redisUtil.del(key);
  251. }
  252. @Override
  253. public WVPResult<String> add(MediaServerItem mediaServerItem) {
  254. WVPResult<String> result = new WVPResult<>();
  255. mediaServerItem.setCreateTime(this.format.format(System.currentTimeMillis()));
  256. mediaServerItem.setUpdateTime(this.format.format(System.currentTimeMillis()));
  257. mediaServerItem.setHookAliveInterval(120);
  258. JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig(mediaServerItem);
  259. if (responseJSON != null) {
  260. JSONArray data = responseJSON.getJSONArray("data");
  261. if (data != null && data.size() > 0) {
  262. ZLMServerConfig zlmServerConfig= JSON.parseObject(JSON.toJSONString(data.get(0)), ZLMServerConfig.class);
  263. if (mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId()) != null) {
  264. result.setCode(-1);
  265. result.setMsg("保存失败,媒体服务ID [ " + zlmServerConfig.getGeneralMediaServerId() + " ] 已存在,请修改媒体服务器配置");
  266. return result;
  267. }
  268. mediaServerItem.setId(zlmServerConfig.getGeneralMediaServerId());
  269. zlmServerConfig.setIp(mediaServerItem.getIp());
  270. mediaServerMapper.add(mediaServerItem);
  271. zlmServerOnline(zlmServerConfig);
  272. result.setCode(0);
  273. result.setMsg("success");
  274. }else {
  275. result.setCode(-1);
  276. result.setMsg("连接失败");
  277. }
  278. }else {
  279. result.setCode(-1);
  280. result.setMsg("连接失败");
  281. }
  282. return result;
  283. }
  284. @Override
  285. public int addToDatabase(MediaServerItem mediaSerItem) {
  286. return mediaServerMapper.add(mediaSerItem);
  287. }
  288. @Override
  289. public int updateToDatabase(MediaServerItem mediaSerItem) {
  290. int result = 0;
  291. if (mediaSerItem.isDefaultServer()) {
  292. TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
  293. int delResult = mediaServerMapper.delDefault();
  294. if (delResult == 0) {
  295. logger.error("移除数据库默认zlm节点失败");
  296. //事务回滚
  297. dataSourceTransactionManager.rollback(transactionStatus);
  298. return 0;
  299. }
  300. result = mediaServerMapper.add(mediaSerItem);
  301. dataSourceTransactionManager.commit(transactionStatus); //手动提交
  302. }else {
  303. result = mediaServerMapper.update(mediaSerItem);
  304. }
  305. return result;
  306. }
  307. /**
  308. * 处理zlm上线
  309. * @param zlmServerConfig zlm上线携带的参数
  310. */
  311. @Override
  312. public void zlmServerOnline(ZLMServerConfig zlmServerConfig) {
  313. logger.info("[ ZLM:{} ]-[ {}:{} ]正在连接",
  314. zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(), zlmServerConfig.getHttpPort());
  315. MediaServerItem serverItem = mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId());
  316. if (serverItem == null) {
  317. logger.warn("[未注册的zlm] 拒接接入:{}来自{}:{}", zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(),zlmServerConfig.getHttpPort() );
  318. logger.warn("请检查ZLM的<general.mediaServerId>配置是否与WVP的<media.id>一致");
  319. return;
  320. }
  321. serverItem.setHookAliveInterval(zlmServerConfig.getHookAliveInterval());
  322. if (serverItem.getHttpPort() == 0) {
  323. serverItem.setHttpPort(zlmServerConfig.getHttpPort());
  324. }
  325. if (serverItem.getHttpSSlPort() == 0) {
  326. serverItem.setHttpSSlPort(zlmServerConfig.getHttpSSLport());
  327. }
  328. if (serverItem.getRtmpPort() == 0) {
  329. serverItem.setRtmpPort(zlmServerConfig.getRtmpPort());
  330. }
  331. if (serverItem.getRtmpSSlPort() == 0) {
  332. serverItem.setRtmpSSlPort(zlmServerConfig.getRtmpSslPort());
  333. }
  334. if (serverItem.getRtspPort() == 0) {
  335. serverItem.setRtspPort(zlmServerConfig.getRtspPort());
  336. }
  337. if (serverItem.getRtspSSLPort() == 0) {
  338. serverItem.setRtspSSLPort(zlmServerConfig.getRtspSSlport());
  339. }
  340. if (serverItem.getRtpProxyPort() == 0) {
  341. serverItem.setRtpProxyPort(zlmServerConfig.getRtpProxyPort());
  342. }
  343. serverItem.setStatus(true);
  344. if (StringUtils.isEmpty(serverItem.getId())) {
  345. logger.warn("[未注册的zlm] serverItem缺少ID, 无法接入:{}:{}", zlmServerConfig.getIp(),zlmServerConfig.getHttpPort() );
  346. return;
  347. }
  348. mediaServerMapper.update(serverItem);
  349. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + zlmServerConfig.getGeneralMediaServerId();
  350. if (redisUtil.get(key) == null) {
  351. SsrcConfig ssrcConfig = new SsrcConfig(zlmServerConfig.getGeneralMediaServerId(), null, sipConfig.getDomain());
  352. serverItem.setSsrcConfig(ssrcConfig);
  353. }else {
  354. MediaServerItem mediaServerItemInRedis = (MediaServerItem)redisUtil.get(key);
  355. serverItem.setSsrcConfig(mediaServerItemInRedis.getSsrcConfig());
  356. }
  357. redisUtil.set(key, serverItem);
  358. resetOnlineServerItem(serverItem);
  359. updateMediaServerKeepalive(serverItem.getId(), null);
  360. setZLMConfig(serverItem, "0".equals(zlmServerConfig.getHookEnable()));
  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 + userSetup.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 + userSetup.getServerId();
  396. redisUtil.zIncrScore(key, mediaServerId, 1);
  397. }
  398. @Override
  399. public void removeCount(String mediaServerId) {
  400. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetup.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 + userSetup.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:{} ]-[ {}:{} ]正在设置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. JSONObject responseJSON = zlmresTfulUtils.setServerConfig(mediaServerItem, param);
  464. if (responseJSON != null && responseJSON.getInteger("code") == 0) {
  465. if (restart) {
  466. logger.info("[ ZLM:{} ]-[ {}:{} ]设置zlm成功, 开始重启以保证配置生效",
  467. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  468. zlmresTfulUtils.restartServer(mediaServerItem);
  469. }else {
  470. logger.info("[ ZLM:{} ]-[ {}:{} ]设置zlm成功",
  471. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  472. }
  473. }else {
  474. logger.info("[ ZLM:{} ]-[ {}:{} ]设置zlm失败",
  475. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  476. }
  477. }
  478. @Override
  479. public WVPResult<MediaServerItem> checkMediaServer(String ip, int port, String secret) {
  480. WVPResult<MediaServerItem> result = new WVPResult<>();
  481. if (mediaServerMapper.queryOneByHostAndPort(ip, port) != null) {
  482. result.setCode(-1);
  483. result.setMsg("此连接已存在");
  484. return result;
  485. }
  486. MediaServerItem mediaServerItem = new MediaServerItem();
  487. mediaServerItem.setIp(ip);
  488. mediaServerItem.setHttpPort(port);
  489. mediaServerItem.setSecret(secret);
  490. JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig(mediaServerItem);
  491. if (responseJSON == null) {
  492. result.setCode(-1);
  493. result.setMsg("连接失败");
  494. return result;
  495. }
  496. JSONArray data = responseJSON.getJSONArray("data");
  497. ZLMServerConfig zlmServerConfig = JSON.parseObject(JSON.toJSONString(data.get(0)), ZLMServerConfig.class);
  498. if (zlmServerConfig == null) {
  499. result.setCode(-1);
  500. result.setMsg("读取配置失败");
  501. return result;
  502. }
  503. if (mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId()) != null) {
  504. result.setCode(-1);
  505. result.setMsg("媒体服务ID [" + zlmServerConfig.getGeneralMediaServerId() + " ] 已存在,请修改媒体服务器配置");
  506. return result;
  507. }
  508. mediaServerItem.setHttpSSlPort(zlmServerConfig.getHttpPort());
  509. mediaServerItem.setRtmpPort(zlmServerConfig.getRtmpPort());
  510. mediaServerItem.setRtmpSSlPort(zlmServerConfig.getRtmpSslPort());
  511. mediaServerItem.setRtspPort(zlmServerConfig.getRtspPort());
  512. mediaServerItem.setRtspSSLPort(zlmServerConfig.getRtspSSlport());
  513. mediaServerItem.setRtpProxyPort(zlmServerConfig.getRtpProxyPort());
  514. mediaServerItem.setStreamIp(ip);
  515. mediaServerItem.setHookIp(sipConfig.getIp());
  516. mediaServerItem.setSdpIp(ip);
  517. mediaServerItem.setStreamNoneReaderDelayMS(zlmServerConfig.getGeneralStreamNoneReaderDelayMS());
  518. result.setCode(0);
  519. result.setMsg("成功");
  520. result.setData(mediaServerItem);
  521. return result;
  522. }
  523. @Override
  524. public boolean checkMediaRecordServer(String ip, int port) {
  525. boolean result = false;
  526. OkHttpClient client = new OkHttpClient();
  527. String url = String.format("http://%s:%s/index/api/record", ip, port);
  528. FormBody.Builder builder = new FormBody.Builder();
  529. Request request = new Request.Builder()
  530. .get()
  531. .url(url)
  532. .build();
  533. try {
  534. Response response = client.newCall(request).execute();
  535. if (response != null) {
  536. result = true;
  537. }
  538. } catch (Exception e) {}
  539. return result;
  540. }
  541. @Override
  542. public void delete(String id) {
  543. redisUtil.zRemove(VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetup.getServerId(), id);
  544. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetup.getServerId() + "_" + id;
  545. redisUtil.del(key);
  546. }
  547. @Override
  548. public void deleteDb(String id){
  549. //同步删除数据库中的数据
  550. mediaServerMapper.delOne(id);
  551. }
  552. @Override
  553. public void updateMediaServerKeepalive(String mediaServerId, JSONObject data) {
  554. MediaServerItem mediaServerItem = getOne(mediaServerId);
  555. if (mediaServerItem == null) {
  556. // zlm连接重试
  557. logger.warn("[更新ZLM 保活信息]失败,未找到流媒体信息");
  558. return;
  559. }
  560. String key = VideoManagerConstants.MEDIA_SERVER_KEEPALIVE_PREFIX + userSetup.getServerId() + "_" + mediaServerId;
  561. int hookAliveInterval = mediaServerItem.getHookAliveInterval() + 2;
  562. redisUtil.set(key, data, hookAliveInterval);
  563. }
  564. @Override
  565. public void syncCatchFromDatabase() {
  566. List<MediaServerItem> allInCatch = getAll();
  567. List<MediaServerItem> allInDatabase = mediaServerMapper.queryAll();
  568. Map<String, MediaServerItem> mediaServerItemMap = new HashMap<>();
  569. for (MediaServerItem mediaServerItem : allInDatabase) {
  570. mediaServerItemMap.put(mediaServerItem.getId(), mediaServerItem);
  571. }
  572. for (MediaServerItem mediaServerItem : allInCatch) {
  573. if (mediaServerItemMap.get(mediaServerItem) == null) {
  574. delete(mediaServerItem.getId());
  575. }
  576. }
  577. }
  578. }