MediaServerServiceImpl.java 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743
  1. package com.genersoft.iot.vmp.service.impl;
  2. import com.alibaba.fastjson2.JSON;
  3. import com.alibaba.fastjson2.JSONArray;
  4. import com.alibaba.fastjson2.JSONObject;
  5. import com.genersoft.iot.vmp.common.CommonCallback;
  6. import com.genersoft.iot.vmp.common.VideoManagerConstants;
  7. import com.genersoft.iot.vmp.conf.DynamicTask;
  8. import com.genersoft.iot.vmp.conf.SipConfig;
  9. import com.genersoft.iot.vmp.conf.UserSetting;
  10. import com.genersoft.iot.vmp.conf.exception.ControllerException;
  11. import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
  12. import com.genersoft.iot.vmp.gb28181.session.SSRCFactory;
  13. import com.genersoft.iot.vmp.media.zlm.AssistRESTfulUtils;
  14. import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
  15. import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
  16. import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig;
  17. import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
  18. import com.genersoft.iot.vmp.media.zlm.dto.ServerKeepaliveData;
  19. import com.genersoft.iot.vmp.service.IMediaServerService;
  20. import com.genersoft.iot.vmp.service.bean.MediaServerLoad;
  21. import com.genersoft.iot.vmp.service.bean.SSRCInfo;
  22. import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
  23. import com.genersoft.iot.vmp.storager.dao.MediaServerMapper;
  24. import com.genersoft.iot.vmp.utils.DateUtil;
  25. import com.genersoft.iot.vmp.utils.JsonUtil;
  26. import com.genersoft.iot.vmp.utils.redis.RedisUtil;
  27. import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
  28. import okhttp3.OkHttpClient;
  29. import okhttp3.Request;
  30. import okhttp3.Response;
  31. import org.slf4j.Logger;
  32. import org.slf4j.LoggerFactory;
  33. import org.springframework.beans.factory.annotation.Autowired;
  34. import org.springframework.beans.factory.annotation.Value;
  35. import org.springframework.data.redis.core.RedisTemplate;
  36. import org.springframework.jdbc.datasource.DataSourceTransactionManager;
  37. import org.springframework.stereotype.Service;
  38. import org.springframework.transaction.TransactionDefinition;
  39. import org.springframework.transaction.TransactionStatus;
  40. import org.springframework.util.ObjectUtils;
  41. import java.io.File;
  42. import java.time.LocalDateTime;
  43. import java.util.*;
  44. /**
  45. * 媒体服务器节点管理
  46. */
  47. @Service
  48. public class MediaServerServiceImpl implements IMediaServerService {
  49. private final static Logger logger = LoggerFactory.getLogger(MediaServerServiceImpl.class);
  50. private final String zlmKeepaliveKeyPrefix = "zlm-keepalive_";
  51. @Autowired
  52. private SipConfig sipConfig;
  53. @Autowired
  54. private SSRCFactory ssrcFactory;
  55. @Value("${server.ssl.enabled:false}")
  56. private boolean sslEnabled;
  57. @Value("${server.port}")
  58. private Integer serverPort;
  59. @Autowired
  60. private UserSetting userSetting;
  61. @Autowired
  62. private AssistRESTfulUtils assistRESTfulUtils;
  63. @Autowired
  64. private ZLMRESTfulUtils zlmresTfulUtils;
  65. @Autowired
  66. private MediaServerMapper mediaServerMapper;
  67. @Autowired
  68. private DataSourceTransactionManager dataSourceTransactionManager;
  69. @Autowired
  70. private TransactionDefinition transactionDefinition;
  71. @Autowired
  72. private ZLMRTPServerFactory zlmrtpServerFactory;
  73. @Autowired
  74. private EventPublisher publisher;
  75. @Autowired
  76. private DynamicTask dynamicTask;
  77. @Autowired
  78. private IRedisCatchStorage redisCatchStorage;
  79. @Autowired
  80. private RedisTemplate<Object, Object> redisTemplate;
  81. /**
  82. * 初始化
  83. */
  84. @Override
  85. public void updateVmServer(List<MediaServerItem> mediaServerItemList) {
  86. logger.info("[zlm] 缓存初始化 ");
  87. for (MediaServerItem mediaServerItem : mediaServerItemList) {
  88. if (ObjectUtils.isEmpty(mediaServerItem.getId())) {
  89. continue;
  90. }
  91. // 更新
  92. if (ssrcFactory.hasMediaServerSSRC(mediaServerItem.getId())) {
  93. ssrcFactory.initMediaServerSSRC(mediaServerItem.getId(), null);
  94. }
  95. // 查询redis是否存在此mediaServer
  96. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId();
  97. Boolean hasKey = redisTemplate.hasKey(key);
  98. if (hasKey != null && ! hasKey) {
  99. redisTemplate.opsForValue().set(key, mediaServerItem);
  100. }
  101. }
  102. }
  103. @Override
  104. public SSRCInfo openRTPServer(MediaServerItem mediaServerItem, String streamId, String presetSsrc, boolean ssrcCheck,
  105. boolean isPlayback, Integer port, Boolean reUsePort, Integer tcpMode) {
  106. if (mediaServerItem == null || mediaServerItem.getId() == null) {
  107. logger.info("[openRTPServer] 失败, mediaServerItem == null || mediaServerItem.getId() == null");
  108. return null;
  109. }
  110. // 获取mediaServer可用的ssrc
  111. String ssrc;
  112. if (presetSsrc != null) {
  113. ssrc = presetSsrc;
  114. }else {
  115. if (isPlayback) {
  116. ssrc = ssrcFactory.getPlayBackSsrc(mediaServerItem.getId());
  117. }else {
  118. ssrc = ssrcFactory.getPlaySsrc(mediaServerItem.getId());
  119. }
  120. }
  121. if (streamId == null) {
  122. streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase();
  123. }
  124. int rtpServerPort;
  125. if (mediaServerItem.isRtpEnable()) {
  126. rtpServerPort = zlmrtpServerFactory.createRTPServer(mediaServerItem, streamId, ssrcCheck?Integer.parseInt(ssrc):0, port, reUsePort, tcpMode);
  127. } else {
  128. rtpServerPort = mediaServerItem.getRtpProxyPort();
  129. }
  130. return new SSRCInfo(rtpServerPort, ssrc, streamId);
  131. }
  132. @Override
  133. public void closeRTPServer(MediaServerItem mediaServerItem, String streamId) {
  134. if (mediaServerItem == null) {
  135. return;
  136. }
  137. zlmrtpServerFactory.closeRtpServer(mediaServerItem, streamId);
  138. }
  139. @Override
  140. public void closeRTPServer(MediaServerItem mediaServerItem, String streamId, CommonCallback<Boolean> callback) {
  141. if (mediaServerItem == null) {
  142. callback.run(false);
  143. return;
  144. }
  145. zlmrtpServerFactory.closeRtpServer(mediaServerItem, streamId, callback);
  146. }
  147. @Override
  148. public void closeRTPServer(String mediaServerId, String streamId) {
  149. MediaServerItem mediaServerItem = this.getOne(mediaServerId);
  150. closeRTPServer(mediaServerItem, streamId);
  151. }
  152. @Override
  153. public Boolean updateRtpServerSSRC(MediaServerItem mediaServerItem, String streamId, String ssrc) {
  154. return zlmrtpServerFactory.updateRtpServerSSRC(mediaServerItem, streamId, ssrc);
  155. }
  156. @Override
  157. public void releaseSsrc(String mediaServerItemId, String ssrc) {
  158. MediaServerItem mediaServerItem = getOne(mediaServerItemId);
  159. if (mediaServerItem == null || ssrc == null) {
  160. return;
  161. }
  162. ssrcFactory.releaseSsrc(mediaServerItemId, ssrc);
  163. }
  164. /**
  165. * zlm 重启后重置他的推流信息, TODO 给正在使用的设备发送停止命令
  166. */
  167. @Override
  168. public void clearRTPServer(MediaServerItem mediaServerItem) {
  169. ssrcFactory.reset(mediaServerItem.getId());
  170. }
  171. @Override
  172. public void update(MediaServerItem mediaSerItem) {
  173. mediaServerMapper.update(mediaSerItem);
  174. MediaServerItem mediaServerItemInRedis = getOne(mediaSerItem.getId());
  175. MediaServerItem mediaServerItemInDataBase = mediaServerMapper.queryOne(mediaSerItem.getId());
  176. if (mediaServerItemInRedis == null || ssrcFactory.hasMediaServerSSRC(mediaSerItem.getId())) {
  177. ssrcFactory.initMediaServerSSRC(mediaServerItemInDataBase.getId(),null);
  178. }
  179. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItemInDataBase.getId();
  180. redisTemplate.opsForValue().set(key, mediaServerItemInDataBase);
  181. }
  182. @Override
  183. public List<MediaServerItem> getAll() {
  184. List<MediaServerItem> result = new ArrayList<>();
  185. List<Object> mediaServerKeys = RedisUtil.scan(redisTemplate, String.format("%S*", VideoManagerConstants.MEDIA_SERVER_PREFIX+ userSetting.getServerId() + "_" ));
  186. String onlineKey = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  187. for (Object mediaServerKey : mediaServerKeys) {
  188. String key = (String) mediaServerKey;
  189. MediaServerItem mediaServerItem = JsonUtil.redisJsonToObject(redisTemplate, key, MediaServerItem.class);
  190. if (Objects.isNull(mediaServerItem)) {
  191. continue;
  192. }
  193. // 检查状态
  194. Double aDouble = redisTemplate.opsForZSet().score(onlineKey, mediaServerItem.getId());
  195. if (aDouble != null) {
  196. mediaServerItem.setStatus(true);
  197. }
  198. result.add(mediaServerItem);
  199. }
  200. result.sort((serverItem1, serverItem2)->{
  201. int sortResult = 0;
  202. LocalDateTime localDateTime1 = LocalDateTime.parse(serverItem1.getCreateTime(), DateUtil.formatter);
  203. LocalDateTime localDateTime2 = LocalDateTime.parse(serverItem2.getCreateTime(), DateUtil.formatter);
  204. sortResult = localDateTime1.compareTo(localDateTime2);
  205. return sortResult;
  206. });
  207. return result;
  208. }
  209. @Override
  210. public List<MediaServerItem> getAllFromDatabase() {
  211. return mediaServerMapper.queryAll();
  212. }
  213. @Override
  214. public List<MediaServerItem> getAllOnline() {
  215. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  216. Set<Object> mediaServerIdSet = redisTemplate.opsForZSet().reverseRange(key, 0, -1);
  217. List<MediaServerItem> result = new ArrayList<>();
  218. if (mediaServerIdSet != null && mediaServerIdSet.size() > 0) {
  219. for (Object mediaServerId : mediaServerIdSet) {
  220. String mediaServerIdStr = (String) mediaServerId;
  221. String serverKey = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerIdStr;
  222. result.add((MediaServerItem) redisTemplate.opsForValue().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 JsonUtil.redisJsonToObject(redisTemplate, key, MediaServerItem.class);
  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. redisTemplate.delete(key);
  249. }
  250. @Override
  251. public void add(MediaServerItem mediaServerItem) {
  252. mediaServerItem.setCreateTime(DateUtil.getNow());
  253. mediaServerItem.setUpdateTime(DateUtil.getNow());
  254. mediaServerItem.setHookAliveInterval(30f);
  255. JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig(mediaServerItem);
  256. if (responseJSON != null) {
  257. JSONArray data = responseJSON.getJSONArray("data");
  258. if (data != null && data.size() > 0) {
  259. ZLMServerConfig zlmServerConfig= JSON.parseObject(JSON.toJSONString(data.get(0)), ZLMServerConfig.class);
  260. if (mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId()) != null) {
  261. throw new ControllerException(ErrorCode.ERROR100.getCode(),"保存失败,媒体服务ID [ " + zlmServerConfig.getGeneralMediaServerId() + " ] 已存在,请修改媒体服务器配置");
  262. }
  263. mediaServerItem.setId(zlmServerConfig.getGeneralMediaServerId());
  264. zlmServerConfig.setIp(mediaServerItem.getIp());
  265. mediaServerMapper.add(mediaServerItem);
  266. zlmServerOnline(zlmServerConfig);
  267. }else {
  268. throw new ControllerException(ErrorCode.ERROR100.getCode(),"连接失败");
  269. }
  270. }else {
  271. throw new ControllerException(ErrorCode.ERROR100.getCode(),"连接失败");
  272. }
  273. }
  274. @Override
  275. public int addToDatabase(MediaServerItem mediaSerItem) {
  276. return mediaServerMapper.add(mediaSerItem);
  277. }
  278. @Override
  279. public int updateToDatabase(MediaServerItem mediaSerItem) {
  280. int result = 0;
  281. if (mediaSerItem.isDefaultServer()) {
  282. TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
  283. int delResult = mediaServerMapper.delDefault();
  284. if (delResult == 0) {
  285. logger.error("移除数据库默认zlm节点失败");
  286. //事务回滚
  287. dataSourceTransactionManager.rollback(transactionStatus);
  288. return 0;
  289. }
  290. result = mediaServerMapper.add(mediaSerItem);
  291. dataSourceTransactionManager.commit(transactionStatus); //手动提交
  292. }else {
  293. result = mediaServerMapper.update(mediaSerItem);
  294. }
  295. return result;
  296. }
  297. /**
  298. * 处理zlm上线
  299. * @param zlmServerConfig zlm上线携带的参数
  300. */
  301. @Override
  302. public void zlmServerOnline(ZLMServerConfig zlmServerConfig) {
  303. MediaServerItem serverItem = mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId());
  304. if (serverItem == null) {
  305. logger.warn("[未注册的zlm] 拒接接入:{}来自{}:{}", zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(),zlmServerConfig.getHttpPort() );
  306. logger.warn("请检查ZLM的<general.mediaServerId>配置是否与WVP的<media.id>一致");
  307. return;
  308. }else {
  309. logger.info("[ZLM] 正在连接 : {} -> {}:{}",
  310. zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(), zlmServerConfig.getHttpPort());
  311. }
  312. serverItem.setHookAliveInterval(zlmServerConfig.getHookAliveInterval());
  313. if (serverItem.getHttpPort() == 0) {
  314. serverItem.setHttpPort(zlmServerConfig.getHttpPort());
  315. }
  316. if (serverItem.getHttpSSlPort() == 0) {
  317. serverItem.setHttpSSlPort(zlmServerConfig.getHttpSSLport());
  318. }
  319. if (serverItem.getRtmpPort() == 0) {
  320. serverItem.setRtmpPort(zlmServerConfig.getRtmpPort());
  321. }
  322. if (serverItem.getRtmpSSlPort() == 0) {
  323. serverItem.setRtmpSSlPort(zlmServerConfig.getRtmpSslPort());
  324. }
  325. if (serverItem.getRtspPort() == 0) {
  326. serverItem.setRtspPort(zlmServerConfig.getRtspPort());
  327. }
  328. if (serverItem.getRtspSSLPort() == 0) {
  329. serverItem.setRtspSSLPort(zlmServerConfig.getRtspSSlport());
  330. }
  331. if (serverItem.getRtpProxyPort() == 0) {
  332. serverItem.setRtpProxyPort(zlmServerConfig.getRtpProxyPort());
  333. }
  334. serverItem.setStatus(true);
  335. if (ObjectUtils.isEmpty(serverItem.getId())) {
  336. logger.warn("[未注册的zlm] serverItem缺少ID, 无法接入:{}:{}", zlmServerConfig.getIp(),zlmServerConfig.getHttpPort() );
  337. return;
  338. }
  339. mediaServerMapper.update(serverItem);
  340. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + zlmServerConfig.getGeneralMediaServerId();
  341. if (ssrcFactory.hasMediaServerSSRC(serverItem.getId())) {
  342. ssrcFactory.initMediaServerSSRC(zlmServerConfig.getGeneralMediaServerId(), null);
  343. }
  344. redisTemplate.opsForValue().set(key, serverItem);
  345. resetOnlineServerItem(serverItem);
  346. if (serverItem.isAutoConfig()) {
  347. // 查看assist服务的录像路径配置
  348. if (serverItem.getRecordAssistPort() > 0 && userSetting.getRecordPath() == null) {
  349. JSONObject info = assistRESTfulUtils.getInfo(serverItem, null);
  350. if (info != null && info.getInteger("code") != null && info.getInteger("code") == 0 ) {
  351. JSONObject dataJson = info.getJSONObject("data");
  352. if (dataJson != null) {
  353. String recordPath = dataJson.getString("record");
  354. userSetting.setRecordPath(recordPath);
  355. }
  356. }
  357. }
  358. setZLMConfig(serverItem, "0".equals(zlmServerConfig.getHookEnable()));
  359. }
  360. final String zlmKeepaliveKey = zlmKeepaliveKeyPrefix + serverItem.getId();
  361. dynamicTask.stop(zlmKeepaliveKey);
  362. dynamicTask.startDelay(zlmKeepaliveKey, new KeepAliveTimeoutRunnable(serverItem), (Math.getExponent(serverItem.getHookAliveInterval()) + 5) * 1000);
  363. publisher.zlmOnlineEventPublish(serverItem.getId());
  364. logger.info("[ZLM] 连接成功 {} - {}:{} ",
  365. zlmServerConfig.getGeneralMediaServerId(), zlmServerConfig.getIp(), zlmServerConfig.getHttpPort());
  366. }
  367. class KeepAliveTimeoutRunnable implements Runnable{
  368. private MediaServerItem serverItem;
  369. public KeepAliveTimeoutRunnable(MediaServerItem serverItem) {
  370. this.serverItem = serverItem;
  371. }
  372. @Override
  373. public void run() {
  374. logger.info("[zlm心跳到期]:" + serverItem.getId());
  375. // 发起http请求验证zlm是否确实无法连接,如果确实无法连接则发送离线事件,否则不作处理
  376. JSONObject mediaServerConfig = zlmresTfulUtils.getMediaServerConfig(serverItem);
  377. if (mediaServerConfig != null && mediaServerConfig.getInteger("code") == 0) {
  378. logger.info("[zlm心跳到期]:{}验证后zlm仍在线,恢复心跳信息,请检查zlm是否可以正常向wvp发送心跳", serverItem.getId());
  379. // 添加zlm信息
  380. updateMediaServerKeepalive(serverItem.getId(), null);
  381. }else {
  382. publisher.zlmOfflineEventPublish(serverItem.getId());
  383. }
  384. }
  385. }
  386. @Override
  387. public void zlmServerOffline(String mediaServerId) {
  388. delete(mediaServerId);
  389. final String zlmKeepaliveKey = zlmKeepaliveKeyPrefix + mediaServerId;
  390. dynamicTask.stop(zlmKeepaliveKey);
  391. }
  392. @Override
  393. public void resetOnlineServerItem(MediaServerItem serverItem) {
  394. // 更新缓存
  395. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  396. // 使用zset的分数作为当前并发量, 默认值设置为0
  397. if (redisTemplate.opsForZSet().score(key, serverItem.getId()) == null) { // 不存在则设置默认值 已存在则重置
  398. redisTemplate.opsForZSet().add(key, serverItem.getId(), 0L);
  399. // 查询服务流数量
  400. zlmresTfulUtils.getMediaList(serverItem, null, null, "rtsp",(mediaList ->{
  401. Integer code = mediaList.getInteger("code");
  402. if (code == 0) {
  403. JSONArray data = mediaList.getJSONArray("data");
  404. if (data != null) {
  405. redisTemplate.opsForZSet().add(key, serverItem.getId(), data.size());
  406. }
  407. }
  408. }));
  409. }else {
  410. clearRTPServer(serverItem);
  411. }
  412. }
  413. @Override
  414. public void addCount(String mediaServerId) {
  415. if (mediaServerId == null) {
  416. return;
  417. }
  418. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  419. redisTemplate.opsForZSet().incrementScore(key, mediaServerId, 1);
  420. }
  421. @Override
  422. public void removeCount(String mediaServerId) {
  423. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  424. redisTemplate.opsForZSet().incrementScore(key, mediaServerId, - 1);
  425. }
  426. /**
  427. * 获取负载最低的节点
  428. * @return MediaServerItem
  429. */
  430. @Override
  431. public MediaServerItem getMediaServerForMinimumLoad(Boolean hasAssist) {
  432. String key = VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId();
  433. Long size = redisTemplate.opsForZSet().zCard(key);
  434. if (size == null || size == 0) {
  435. logger.info("获取负载最低的节点时无在线节点");
  436. return null;
  437. }
  438. // 获取分数最低的,及并发最低的
  439. Set<Object> objects = redisTemplate.opsForZSet().range(key, 0, -1);
  440. ArrayList<Object> mediaServerObjectS = new ArrayList<>(objects);
  441. MediaServerItem mediaServerItem = null;
  442. if (hasAssist == null) {
  443. String mediaServerId = (String)mediaServerObjectS.get(0);
  444. mediaServerItem = getOne(mediaServerId);
  445. }else if (hasAssist) {
  446. for (Object mediaServerObject : mediaServerObjectS) {
  447. String mediaServerId = (String)mediaServerObject;
  448. MediaServerItem serverItem = getOne(mediaServerId);
  449. if (serverItem.getRecordAssistPort() > 0) {
  450. mediaServerItem = serverItem;
  451. break;
  452. }
  453. }
  454. }else if (!hasAssist) {
  455. for (Object mediaServerObject : mediaServerObjectS) {
  456. String mediaServerId = (String)mediaServerObject;
  457. MediaServerItem serverItem = getOne(mediaServerId);
  458. if (serverItem.getRecordAssistPort() == 0) {
  459. mediaServerItem = serverItem;
  460. break;
  461. }
  462. }
  463. }
  464. return mediaServerItem;
  465. }
  466. /**
  467. * 对zlm服务器进行基础配置
  468. * @param mediaServerItem 服务ID
  469. * @param restart 是否重启zlm
  470. */
  471. @Override
  472. public void setZLMConfig(MediaServerItem mediaServerItem, boolean restart) {
  473. logger.info("[ZLM] 正在设置 :{} -> {}:{}",
  474. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  475. String protocol = sslEnabled ? "https" : "http";
  476. String hookPrex = String.format("%s://%s:%s/index/hook", protocol, mediaServerItem.getHookIp(), serverPort);
  477. Map<String, Object> param = new HashMap<>();
  478. param.put("api.secret",mediaServerItem.getSecret()); // -profile:v Baseline
  479. if (mediaServerItem.getRtspPort() != 0) {
  480. param.put("ffmpeg.snap", "%s -rtsp_transport tcp -i %s -y -f mjpeg -t 0.001 %s");
  481. }
  482. param.put("hook.enable","1");
  483. param.put("hook.on_flow_report","");
  484. param.put("hook.on_play",String.format("%s/on_play", hookPrex));
  485. param.put("hook.on_http_access","");
  486. param.put("hook.on_publish", String.format("%s/on_publish", hookPrex));
  487. param.put("hook.on_record_ts","");
  488. param.put("hook.on_rtsp_auth","");
  489. param.put("hook.on_rtsp_realm","");
  490. param.put("hook.on_server_started",String.format("%s/on_server_started", hookPrex));
  491. param.put("hook.on_shell_login","");
  492. param.put("hook.on_stream_changed",String.format("%s/on_stream_changed", hookPrex));
  493. param.put("hook.on_stream_none_reader",String.format("%s/on_stream_none_reader", hookPrex));
  494. param.put("hook.on_stream_not_found",String.format("%s/on_stream_not_found", hookPrex));
  495. param.put("hook.on_server_keepalive",String.format("%s/on_server_keepalive", hookPrex));
  496. param.put("hook.on_send_rtp_stopped",String.format("%s/on_send_rtp_stopped", hookPrex));
  497. param.put("hook.on_rtp_server_timeout",String.format("%s/on_rtp_server_timeout", hookPrex));
  498. if (mediaServerItem.getRecordAssistPort() > 0) {
  499. param.put("hook.on_record_mp4",String.format("http://127.0.0.1:%s/api/record/on_record_mp4", mediaServerItem.getRecordAssistPort()));
  500. }else {
  501. param.put("hook.on_record_mp4","");
  502. }
  503. param.put("hook.timeoutSec","20");
  504. // 推流断开后可以在超时时间内重新连接上继续推流,这样播放器会接着播放。
  505. // 置0关闭此特性(推流断开会导致立即断开播放器)
  506. // 此参数不应大于播放器超时时间
  507. // 优化此消息以更快的收到流注销事件
  508. param.put("protocol.continue_push_ms", "3000" );
  509. // 最多等待未初始化的Track时间,单位毫秒,超时之后会忽略未初始化的Track, 设置此选项优化那些音频错误的不规范流,
  510. // 等zlm支持给每个rtpServer设置关闭音频的时候可以不设置此选项
  511. // param.put("general.wait_track_ready_ms", "3000" );
  512. if (mediaServerItem.isRtpEnable() && !ObjectUtils.isEmpty(mediaServerItem.getRtpPortRange())) {
  513. param.put("rtp_proxy.port_range", mediaServerItem.getRtpPortRange().replace(",", "-"));
  514. }
  515. if (userSetting.getRecordPath() != null) {
  516. File recordPathFile = new File(userSetting.getRecordPath());
  517. File mp4SavePathFile = recordPathFile.getParentFile().getAbsoluteFile();
  518. param.put("protocol.mp4_save_path", mp4SavePathFile.getAbsoluteFile());
  519. param.put("record.appName", recordPathFile.getName());
  520. }
  521. JSONObject responseJSON = zlmresTfulUtils.setServerConfig(mediaServerItem, param);
  522. if (responseJSON != null && responseJSON.getInteger("code") == 0) {
  523. if (restart) {
  524. logger.info("[ZLM] 设置成功,开始重启以保证配置生效 {} -> {}:{}",
  525. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  526. zlmresTfulUtils.restartServer(mediaServerItem);
  527. }else {
  528. logger.info("[ZLM] 设置成功 {} -> {}:{}",
  529. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  530. }
  531. }else {
  532. logger.info("[ZLM] 设置zlm失败 {} -> {}:{}",
  533. mediaServerItem.getId(), mediaServerItem.getIp(), mediaServerItem.getHttpPort());
  534. }
  535. }
  536. @Override
  537. public MediaServerItem checkMediaServer(String ip, int port, String secret) {
  538. if (mediaServerMapper.queryOneByHostAndPort(ip, port) != null) {
  539. throw new ControllerException(ErrorCode.ERROR100.getCode(), "此连接已存在");
  540. }
  541. MediaServerItem mediaServerItem = new MediaServerItem();
  542. mediaServerItem.setIp(ip);
  543. mediaServerItem.setHttpPort(port);
  544. mediaServerItem.setSecret(secret);
  545. JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig(mediaServerItem);
  546. if (responseJSON == null) {
  547. throw new ControllerException(ErrorCode.ERROR100.getCode(), "连接失败");
  548. }
  549. JSONArray data = responseJSON.getJSONArray("data");
  550. ZLMServerConfig zlmServerConfig = JSON.parseObject(JSON.toJSONString(data.get(0)), ZLMServerConfig.class);
  551. if (zlmServerConfig == null) {
  552. throw new ControllerException(ErrorCode.ERROR100.getCode(), "读取配置失败");
  553. }
  554. if (mediaServerMapper.queryOne(zlmServerConfig.getGeneralMediaServerId()) != null) {
  555. throw new ControllerException(ErrorCode.ERROR100.getCode(), "媒体服务ID [" + zlmServerConfig.getGeneralMediaServerId() + " ] 已存在,请修改媒体服务器配置");
  556. }
  557. mediaServerItem.setHttpSSlPort(zlmServerConfig.getHttpPort());
  558. mediaServerItem.setRtmpPort(zlmServerConfig.getRtmpPort());
  559. mediaServerItem.setRtmpSSlPort(zlmServerConfig.getRtmpSslPort());
  560. mediaServerItem.setRtspPort(zlmServerConfig.getRtspPort());
  561. mediaServerItem.setRtspSSLPort(zlmServerConfig.getRtspSSlport());
  562. mediaServerItem.setRtpProxyPort(zlmServerConfig.getRtpProxyPort());
  563. mediaServerItem.setStreamIp(ip);
  564. mediaServerItem.setHookIp(sipConfig.getIp().split(",")[0]);
  565. mediaServerItem.setSdpIp(ip);
  566. return mediaServerItem;
  567. }
  568. @Override
  569. public boolean checkMediaRecordServer(String ip, int port) {
  570. boolean result = false;
  571. OkHttpClient client = new OkHttpClient();
  572. String url = String.format("http://%s:%s/index/api/record", ip, port);
  573. Request request = new Request.Builder()
  574. .get()
  575. .url(url)
  576. .build();
  577. try {
  578. Response response = client.newCall(request).execute();
  579. if (response != null) {
  580. result = true;
  581. }
  582. } catch (Exception e) {}
  583. return result;
  584. }
  585. @Override
  586. public void delete(String id) {
  587. redisTemplate.opsForZSet().remove(VideoManagerConstants.MEDIA_SERVERS_ONLINE_PREFIX + userSetting.getServerId(), id);
  588. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + id;
  589. redisTemplate.delete(key);
  590. }
  591. @Override
  592. public void deleteDb(String id){
  593. //同步删除数据库中的数据
  594. mediaServerMapper.delOne(id);
  595. }
  596. @Override
  597. public void updateMediaServerKeepalive(String mediaServerId, ServerKeepaliveData data) {
  598. MediaServerItem mediaServerItem = getOne(mediaServerId);
  599. if (mediaServerItem == null) {
  600. // 缓存不存在,从数据库查询,如果数据库不存在则是错误的
  601. mediaServerItem = getOneFromDatabase(mediaServerId);
  602. if (mediaServerItem == null) {
  603. logger.warn("[更新ZLM 保活信息] 流媒体{}尚未加入使用,请检查节点中是否含有此流媒体 ", mediaServerId);
  604. return;
  605. }
  606. // zlm连接重试
  607. logger.warn("[更新ZLM 保活信息]尝试链接zml id {}", mediaServerId);
  608. ssrcFactory.initMediaServerSSRC(mediaServerItem.getId(), null);
  609. String key = VideoManagerConstants.MEDIA_SERVER_PREFIX + userSetting.getServerId() + "_" + mediaServerItem.getId();
  610. redisTemplate.opsForValue().set(key, mediaServerItem);
  611. clearRTPServer(mediaServerItem);
  612. }
  613. final String zlmKeepaliveKey = zlmKeepaliveKeyPrefix + mediaServerItem.getId();
  614. dynamicTask.stop(zlmKeepaliveKey);
  615. dynamicTask.startDelay(zlmKeepaliveKey, new KeepAliveTimeoutRunnable(mediaServerItem), (mediaServerItem.getHookAliveInterval().intValue() + 5) * 1000);
  616. }
  617. private MediaServerItem getOneFromDatabase(String mediaServerId) {
  618. return mediaServerMapper.queryOne(mediaServerId);
  619. }
  620. @Override
  621. public void syncCatchFromDatabase() {
  622. List<MediaServerItem> allInCatch = getAll();
  623. List<MediaServerItem> allInDatabase = mediaServerMapper.queryAll();
  624. Map<String, MediaServerItem> mediaServerItemMap = new HashMap<>();
  625. for (MediaServerItem mediaServerItem : allInDatabase) {
  626. mediaServerItemMap.put(mediaServerItem.getId(), mediaServerItem);
  627. }
  628. for (MediaServerItem mediaServerItem : allInCatch) {
  629. if (!mediaServerItemMap.containsKey(mediaServerItem.getId())) {
  630. delete(mediaServerItem.getId());
  631. }
  632. }
  633. }
  634. @Override
  635. public boolean checkRtpServer(MediaServerItem mediaServerItem, String app, String stream) {
  636. JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(mediaServerItem, stream);
  637. if(rtpInfo.getInteger("code") == 0){
  638. return rtpInfo.getBoolean("exist");
  639. }
  640. return false;
  641. }
  642. @Override
  643. public MediaServerLoad getLoad(MediaServerItem mediaServerItem) {
  644. MediaServerLoad result = new MediaServerLoad();
  645. result.setId(mediaServerItem.getId());
  646. result.setPush(redisCatchStorage.getPushStreamCount(mediaServerItem.getId()));
  647. result.setProxy(redisCatchStorage.getProxyStreamCount(mediaServerItem.getId()));
  648. result.setGbReceive(redisCatchStorage.getGbReceiveCount(mediaServerItem.getId()));
  649. result.setGbSend(redisCatchStorage.getGbSendCount(mediaServerItem.getId()));
  650. return result;
  651. }
  652. }