MediaServerServiceImpl.java 24 KB

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