MediaServerServiceImpl.java 27 KB

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