MediaServerServiceImpl.java 25 KB

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