guzijian / wvp-GB28181-pro · Commits

Commit c19ad94c3ec45245914a7682174db5fb650ee08e

Authored by 648540858
1 parent ebcd2320

优化目录更新，更新失败时只更新收到的内容不重置所有通道和自动拉流

Inline Side-by-side
Showing 11 changed files with 848 additions and 693 deletions
src/main/java/com/genersoft/iot/vmp/gb28181/session/CatalogDataCatch.java
  View file @c19ad94
... ... @@ -109,12 +109,18 @@ public class CatalogDataCatch {
109 109  
110 110 for (String deviceId : keys) {
111 111 CatalogData catalogData = data.get(deviceId);
112   - if ( catalogData.getLastTime().isBefore(instantBefore5S)) { // 超过五秒收不到消息任务超时， 只更新这一部分数据
  112 + if ( catalogData.getLastTime().isBefore(instantBefore5S)) {
  113 + // 超过五秒收不到消息任务超时， 只更新这一部分数据, 收到数据与声明的总数一致，则重置通道数据，数据不全则只对收到的数据做更新操作
113 114 if (catalogData.getStatus().equals(CatalogData.CatalogDataStatus.runIng)) {
114   - storager.resetChannels(catalogData.getDevice().getDeviceId(), catalogData.getChannelList());
  115 + if (catalogData.getTotal() == catalogData.getChannelList().size()) {
  116 + storager.resetChannels(catalogData.getDevice().getDeviceId(), catalogData.getChannelList());
  117 + }else {
  118 + storager.updateChannels(catalogData.getDevice().getDeviceId(), catalogData.getChannelList());
  119 + }
  120 + String errorMsg = "更新成功，共" + catalogData.getTotal() + "条，已更新" + catalogData.getChannelList().size() + "条";
  121 + catalogData.setErrorMsg(errorMsg);
115 122 if (catalogData.getTotal() != catalogData.getChannelList().size()) {
116   - String errorMsg = "更新成功，共" + catalogData.getTotal() + "条，已更新" + catalogData.getChannelList().size() + "条";
117   - catalogData.setErrorMsg(errorMsg);
  123 +
118 124 }
119 125 }else if (catalogData.getStatus().equals(CatalogData.CatalogDataStatus.ready)) {
120 126 String errorMsg = "同步失败，等待回复超时";
... ...
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/SIPRequestProcessorParent.java
  View file @c19ad94
... ... @@ -3,7 +3,6 @@ package com.genersoft.iot.vmp.gb28181.transmit.event.request;
3 3 import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
4 4 import com.genersoft.iot.vmp.gb28181.transmit.SIPSender;
5 5 import com.genersoft.iot.vmp.gb28181.utils.SipUtils;
6   -import gov.nist.javax.sip.SipProviderImpl;
7 6 import gov.nist.javax.sip.message.SIPRequest;
8 7 import gov.nist.javax.sip.message.SIPResponse;
9 8 import org.apache.commons.lang3.ArrayUtils;
... ... @@ -14,19 +13,17 @@ import org.dom4j.io.SAXReader;
14 13 import org.slf4j.Logger;
15 14 import org.slf4j.LoggerFactory;
16 15 import org.springframework.beans.factory.annotation.Autowired;
17   -import org.springframework.beans.factory.annotation.Qualifier;
18 16  
19 17 import javax.sip.*;
20 18 import javax.sip.address.Address;
21   -import javax.sip.address.AddressFactory;
22 19 import javax.sip.address.SipURI;
23   -import javax.sip.header.*;
  20 +import javax.sip.header.ContentTypeHeader;
  21 +import javax.sip.header.ExpiresHeader;
  22 +import javax.sip.header.HeaderFactory;
24 23 import javax.sip.message.MessageFactory;
25 24 import javax.sip.message.Request;
26 25 import javax.sip.message.Response;
27 26 import java.io.ByteArrayInputStream;
28   -import java.nio.ByteBuffer;
29   -import java.nio.charset.StandardCharsets;
30 27 import java.text.ParseException;
31 28 import java.util.ArrayList;
32 29 import java.util.Arrays;
... ... @@ -44,15 +41,6 @@ public abstract class SIPRequestProcessorParent {
44 41 @Autowired
45 42 private SIPSender sipSender;
46 43  
47   - public AddressFactory getAddressFactory() {
48   - try {
49   - return SipFactory.getInstance().createAddressFactory();
50   - } catch (PeerUnavailableException e) {
51   - e.printStackTrace();
52   - }
53   - return null;
54   - }
55   -
56 44 public HeaderFactory getHeaderFactory() {
57 45 try {
58 46 return SipFactory.getInstance().createHeaderFactory();
... ... @@ -93,53 +81,6 @@ public abstract class SIPRequestProcessorParent {
93 81 return responseAck(sipRequest, statusCode, msg, null);
94 82 }
95 83  
96   -// public SIPResponse responseAck(ServerTransaction serverTransaction, int statusCode, String msg, ResponseAckExtraParam responseAckExtraParam) throws SipException, InvalidArgumentException, ParseException {
97   -// if (serverTransaction == null) {
98   -// logger.warn("[回复消息] ServerTransaction 为null");
99   -// return null;
100   -// }
101   -// ToHeader toHeader = (ToHeader) serverTransaction.getRequest().getHeader(ToHeader.NAME);
102   -// if (toHeader.getTag() == null) {
103   -// toHeader.setTag(SipUtils.getNewTag());
104   -// }
105   -// SIPResponse response = (SIPResponse)getMessageFactory().createResponse(statusCode, serverTransaction.getRequest());
106   -// if (msg != null) {
107   -// response.setReasonPhrase(msg);
108   -// }
109   -// if (responseAckExtraParam != null) {
110   -// if (responseAckExtraParam.sipURI != null && serverTransaction.getRequest().getMethod().equals(Request.INVITE)) {
111   -// logger.debug("responseSdpAck SipURI: {}:{}", responseAckExtraParam.sipURI.getHost(), responseAckExtraParam.sipURI.getPort());
112   -// Address concatAddress = SipFactory.getInstance().createAddressFactory().createAddress(
113   -// SipFactory.getInstance().createAddressFactory().createSipURI(responseAckExtraParam.sipURI.getUser(), responseAckExtraParam.sipURI.getHost()+":"+responseAckExtraParam.sipURI.getPort()
114   -// ));
115   -// response.addHeader(SipFactory.getInstance().createHeaderFactory().createContactHeader(concatAddress));
116   -// }
117   -// if (responseAckExtraParam.contentTypeHeader != null) {
118   -// response.setContent(responseAckExtraParam.content, responseAckExtraParam.contentTypeHeader);
119   -// }
120   -//
121   -// if (serverTransaction.getRequest().getMethod().equals(Request.SUBSCRIBE)) {
122   -// if (responseAckExtraParam.expires == -1) {
123   -// logger.error("[参数不全] 2xx的SUBSCRIBE回复，必须设置Expires header");
124   -// }else {
125   -// ExpiresHeader expiresHeader = SipFactory.getInstance().createHeaderFactory().createExpiresHeader(responseAckExtraParam.expires);
126   -// response.addHeader(expiresHeader);
127   -// }
128   -// }
129   -// }else {
130   -// if (serverTransaction.getRequest().getMethod().equals(Request.SUBSCRIBE)) {
131   -// logger.error("[参数不全] 2xx的SUBSCRIBE回复，必须设置Expires header");
132   -// }
133   -// }
134   -// serverTransaction.sendResponse(response);
135   -// if (statusCode >= 200 && !"NOTIFY".equalsIgnoreCase(serverTransaction.getRequest().getMethod())) {
136   -// if (serverTransaction.getDialog() != null) {
137   -// serverTransaction.getDialog().delete();
138   -// }
139   -// }
140   -// return response;
141   -// }
142   -
143 84 public SIPResponse responseAck(SIPRequest sipRequest, int statusCode, String msg, ResponseAckExtraParam responseAckExtraParam) throws SipException, InvalidArgumentException, ParseException {
144 85 if (sipRequest.getToHeader().getTag() == null) {
145 86 sipRequest.getToHeader().setTag(SipUtils.getNewTag());
... ...
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/impl/InviteRequestProcessor.java
  View file @c19ad94
... ... @@ -228,7 +228,7 @@ public class InviteRequestProcessor extends SIPRequestProcessorParent implements
228 228 }
229 229 return;
230 230 } else {
231   - logger.info("通道不存在，返回404");
  231 + logger.info("通道不存在，返回404: {}", channelId);
232 232 try {
233 233 // 通道不存在，发404，资源不存在
234 234 responseAck(request, Response.NOT_FOUND);
... ...
src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMHttpHookListener.java
  View file @c19ad94
... ... @@ -8,6 +8,8 @@ import com.genersoft.iot.vmp.conf.exception.SsrcTransactionNotFoundException;
8 8 import com.genersoft.iot.vmp.gb28181.bean.*;
9 9 import com.genersoft.iot.vmp.gb28181.event.EventPublisher;
10 10 import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
  11 +import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
  12 +import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
11 13 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
12 14 import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommanderFroPlatform;
13 15 import com.genersoft.iot.vmp.media.zlm.dto.HookType;
... ... @@ -18,7 +20,10 @@ import com.genersoft.iot.vmp.media.zlm.dto.hook.*;
18 20 import com.genersoft.iot.vmp.service.*;
19 21 import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
20 22 import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
  23 +import com.genersoft.iot.vmp.vmanager.bean.DeferredResultEx;
  24 +import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
21 25 import com.genersoft.iot.vmp.vmanager.bean.StreamContent;
  26 +import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
22 27 import org.slf4j.Logger;
23 28 import org.slf4j.LoggerFactory;
24 29 import org.springframework.beans.factory.annotation.Autowired;
... ... @@ -26,6 +31,7 @@ import org.springframework.beans.factory.annotation.Qualifier;
26 31 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
27 32 import org.springframework.util.ObjectUtils;
28 33 import org.springframework.web.bind.annotation.*;
  34 +import org.springframework.web.context.request.async.DeferredResult;
29 35  
30 36 import javax.servlet.http.HttpServletRequest;
31 37 import javax.sip.InvalidArgumentException;
... ... @@ -34,638 +40,644 @@ import java.text.ParseException;
34 40 import java.util.HashMap;
35 41 import java.util.List;
36 42 import java.util.Map;
  43 +import java.util.UUID;
37 44  
38   -/**
  45 +/**
39 46 * @description:针对 ZLMediaServer的hook事件监听
40 47 * @author: swwheihei
41   - * @date: 2020年5月8日 上午10:46:48
  48 + * @date: 2020年5月8日 上午10:46:48
42 49 */
43 50 @RestController
44 51 @RequestMapping("/index/hook")
45 52 public class ZLMHttpHookListener {
46 53  
47   - private final static Logger logger = LoggerFactory.getLogger(ZLMHttpHookListener.class);
  54 + private final static Logger logger = LoggerFactory.getLogger(ZLMHttpHookListener.class);
48 55  
49   - @Autowired
50   - private SIPCommander cmder;
51   -
52   - @Autowired
53   - private SIPCommanderFroPlatform commanderFroPlatform;
54   -
55   - @Autowired
56   - private IPlayService playService;
57   -
58   - @Autowired
59   - private IVideoManagerStorage storager;
60   -
61   - @Autowired
62   - private IRedisCatchStorage redisCatchStorage;
63   -
64   - @Autowired
65   - private IDeviceService deviceService;
66   -
67   - @Autowired
68   - private IMediaServerService mediaServerService;
69   -
70   - @Autowired
71   - private IStreamProxyService streamProxyService;
72   -
73   - @Autowired
74   - private IStreamPushService streamPushService;
75   -
76   - @Autowired
77   - private IMediaService mediaService;
78   -
79   - @Autowired
80   - private EventPublisher eventPublisher;
81   -
82   - @Autowired
83   - private ZLMMediaListManager zlmMediaListManager;
84   -
85   - @Autowired
86   - private ZlmHttpHookSubscribe subscribe;
87   -
88   - @Autowired
89   - private UserSetting userSetting;
90   -
91   - @Autowired
92   - private IUserService userService;
93   -
94   - @Autowired
95   - private VideoStreamSessionManager sessionManager;
96   -
97   - @Autowired
98   - private AssistRESTfulUtils assistRESTfulUtils;
99   -
100   - @Qualifier("taskExecutor")
101   - @Autowired
102   - private ThreadPoolTaskExecutor taskExecutor;
103   -
104   - /**
105   - * 服务器定时上报时间，上报间隔可配置，默认10s上报一次
106   - *
107   - */
108   - @ResponseBody
109   - @PostMapping(value = "/on_server_keepalive", produces = "application/json;charset=UTF-8")
110   - public JSONObject onServerKeepalive(@RequestBody OnServerKeepaliveHookParam param){
111   -
112   - logger.info("[ZLM HOOK] 收到zlm心跳：" + param.getMediaServerId());
113   -
114   - taskExecutor.execute(()->{
115   - List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_keepalive);
116   - JSONObject json = (JSONObject) JSON.toJSON(param);
117   - if (subscribes != null && subscribes.size() > 0) {
118   - for (ZlmHttpHookSubscribe.Event subscribe : subscribes) {
119   - subscribe.response(null, json);
120   - }
121   - }
122   - });
123   - mediaServerService.updateMediaServerKeepalive(param.getMediaServerId(), param.getData());
124   -
125   - JSONObject ret = new JSONObject();
126   - ret.put("code", 0);
127   - ret.put("msg", "success");
128   -
129   - return ret;
130   - }
131   -
132   - /**
133   - * 播放器鉴权事件，rtsp/rtmp/http-flv/ws-flv/hls的播放都将触发此鉴权事件。
134   - *
135   - */
136   - @ResponseBody
137   - @PostMapping(value = "/on_play", produces = "application/json;charset=UTF-8")
138   - public JSONObject onPlay(@RequestBody OnPlayHookParam param){
139   - if (logger.isDebugEnabled()) {
140   - logger.debug("[ZLM HOOK] 播放鉴权：{}->{}" + param.getMediaServerId(), param);
141   - }
142   - String mediaServerId = param.getMediaServerId();
143   -
144   - taskExecutor.execute(()->{
145   - JSONObject json = (JSONObject) JSON.toJSON(param);
146   - ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_play, json);
147   - if (subscribe != null ) {
148   - MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId);
149   - if (mediaInfo != null) {
150   - subscribe.response(mediaInfo, json);
151   - }
152   - }
153   - });
154   - JSONObject ret = new JSONObject();
155   - if (!"rtp".equals(param.getApp())) {
156   - Map<String, String> paramMap = urlParamToMap(param.getParams());
157   - StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream());
158   - if (streamAuthorityInfo != null && streamAuthorityInfo.getCallId() != null && !streamAuthorityInfo.getCallId().equals(paramMap.get("callId"))) {
159   - ret.put("code", 401);
160   - ret.put("msg", "Unauthorized");
161   - return ret;
162   - }
163   - }
164   -
165   - ret.put("code", 0);
166   - ret.put("msg", "success");
167   - return ret;
168   - }
169   -
170   - /**
171   - * rtsp/rtmp/rtp推流鉴权事件。
172   - *
173   - */
174   - @ResponseBody
175   - @PostMapping(value = "/on_publish", produces = "application/json;charset=UTF-8")
176   - public JSONObject onPublish(@RequestBody OnPublishHookParam param) {
177   -
178   - JSONObject json = (JSONObject) JSON.toJSON(param);
179   -
180   - logger.info("[ZLM HOOK]推流鉴权：{}->{}", param.getMediaServerId(), param);
181   - JSONObject ret = new JSONObject();
182   - String mediaServerId = json.getString("mediaServerId");
183   - MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId);
184   -
185   - if (!"rtp".equals(param.getApp())) {
186   - if (userSetting.getPushAuthority()) {
187   - // 推流鉴权
188   - if (param.getParams() == null) {
189   - logger.info("推流鉴权失败： 缺少不要参数：sign=md5(user表的pushKey)");
190   - ret.put("code", 401);
191   - ret.put("msg", "Unauthorized");
192   - return ret;
193   - }
194   - Map<String, String> paramMap = urlParamToMap(param.getParams());
195   - String sign = paramMap.get("sign");
196   - if (sign == null) {
197   - logger.info("推流鉴权失败： 缺少不要参数：sign=md5(user表的pushKey)");
198   - ret.put("code", 401);
199   - ret.put("msg", "Unauthorized");
200   - return ret;
201   - }
202   - // 推流自定义播放鉴权码
203   - String callId = paramMap.get("callId");
204   - // 鉴权配置
205   - boolean hasAuthority = userService.checkPushAuthority(callId, sign);
206   - if (!hasAuthority) {
207   - logger.info("推流鉴权失败： sign 无权限: callId={}. sign={}", callId, sign);
208   - ret.put("code", 401);
209   - ret.put("msg", "Unauthorized");
210   - return ret;
211   - }
212   - StreamAuthorityInfo streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(param);
213   - streamAuthorityInfo.setCallId(callId);
214   - streamAuthorityInfo.setSign(sign);
215   - // 鉴权通过
216   - redisCatchStorage.updateStreamAuthorityInfo(param.getApp(), param.getStream(), streamAuthorityInfo);
217   - // 通知assist新的callId
218   - if (mediaInfo != null && mediaInfo.getRecordAssistPort() > 0) {
219   - taskExecutor.execute(()->{
220   - assistRESTfulUtils.addStreamCallInfo(mediaInfo, param.getApp(), param.getStream(), callId, null);
221   - });
222   - }
223   - }
224   - }else {
225   - zlmMediaListManager.sendStreamEvent(param.getApp(),param.getStream(), param.getMediaServerId());
226   - }
227   -
228   - ret.put("code", 0);
229   - ret.put("msg", "success");
230   -
231   - if (!"rtp".equals(param.getApp())) {
232   - ret.put("enable_audio", true);
233   - }
234   -
235   - taskExecutor.execute(()->{
236   - ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_publish, json);
237   - if (subscribe != null) {
238   - if (mediaInfo != null) {
239   - subscribe.response(mediaInfo, json);
240   - }else {
241   - ret.put("code", 1);
242   - ret.put("msg", "zlm not register");
243   - }
244   - }
245   - });
246   -
247   - if ("rtp".equals(param.getApp())) {
248   - ret.put("enable_mp4", userSetting.getRecordSip());
249   - }else {
250   - ret.put("enable_mp4", userSetting.isRecordPushLive());
251   - }
252   - List<SsrcTransaction> ssrcTransactionForAll = sessionManager.getSsrcTransactionForAll(null, null, null, param.getStream());
253   - if (ssrcTransactionForAll != null && ssrcTransactionForAll.size() == 1) {
254   - String deviceId = ssrcTransactionForAll.get(0).getDeviceId();
255   - String channelId = ssrcTransactionForAll.get(0).getChannelId();
256   - DeviceChannel deviceChannel = storager.queryChannel(deviceId, channelId);
257   - if (deviceChannel != null) {
258   - ret.put("enable_audio", deviceChannel.isHasAudio());
259   - }
260   - // 如果是录像下载就设置视频间隔十秒
261   - if (ssrcTransactionForAll.get(0).getType() == VideoStreamSessionManager.SessionType.download) {
262   - ret.put("mp4_max_second", 10);
263   - ret.put("enable_mp4", true);
264   - ret.put("enable_audio", true);
265   - }
266   - }
267   - return ret;
268   - }
269   -
270   - /**
271   - * rtsp/rtmp流注册或注销时触发此事件；此事件对回复不敏感。
272   - *
273   - */
274   - @ResponseBody
275   - @PostMapping(value = "/on_stream_changed", produces = "application/json;charset=UTF-8")
276   - public JSONObject onStreamChanged(@RequestBody OnStreamChangedHookParam param){
277   -
278   - if (param.isRegist()) {
279   - logger.info("[ZLM HOOK] 流注册, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
280   - }else {
281   - logger.info("[ZLM HOOK] 流注销, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
282   - }
283   -
284   -
285   - JSONObject json = (JSONObject) JSON.toJSON(param);
286   - taskExecutor.execute(()->{
287   - ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_stream_changed, json);
288   - if (subscribe != null ) {
289   - MediaServerItem mediaInfo = mediaServerService.getOne(param.getMediaServerId());
290   - if (mediaInfo != null) {
291   - subscribe.response(mediaInfo, json);
292   - }
293   - }
294   - // 流消失移除redis play
295   - List<OnStreamChangedHookParam.MediaTrack> tracks = param.getTracks();
296   - if (param.isRegist()) {
297   - if (param.getOriginType() == OriginType.RTMP_PUSH.ordinal()
298   - || param.getOriginType() == OriginType.RTSP_PUSH.ordinal()
299   - || param.getOriginType() == OriginType.RTC_PUSH.ordinal()) {
300   -
301   - StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream());
302   - if (streamAuthorityInfo == null) {
303   - streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(param);
304   - }else {
305   - streamAuthorityInfo.setOriginType(param.getOriginType());
306   - streamAuthorityInfo.setOriginTypeStr(param.getOriginTypeStr());
307   - }
308   - redisCatchStorage.updateStreamAuthorityInfo(param.getApp(), param.getStream(), streamAuthorityInfo);
309   - }
310   - }else {
311   - redisCatchStorage.removeStreamAuthorityInfo(param.getApp(), param.getStream());
312   - }
313   -
314   - if ("rtsp".equals(param.getSchema())){
315   - if (param.isRegist()) {
316   - mediaServerService.addCount(param.getMediaServerId());
317   - }else {
318   - mediaServerService.removeCount(param.getMediaServerId());
319   - }
320   - if (param.getOriginType() == OriginType.PULL.ordinal()
321   - || param.getOriginType() == OriginType.FFMPEG_PULL.ordinal()) {
322   - // 设置拉流代理上线/离线
323   - streamProxyService.updateStatus(param.isRegist(), param.getApp(), param.getStream());
324   - }
325   - if ("rtp".equals(param.getApp()) && !param.isRegist() ) {
326   - StreamInfo streamInfo = redisCatchStorage.queryPlayByStreamId(param.getStream());
327   - if (streamInfo!=null){
328   - redisCatchStorage.stopPlay(streamInfo);
329   - storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId());
330   - }else{
331   - streamInfo = redisCatchStorage.queryPlayback(null, null, param.getStream(), null);
332   - if (streamInfo != null) {
333   - redisCatchStorage.stopPlayback(streamInfo.getDeviceID(), streamInfo.getChannelId(),
334   - streamInfo.getStream(), null);
335   - }
336   - }
337   - }else {
338   - if (!"rtp".equals(param.getApp())){
339   - String type = OriginType.values()[param.getOriginType()].getType();
340   - MediaServerItem mediaServerItem = mediaServerService.getOne(param.getMediaServerId());
341   -
342   - if (mediaServerItem != null){
343   - if (param.isRegist()) {
344   - StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream());
345   - String callId = null;
346   - if (streamAuthorityInfo != null) {
347   - callId = streamAuthorityInfo.getCallId();
348   - }
349   - StreamInfo streamInfoByAppAndStream = mediaService.getStreamInfoByAppAndStream(mediaServerItem,
350   - param.getApp(), param.getStream(), tracks, callId);
351   - param.setStreamInfo(new StreamContent(streamInfoByAppAndStream));
352   - redisCatchStorage.addStream(mediaServerItem, type, param.getApp(), param.getStream(), param);
353   - if (param.getOriginType() == OriginType.RTSP_PUSH.ordinal()
354   - || param.getOriginType() == OriginType.RTMP_PUSH.ordinal()
355   - || param.getOriginType() == OriginType.RTC_PUSH.ordinal() ) {
356   - param.setSeverId(userSetting.getServerId());
357   - zlmMediaListManager.addPush(param);
358   - }
359   - }else {
360   - // 兼容流注销时类型从redis记录获取
361   - OnStreamChangedHookParam onStreamChangedHookParam = redisCatchStorage.getStreamInfo(param.getApp(), param.getStream(), param.getMediaServerId());
362   - if (onStreamChangedHookParam != null) {
363   - type = OriginType.values()[onStreamChangedHookParam.getOriginType()].getType();
364   - redisCatchStorage.removeStream(mediaServerItem.getId(), type, param.getApp(), param.getStream());
365   - }
366   - GbStream gbStream = storager.getGbStream(param.getApp(), param.getStream());
367   - if (gbStream != null) {
  56 + @Autowired
  57 + private SIPCommander cmder;
  58 +
  59 + @Autowired
  60 + private SIPCommanderFroPlatform commanderFroPlatform;
  61 +
  62 + @Autowired
  63 + private IPlayService playService;
  64 +
  65 + @Autowired
  66 + private IVideoManagerStorage storager;
  67 +
  68 + @Autowired
  69 + private IRedisCatchStorage redisCatchStorage;
  70 +
  71 + @Autowired
  72 + private IDeviceService deviceService;
  73 +
  74 + @Autowired
  75 + private IMediaServerService mediaServerService;
  76 +
  77 + @Autowired
  78 + private IStreamProxyService streamProxyService;
  79 +
  80 + @Autowired
  81 + private DeferredResultHolder resultHolder;
  82 +
  83 + @Autowired
  84 + private IMediaService mediaService;
  85 +
  86 + @Autowired
  87 + private EventPublisher eventPublisher;
  88 +
  89 + @Autowired
  90 + private ZLMMediaListManager zlmMediaListManager;
  91 +
  92 + @Autowired
  93 + private ZlmHttpHookSubscribe subscribe;
  94 +
  95 + @Autowired
  96 + private UserSetting userSetting;
  97 +
  98 + @Autowired
  99 + private IUserService userService;
  100 +
  101 + @Autowired
  102 + private VideoStreamSessionManager sessionManager;
  103 +
  104 + @Autowired
  105 + private AssistRESTfulUtils assistRESTfulUtils;
  106 +
  107 + @Qualifier("taskExecutor")
  108 + @Autowired
  109 + private ThreadPoolTaskExecutor taskExecutor;
  110 +
  111 + /**
  112 + * 服务器定时上报时间，上报间隔可配置，默认10s上报一次
  113 + */
  114 + @ResponseBody
  115 + @PostMapping(value = "/on_server_keepalive", produces = "application/json;charset=UTF-8")
  116 + public HookResult onServerKeepalive(@RequestBody OnServerKeepaliveHookParam param) {
  117 +
  118 + logger.info("[ZLM HOOK] 收到zlm心跳：" + param.getMediaServerId());
  119 +
  120 + taskExecutor.execute(() -> {
  121 + List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_keepalive);
  122 + JSONObject json = (JSONObject) JSON.toJSON(param);
  123 + if (subscribes != null && subscribes.size() > 0) {
  124 + for (ZlmHttpHookSubscribe.Event subscribe : subscribes) {
  125 + subscribe.response(null, json);
  126 + }
  127 + }
  128 + });
  129 + mediaServerService.updateMediaServerKeepalive(param.getMediaServerId(), param.getData());
  130 +
  131 + return HookResult.SUCCESS();
  132 + }
  133 +
  134 + /**
  135 + * 播放器鉴权事件，rtsp/rtmp/http-flv/ws-flv/hls的播放都将触发此鉴权事件。
  136 + */
  137 + @ResponseBody
  138 + @PostMapping(value = "/on_play", produces = "application/json;charset=UTF-8")
  139 + public HookResult onPlay(@RequestBody OnPlayHookParam param) {
  140 + if (logger.isDebugEnabled()) {
  141 + logger.debug("[ZLM HOOK] 播放鉴权：{}->{}" + param.getMediaServerId(), param);
  142 + }
  143 + String mediaServerId = param.getMediaServerId();
  144 +
  145 + taskExecutor.execute(() -> {
  146 + JSONObject json = (JSONObject) JSON.toJSON(param);
  147 + ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_play, json);
  148 + if (subscribe != null) {
  149 + MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId);
  150 + if (mediaInfo != null) {
  151 + subscribe.response(mediaInfo, json);
  152 + }
  153 + }
  154 + });
  155 + if (!"rtp".equals(param.getApp())) {
  156 + Map<String, String> paramMap = urlParamToMap(param.getParams());
  157 + StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream());
  158 + if (streamAuthorityInfo != null && streamAuthorityInfo.getCallId() != null && !streamAuthorityInfo.getCallId().equals(paramMap.get("callId"))) {
  159 + return new HookResult(401, "Unauthorized");
  160 + }
  161 + }
  162 +
  163 + return HookResult.SUCCESS();
  164 + }
  165 +
  166 + /**
  167 + * rtsp/rtmp/rtp推流鉴权事件。
  168 + */
  169 + @ResponseBody
  170 + @PostMapping(value = "/on_publish", produces = "application/json;charset=UTF-8")
  171 + public HookResultForOnPublish onPublish(@RequestBody OnPublishHookParam param) {
  172 +
  173 + JSONObject json = (JSONObject) JSON.toJSON(param);
  174 +
  175 + logger.info("[ZLM HOOK]推流鉴权：{}->{}", param.getMediaServerId(), param);
  176 +
  177 + String mediaServerId = json.getString("mediaServerId");
  178 + MediaServerItem mediaInfo = mediaServerService.getOne(mediaServerId);
  179 +
  180 + if (!"rtp".equals(param.getApp())) {
  181 + if (userSetting.getPushAuthority()) {
  182 + // 推流鉴权
  183 + if (param.getParams() == null) {
  184 + logger.info("推流鉴权失败： 缺少不要参数：sign=md5(user表的pushKey)");
  185 + return new HookResultForOnPublish(401, "Unauthorized");
  186 + }
  187 + Map<String, String> paramMap = urlParamToMap(param.getParams());
  188 + String sign = paramMap.get("sign");
  189 + if (sign == null) {
  190 + logger.info("推流鉴权失败： 缺少不要参数：sign=md5(user表的pushKey)");
  191 + return new HookResultForOnPublish(401, "Unauthorized");
  192 + }
  193 + // 推流自定义播放鉴权码
  194 + String callId = paramMap.get("callId");
  195 + // 鉴权配置
  196 + boolean hasAuthority = userService.checkPushAuthority(callId, sign);
  197 + if (!hasAuthority) {
  198 + logger.info("推流鉴权失败： sign 无权限: callId={}. sign={}", callId, sign);
  199 + return new HookResultForOnPublish(401, "Unauthorized");
  200 + }
  201 + StreamAuthorityInfo streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(param);
  202 + streamAuthorityInfo.setCallId(callId);
  203 + streamAuthorityInfo.setSign(sign);
  204 + // 鉴权通过
  205 + redisCatchStorage.updateStreamAuthorityInfo(param.getApp(), param.getStream(), streamAuthorityInfo);
  206 + // 通知assist新的callId
  207 + if (mediaInfo != null && mediaInfo.getRecordAssistPort() > 0) {
  208 + taskExecutor.execute(() -> {
  209 + assistRESTfulUtils.addStreamCallInfo(mediaInfo, param.getApp(), param.getStream(), callId, null);
  210 + });
  211 + }
  212 + }
  213 + } else {
  214 + zlmMediaListManager.sendStreamEvent(param.getApp(), param.getStream(), param.getMediaServerId());
  215 + }
  216 +
  217 +
  218 + HookResultForOnPublish result = HookResultForOnPublish.SUCCESS();
  219 + if (!"rtp".equals(param.getApp())) {
  220 + result.setEnable_audio(true);
  221 + }
  222 +
  223 + taskExecutor.execute(() -> {
  224 + ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_publish, json);
  225 + if (subscribe != null) {
  226 + if (mediaInfo != null) {
  227 + subscribe.response(mediaInfo, json);
  228 + } else {
  229 + new HookResultForOnPublish(1, "zlm not register");
  230 + }
  231 + }
  232 + });
  233 +
  234 + if ("rtp".equals(param.getApp())) {
  235 + result.setEnable_mp4(userSetting.getRecordSip());
  236 + } else {
  237 + result.setEnable_mp4(userSetting.isRecordPushLive());
  238 + }
  239 + List<SsrcTransaction> ssrcTransactionForAll = sessionManager.getSsrcTransactionForAll(null, null, null, param.getStream());
  240 + if (ssrcTransactionForAll != null && ssrcTransactionForAll.size() == 1) {
  241 + String deviceId = ssrcTransactionForAll.get(0).getDeviceId();
  242 + String channelId = ssrcTransactionForAll.get(0).getChannelId();
  243 + DeviceChannel deviceChannel = storager.queryChannel(deviceId, channelId);
  244 + if (deviceChannel != null) {
  245 + result.setEnable_audio(deviceChannel.isHasAudio());
  246 + }
  247 + // 如果是录像下载就设置视频间隔十秒
  248 + if (ssrcTransactionForAll.get(0).getType() == VideoStreamSessionManager.SessionType.download) {
  249 + result.setMp4_max_second(10);
  250 + result.setEnable_audio(true);
  251 + result.setEnable_mp4(true);
  252 + }
  253 + }
  254 + return result;
  255 + }
  256 +
  257 + /**
  258 + * rtsp/rtmp流注册或注销时触发此事件；此事件对回复不敏感。
  259 + */
  260 + @ResponseBody
  261 + @PostMapping(value = "/on_stream_changed", produces = "application/json;charset=UTF-8")
  262 + public HookResult onStreamChanged(@RequestBody OnStreamChangedHookParam param) {
  263 +
  264 + if (param.isRegist()) {
  265 + logger.info("[ZLM HOOK] 流注册, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
  266 + } else {
  267 + logger.info("[ZLM HOOK] 流注销, {}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
  268 + }
  269 +
  270 +
  271 + JSONObject json = (JSONObject) JSON.toJSON(param);
  272 + taskExecutor.execute(() -> {
  273 + ZlmHttpHookSubscribe.Event subscribe = this.subscribe.sendNotify(HookType.on_stream_changed, json);
  274 + if (subscribe != null) {
  275 + MediaServerItem mediaInfo = mediaServerService.getOne(param.getMediaServerId());
  276 + if (mediaInfo != null) {
  277 + subscribe.response(mediaInfo, json);
  278 + }
  279 + }
  280 + // 流消失移除redis play
  281 + List<OnStreamChangedHookParam.MediaTrack> tracks = param.getTracks();
  282 + if (param.isRegist()) {
  283 + if (param.getOriginType() == OriginType.RTMP_PUSH.ordinal()
  284 + || param.getOriginType() == OriginType.RTSP_PUSH.ordinal()
  285 + || param.getOriginType() == OriginType.RTC_PUSH.ordinal()) {
  286 +
  287 + StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream());
  288 + if (streamAuthorityInfo == null) {
  289 + streamAuthorityInfo = StreamAuthorityInfo.getInstanceByHook(param);
  290 + } else {
  291 + streamAuthorityInfo.setOriginType(param.getOriginType());
  292 + streamAuthorityInfo.setOriginTypeStr(param.getOriginTypeStr());
  293 + }
  294 + redisCatchStorage.updateStreamAuthorityInfo(param.getApp(), param.getStream(), streamAuthorityInfo);
  295 + }
  296 + } else {
  297 + redisCatchStorage.removeStreamAuthorityInfo(param.getApp(), param.getStream());
  298 + }
  299 +
  300 + if ("rtsp".equals(param.getSchema())) {
  301 + if (param.isRegist()) {
  302 + mediaServerService.addCount(param.getMediaServerId());
  303 + } else {
  304 + mediaServerService.removeCount(param.getMediaServerId());
  305 + }
  306 + if (param.getOriginType() == OriginType.PULL.ordinal()
  307 + || param.getOriginType() == OriginType.FFMPEG_PULL.ordinal()) {
  308 + // 设置拉流代理上线/离线
  309 + streamProxyService.updateStatus(param.isRegist(), param.getApp(), param.getStream());
  310 + }
  311 + if ("rtp".equals(param.getApp()) && !param.isRegist()) {
  312 + StreamInfo streamInfo = redisCatchStorage.queryPlayByStreamId(param.getStream());
  313 + if (streamInfo != null) {
  314 + redisCatchStorage.stopPlay(streamInfo);
  315 + storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId());
  316 + } else {
  317 + streamInfo = redisCatchStorage.queryPlayback(null, null, param.getStream(), null);
  318 + if (streamInfo != null) {
  319 + redisCatchStorage.stopPlayback(streamInfo.getDeviceID(), streamInfo.getChannelId(),
  320 + streamInfo.getStream(), null);
  321 + }
  322 + }
  323 + } else {
  324 + if (!"rtp".equals(param.getApp())) {
  325 + String type = OriginType.values()[param.getOriginType()].getType();
  326 + MediaServerItem mediaServerItem = mediaServerService.getOne(param.getMediaServerId());
  327 +
  328 + if (mediaServerItem != null) {
  329 + if (param.isRegist()) {
  330 + StreamAuthorityInfo streamAuthorityInfo = redisCatchStorage.getStreamAuthorityInfo(param.getApp(), param.getStream());
  331 + String callId = null;
  332 + if (streamAuthorityInfo != null) {
  333 + callId = streamAuthorityInfo.getCallId();
  334 + }
  335 + StreamInfo streamInfoByAppAndStream = mediaService.getStreamInfoByAppAndStream(mediaServerItem,
  336 + param.getApp(), param.getStream(), tracks, callId);
  337 + param.setStreamInfo(new StreamContent(streamInfoByAppAndStream));
  338 + redisCatchStorage.addStream(mediaServerItem, type, param.getApp(), param.getStream(), param);
  339 + if (param.getOriginType() == OriginType.RTSP_PUSH.ordinal()
  340 + || param.getOriginType() == OriginType.RTMP_PUSH.ordinal()
  341 + || param.getOriginType() == OriginType.RTC_PUSH.ordinal()) {
  342 + param.setSeverId(userSetting.getServerId());
  343 + zlmMediaListManager.addPush(param);
  344 + }
  345 + } else {
  346 + // 兼容流注销时类型从redis记录获取
  347 + OnStreamChangedHookParam onStreamChangedHookParam = redisCatchStorage.getStreamInfo(param.getApp(), param.getStream(), param.getMediaServerId());
  348 + if (onStreamChangedHookParam != null) {
  349 + type = OriginType.values()[onStreamChangedHookParam.getOriginType()].getType();
  350 + redisCatchStorage.removeStream(mediaServerItem.getId(), type, param.getApp(), param.getStream());
  351 + }
  352 + GbStream gbStream = storager.getGbStream(param.getApp(), param.getStream());
  353 + if (gbStream != null) {
368 354 // eventPublisher.catalogEventPublishForStream(null, gbStream, CatalogEvent.OFF);
369   - }
370   - zlmMediaListManager.removeMedia(param.getApp(), param.getStream());
371   - }
372   - if (type != null) {
373   - // 发送流变化redis消息
374   - JSONObject jsonObject = new JSONObject();
375   - jsonObject.put("serverId", userSetting.getServerId());
376   - jsonObject.put("app", param.getApp());
377   - jsonObject.put("stream", param.getStream());
378   - jsonObject.put("register", param.isRegist());
379   - jsonObject.put("mediaServerId", param.getMediaServerId());
380   - redisCatchStorage.sendStreamChangeMsg(type, jsonObject);
381   - }
382   - }
383   - }
384   - }
385   - if (!param.isRegist()) {
386   - List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(param.getStream());
387   - if (sendRtpItems.size() > 0) {
388   - for (SendRtpItem sendRtpItem : sendRtpItems) {
389   - if (sendRtpItem.getApp().equals(param.getApp())) {
390   - String platformId = sendRtpItem.getPlatformId();
391   - ParentPlatform platform = storager.queryParentPlatByServerGBId(platformId);
392   - Device device = deviceService.getDevice(platformId);
393   -
394   - try {
395   - if (platform != null) {
396   - commanderFroPlatform.streamByeCmd(platform, sendRtpItem);
397   - }else {
398   - cmder.streamByeCmd(device, sendRtpItem.getChannelId(), param.getStream(), sendRtpItem.getCallId());
399   - }
400   - } catch (SipException | InvalidArgumentException | ParseException | SsrcTransactionNotFoundException e) {
401   - logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
402   - }
403   - }
404   - }
405   - }
406   - }
407   - }
408   - });
409   -
410   - JSONObject ret = new JSONObject();
411   - ret.put("code", 0);
412   - ret.put("msg", "success");
413   - return ret;
414   - }
415   -
416   - /**
417   - * 流无人观看时事件，用户可以通过此事件选择是否关闭无人看的流。
418   - *
419   - */
420   - @ResponseBody
421   - @PostMapping(value = "/on_stream_none_reader", produces = "application/json;charset=UTF-8")
422   - public JSONObject onStreamNoneReader(@RequestBody OnStreamNoneReaderHookParam param){
423   -
424   - logger.info("[ZLM HOOK]流无人观看：{]->{}->{}/{}" + param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
425   - JSONObject ret = new JSONObject();
426   - ret.put("code", 0);
427   - // 国标类型的流
428   - if ("rtp".equals(param.getApp())){
429   - ret.put("close", userSetting.getStreamOnDemand());
430   - // 国标流， 点播/录像回放/录像下载
431   - StreamInfo streamInfoForPlayCatch = redisCatchStorage.queryPlayByStreamId(param.getStream());
432   - // 点播
433   - if (streamInfoForPlayCatch != null) {
434   - // 收到无人观看说明流也没有在往上级推送
435   - if (redisCatchStorage.isChannelSendingRTP(streamInfoForPlayCatch.getChannelId())) {
436   - List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByChnnelId(streamInfoForPlayCatch.getChannelId());
437   - if (sendRtpItems.size() > 0) {
438   - for (SendRtpItem sendRtpItem : sendRtpItems) {
439   - ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId());
440   - try {
441   - commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
442   - } catch (SipException | InvalidArgumentException | ParseException e) {
443   - logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
444   - }
445   - redisCatchStorage.deleteSendRTPServer(parentPlatform.getServerGBId(), sendRtpItem.getChannelId(),
446   - sendRtpItem.getCallId(), sendRtpItem.getStreamId());
447   - }
448   - }
449   - }
450   - Device device = deviceService.getDevice(streamInfoForPlayCatch.getDeviceID());
451   - if (device != null) {
452   - try {
453   - cmder.streamByeCmd(device, streamInfoForPlayCatch.getChannelId(),
454   - streamInfoForPlayCatch.getStream(), null);
455   - } catch (InvalidArgumentException | ParseException | SipException | SsrcTransactionNotFoundException e) {
456   - logger.error("[无人观看]点播， 发送BYE失败 {}", e.getMessage());
457   - }
458   - }
459   -
460   - redisCatchStorage.stopPlay(streamInfoForPlayCatch);
461   - storager.stopPlay(streamInfoForPlayCatch.getDeviceID(), streamInfoForPlayCatch.getChannelId());
462   - return ret;
463   - }
464   - // 录像回放
465   - StreamInfo streamInfoForPlayBackCatch = redisCatchStorage.queryPlayback(null, null, param.getStream(), null);
466   - if (streamInfoForPlayBackCatch != null ) {
467   - if (streamInfoForPlayBackCatch.isPause()) {
468   - ret.put("close", false);
469   - }else {
470   - Device device = deviceService.getDevice(streamInfoForPlayBackCatch.getDeviceID());
471   - if (device != null) {
472   - try {
473   - cmder.streamByeCmd(device,streamInfoForPlayBackCatch.getChannelId(),
474   - streamInfoForPlayBackCatch.getStream(), null);
475   - } catch (InvalidArgumentException | ParseException | SipException |
476   - SsrcTransactionNotFoundException e) {
477   - logger.error("[无人观看]回放， 发送BYE失败 {}", e.getMessage());
478   - }
479   - }
480   - redisCatchStorage.stopPlayback(streamInfoForPlayBackCatch.getDeviceID(),
481   - streamInfoForPlayBackCatch.getChannelId(), streamInfoForPlayBackCatch.getStream(), null);
482   - }
483   - return ret;
484   - }
485   - // 录像下载
486   - StreamInfo streamInfoForDownload = redisCatchStorage.queryDownload(null, null, param.getStream(), null);
487   - // 进行录像下载时无人观看不断流
488   - if (streamInfoForDownload != null) {
489   - ret.put("close", false);
490   - return ret;
491   - }
492   - }else {
493   - // 非国标流 推流/拉流代理
494   - // 拉流代理
495   - StreamProxyItem streamProxyItem = streamProxyService.getStreamProxyByAppAndStream(param.getApp(), param.getStream());
496   - if (streamProxyItem != null ) {
497   - if (streamProxyItem.isEnable_remove_none_reader()) {
498   - // 无人观看自动移除
499   - ret.put("close", true);
500   - streamProxyService.del(param.getApp(), param.getStream());
501   - String url = streamProxyItem.getUrl() != null?streamProxyItem.getUrl():streamProxyItem.getSrc_url();
502   - logger.info("[{}/{}]<-[{}] 拉流代理无人观看已经移除", param.getApp(), param.getStream(), url);
503   - }else if (streamProxyItem.isEnable_disable_none_reader()) {
504   - // 无人观看停用
505   - ret.put("close", true);
506   - // 修改数据
507   - streamProxyService.stop(param.getApp(), param.getStream());
508   - }else {
509   - // 无人观看不做处理
510   - ret.put("close", false);
511   - }
512   - return ret;
513   - }
514   - // 推流具有主动性，暂时不做处理
  355 + }
  356 + zlmMediaListManager.removeMedia(param.getApp(), param.getStream());
  357 + }
  358 + if (type != null) {
  359 + // 发送流变化redis消息
  360 + JSONObject jsonObject = new JSONObject();
  361 + jsonObject.put("serverId", userSetting.getServerId());
  362 + jsonObject.put("app", param.getApp());
  363 + jsonObject.put("stream", param.getStream());
  364 + jsonObject.put("register", param.isRegist());
  365 + jsonObject.put("mediaServerId", param.getMediaServerId());
  366 + redisCatchStorage.sendStreamChangeMsg(type, jsonObject);
  367 + }
  368 + }
  369 + }
  370 + }
  371 + if (!param.isRegist()) {
  372 + List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(param.getStream());
  373 + if (sendRtpItems.size() > 0) {
  374 + for (SendRtpItem sendRtpItem : sendRtpItems) {
  375 + if (sendRtpItem.getApp().equals(param.getApp())) {
  376 + String platformId = sendRtpItem.getPlatformId();
  377 + ParentPlatform platform = storager.queryParentPlatByServerGBId(platformId);
  378 + Device device = deviceService.getDevice(platformId);
  379 +
  380 + try {
  381 + if (platform != null) {
  382 + commanderFroPlatform.streamByeCmd(platform, sendRtpItem);
  383 + } else {
  384 + cmder.streamByeCmd(device, sendRtpItem.getChannelId(), param.getStream(), sendRtpItem.getCallId());
  385 + }
  386 + } catch (SipException | InvalidArgumentException | ParseException |
  387 + SsrcTransactionNotFoundException e) {
  388 + logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
  389 + }
  390 + }
  391 + }
  392 + }
  393 + }
  394 + }
  395 + });
  396 +
  397 + return HookResult.SUCCESS();
  398 + }
  399 +
  400 + /**
  401 + * 流无人观看时事件，用户可以通过此事件选择是否关闭无人看的流。
  402 + */
  403 + @ResponseBody
  404 + @PostMapping(value = "/on_stream_none_reader", produces = "application/json;charset=UTF-8")
  405 + public JSONObject onStreamNoneReader(@RequestBody OnStreamNoneReaderHookParam param) {
  406 +
  407 + logger.info("[ZLM HOOK]流无人观看：{]->{}->{}/{}" + param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
  408 + JSONObject ret = new JSONObject();
  409 + ret.put("code", 0);
  410 + // 国标类型的流
  411 + if ("rtp".equals(param.getApp())) {
  412 + ret.put("close", userSetting.getStreamOnDemand());
  413 + // 国标流， 点播/录像回放/录像下载
  414 + StreamInfo streamInfoForPlayCatch = redisCatchStorage.queryPlayByStreamId(param.getStream());
  415 + // 点播
  416 + if (streamInfoForPlayCatch != null) {
  417 + // 收到无人观看说明流也没有在往上级推送
  418 + if (redisCatchStorage.isChannelSendingRTP(streamInfoForPlayCatch.getChannelId())) {
  419 + List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByChnnelId(streamInfoForPlayCatch.getChannelId());
  420 + if (sendRtpItems.size() > 0) {
  421 + for (SendRtpItem sendRtpItem : sendRtpItems) {
  422 + ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId());
  423 + try {
  424 + commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
  425 + } catch (SipException | InvalidArgumentException | ParseException e) {
  426 + logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
  427 + }
  428 + redisCatchStorage.deleteSendRTPServer(parentPlatform.getServerGBId(), sendRtpItem.getChannelId(),
  429 + sendRtpItem.getCallId(), sendRtpItem.getStreamId());
  430 + }
  431 + }
  432 + }
  433 + Device device = deviceService.getDevice(streamInfoForPlayCatch.getDeviceID());
  434 + if (device != null) {
  435 + try {
  436 + cmder.streamByeCmd(device, streamInfoForPlayCatch.getChannelId(),
  437 + streamInfoForPlayCatch.getStream(), null);
  438 + } catch (InvalidArgumentException | ParseException | SipException |
  439 + SsrcTransactionNotFoundException e) {
  440 + logger.error("[无人观看]点播， 发送BYE失败 {}", e.getMessage());
  441 + }
  442 + }
  443 +
  444 + redisCatchStorage.stopPlay(streamInfoForPlayCatch);
  445 + storager.stopPlay(streamInfoForPlayCatch.getDeviceID(), streamInfoForPlayCatch.getChannelId());
  446 + return ret;
  447 + }
  448 + // 录像回放
  449 + StreamInfo streamInfoForPlayBackCatch = redisCatchStorage.queryPlayback(null, null, param.getStream(), null);
  450 + if (streamInfoForPlayBackCatch != null) {
  451 + if (streamInfoForPlayBackCatch.isPause()) {
  452 + ret.put("close", false);
  453 + } else {
  454 + Device device = deviceService.getDevice(streamInfoForPlayBackCatch.getDeviceID());
  455 + if (device != null) {
  456 + try {
  457 + cmder.streamByeCmd(device, streamInfoForPlayBackCatch.getChannelId(),
  458 + streamInfoForPlayBackCatch.getStream(), null);
  459 + } catch (InvalidArgumentException | ParseException | SipException |
  460 + SsrcTransactionNotFoundException e) {
  461 + logger.error("[无人观看]回放， 发送BYE失败 {}", e.getMessage());
  462 + }
  463 + }
  464 + redisCatchStorage.stopPlayback(streamInfoForPlayBackCatch.getDeviceID(),
  465 + streamInfoForPlayBackCatch.getChannelId(), streamInfoForPlayBackCatch.getStream(), null);
  466 + }
  467 + return ret;
  468 + }
  469 + // 录像下载
  470 + StreamInfo streamInfoForDownload = redisCatchStorage.queryDownload(null, null, param.getStream(), null);
  471 + // 进行录像下载时无人观看不断流
  472 + if (streamInfoForDownload != null) {
  473 + ret.put("close", false);
  474 + return ret;
  475 + }
  476 + } else {
  477 + // 非国标流 推流/拉流代理
  478 + // 拉流代理
  479 + StreamProxyItem streamProxyItem = streamProxyService.getStreamProxyByAppAndStream(param.getApp(), param.getStream());
  480 + if (streamProxyItem != null) {
  481 + if (streamProxyItem.isEnable_remove_none_reader()) {
  482 + // 无人观看自动移除
  483 + ret.put("close", true);
  484 + streamProxyService.del(param.getApp(), param.getStream());
  485 + String url = streamProxyItem.getUrl() != null ? streamProxyItem.getUrl() : streamProxyItem.getSrc_url();
  486 + logger.info("[{}/{}]<-[{}] 拉流代理无人观看已经移除", param.getApp(), param.getStream(), url);
  487 + } else if (streamProxyItem.isEnable_disable_none_reader()) {
  488 + // 无人观看停用
  489 + ret.put("close", true);
  490 + // 修改数据
  491 + streamProxyService.stop(param.getApp(), param.getStream());
  492 + } else {
  493 + // 无人观看不做处理
  494 + ret.put("close", false);
  495 + }
  496 + return ret;
  497 + }
  498 + // 推流具有主动性，暂时不做处理
515 499 // StreamPushItem streamPushItem = streamPushService.getPush(app, streamId);
516 500 // if (streamPushItem != null) {
517 501 // // TODO 发送停止
518 502 //
519 503 // }
520   - }
521   - return ret;
522   - }
523   -
524   - /**
525   - * 流未找到事件，用户可以在此事件触发时，立即去拉流，这样可以实现按需拉流；此事件对回复不敏感。
526   - *
527   - */
528   - @ResponseBody
529   - @PostMapping(value = "/on_stream_not_found", produces = "application/json;charset=UTF-8")
530   - public JSONObject onStreamNotFound(@RequestBody OnStreamNotFoundHookParam param){
531   - logger.info("[ZLM HOOK] 流未找到：{}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
532   - taskExecutor.execute(()->{
533   - MediaServerItem mediaInfo = mediaServerService.getOne(param.getMediaServerId());
534   - if (userSetting.isAutoApplyPlay() && mediaInfo != null) {
535   - if ("rtp".equals(param.getApp())) {
536   - if (mediaInfo.isRtpEnable()) {
537   - String[] s = param.getStream().split("_");
538   - if (s.length == 2) {
539   - String deviceId = s[0];
540   - String channelId = s[1];
541   - Device device = redisCatchStorage.getDevice(deviceId);
542   - if (device != null) {
543   - playService.play(mediaInfo,deviceId, channelId, null, null, null);
544   - }
545   - }
546   - }
547   - }else {
548   - // 拉流代理
549   - StreamProxyItem streamProxyByAppAndStream = streamProxyService.getStreamProxyByAppAndStream(param.getApp(), param.getStream());
550   - if (streamProxyByAppAndStream != null && streamProxyByAppAndStream.isEnable_disable_none_reader()) {
551   - streamProxyService.start(param.getApp(), param.getStream());
552   - }
553   - }
554   - }
555   - });
556   -
557   -
558   - JSONObject ret = new JSONObject();
559   - ret.put("code", 0);
560   - ret.put("msg", "success");
561   - return ret;
562   - }
563   -
564   - /**
565   - * 服务器启动事件，可以用于监听服务器崩溃重启；此事件对回复不敏感。
566   - *
567   - */
568   - @ResponseBody
569   - @PostMapping(value = "/on_server_started", produces = "application/json;charset=UTF-8")
570   - public JSONObject onServerStarted(HttpServletRequest request, @RequestBody JSONObject jsonObject){
571   -
572   - jsonObject.put("ip", request.getRemoteAddr());
573   - ZLMServerConfig zlmServerConfig = JSON.to(ZLMServerConfig.class, jsonObject);
574   - zlmServerConfig.setIp(request.getRemoteAddr());
575   - logger.info("[ZLM HOOK] zlm 启动 " + zlmServerConfig.getGeneralMediaServerId());
576   - taskExecutor.execute(()->{
577   - List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_started);
578   - if (subscribes != null && subscribes.size() > 0) {
579   - for (ZlmHttpHookSubscribe.Event subscribe : subscribes) {
580   - subscribe.response(null, jsonObject);
581   - }
582   - }
583   - mediaServerService.zlmServerOnline(zlmServerConfig);
584   - });
585   -
586   - JSONObject ret = new JSONObject();
587   - ret.put("code", 0);
588   - ret.put("msg", "success");
589   - return ret;
590   - }
591   -
592   - /**
593   - * 发送rtp(startSendRtp)被动关闭时回调
594   - */
595   - @ResponseBody
596   - @PostMapping(value = "/on_send_rtp_stopped", produces = "application/json;charset=UTF-8")
597   - public JSONObject onSendRtpStopped(HttpServletRequest request, @RequestBody OnSendRtpStoppedHookParam param){
598   -
599   - logger.info("[ZLM HOOK] rtp发送关闭：{}->{}/{}", param.getMediaServerId(), param.getApp(), param.getStream());
600   -
601   - JSONObject ret = new JSONObject();
602   - ret.put("code", 0);
603   - ret.put("msg", "success");
604   -
605   - // 查找对应的上级推流，发送停止
606   - if (!"rtp".equals(param.getApp())) {
607   - return ret;
608   - }
609   - taskExecutor.execute(()->{
610   - List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(param.getStream());
611   - if (sendRtpItems.size() > 0) {
612   - for (SendRtpItem sendRtpItem : sendRtpItems) {
613   - ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId());
614   - try {
615   - commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
616   - } catch (SipException | InvalidArgumentException | ParseException e) {
617   - logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
618   - }
619   - redisCatchStorage.deleteSendRTPServer(parentPlatform.getServerGBId(), sendRtpItem.getChannelId(),
620   - sendRtpItem.getCallId(), sendRtpItem.getStreamId());
621   - }
622   - }
623   - });
624   -
625   -
626   - return ret;
627   - }
628   -
629   - /**
630   - * rtpServer收流超时
631   - */
632   - @ResponseBody
633   - @PostMapping(value = "/on_rtp_server_timeout", produces = "application/json;charset=UTF-8")
634   - public JSONObject onRtpServerTimeout(HttpServletRequest request, @RequestBody OnRtpServerTimeoutHookParam param){
635   - logger.info("[ZLM HOOK] rtpServer收流超时：{}->{}({})", param.getMediaServerId(), param.getStream_id(), param.getSsrc());
636   -
637   - JSONObject ret = new JSONObject();
638   - ret.put("code", 0);
639   - ret.put("msg", "success");
640   -
641   - taskExecutor.execute(()->{
642   - JSONObject json = (JSONObject) JSON.toJSON(param);
643   - List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_rtp_server_timeout);
644   - if (subscribes != null && subscribes.size() > 0) {
645   - for (ZlmHttpHookSubscribe.Event subscribe : subscribes) {
646   - subscribe.response(null, json);
647   - }
648   - }
649   - });
650   -
651   - return ret;
652   - }
653   -
654   - private Map<String, String> urlParamToMap(String params) {
655   - HashMap<String, String> map = new HashMap<>();
656   - if (ObjectUtils.isEmpty(params)) {
657   - return map;
658   - }
659   - String[] paramsArray = params.split("&");
660   - if (paramsArray.length == 0) {
661   - return map;
662   - }
663   - for (String param : paramsArray) {
664   - String[] paramArray = param.split("=");
665   - if (paramArray.length == 2){
666   - map.put(paramArray[0], paramArray[1]);
667   - }
668   - }
669   - return map;
670   - }
  504 + }
  505 + return ret;
  506 + }
  507 +
  508 + /**
  509 + * 流未找到事件，用户可以在此事件触发时，立即去拉流，这样可以实现按需拉流；此事件对回复不敏感。
  510 + */
  511 + @ResponseBody
  512 + @PostMapping(value = "/on_stream_not_found", produces = "application/json;charset=UTF-8")
  513 + public DeferredResult<HookResult> onStreamNotFound(@RequestBody OnStreamNotFoundHookParam param) {
  514 + logger.info("[ZLM HOOK] 流未找到：{}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
  515 +
  516 + DeferredResult<HookResult> defaultResult = new DeferredResult<>();
  517 +
  518 + MediaServerItem mediaInfo = mediaServerService.getOne(param.getMediaServerId());
  519 + if (!userSetting.isAutoApplyPlay() || mediaInfo == null) {
  520 + defaultResult.setResult(new HookResult(ErrorCode.ERROR404.getCode(), ErrorCode.ERROR404.getMsg()));
  521 + return defaultResult;
  522 + }
  523 +
  524 + if ("rtp".equals(param.getApp())) {
  525 + String[] s = param.getStream().split("_");
  526 + if (!mediaInfo.isRtpEnable() || s.length != 2) {
  527 + defaultResult.setResult(HookResult.SUCCESS());
  528 + return defaultResult;
  529 + }
  530 + String deviceId = s[0];
  531 + String channelId = s[1];
  532 + Device device = redisCatchStorage.getDevice(deviceId);
  533 + if (device == null) {
  534 + defaultResult.setResult(new HookResult(ErrorCode.ERROR404.getCode(), ErrorCode.ERROR404.getMsg()));
  535 + return defaultResult;
  536 + }
  537 + DeviceChannel deviceChannel = storager.queryChannel(deviceId, channelId);
  538 + if (deviceChannel == null) {
  539 + defaultResult.setResult(new HookResult(ErrorCode.ERROR404.getCode(), ErrorCode.ERROR404.getMsg()));
  540 + return defaultResult;
  541 + }
  542 + logger.info("[ZLM HOOK] 流未找到, 发起自动点播：{}->{}->{}/{}", param.getMediaServerId(), param.getSchema(), param.getApp(), param.getStream());
  543 + RequestMessage msg = new RequestMessage();
  544 + String key = DeferredResultHolder.CALLBACK_CMD_PLAY + deviceId + channelId;
  545 + boolean exist = resultHolder.exist(key, null);
  546 + msg.setKey(key);
  547 + String uuid = UUID.randomUUID().toString();
  548 + msg.setId(uuid);
  549 + DeferredResult<HookResult> result = new DeferredResult<>(userSetting.getPlayTimeout().longValue());
  550 + DeferredResultEx<HookResult> deferredResultEx = new DeferredResultEx<>(result);
  551 +
  552 + result.onTimeout(() -> {
  553 + logger.info("点播接口等待超时");
  554 + // 释放rtpserver
  555 + msg.setData(new HookResult(ErrorCode.ERROR100.getCode(), "点播超时"));
  556 + resultHolder.invokeResult(msg);
  557 + });
  558 + // TODO 在点播未成功的情况下在此调用接口点播会导致返回的流地址ip错误
  559 + deferredResultEx.setFilter(result1 -> {
  560 + WVPResult<StreamInfo> wvpResult1 = (WVPResult<StreamInfo>) result1;
  561 + HookResult resultForEnd = new HookResult();
  562 + resultForEnd.setCode(wvpResult1.getCode());
  563 + resultForEnd.setMsg(wvpResult1.getMsg());
  564 + return resultForEnd;
  565 + });
  566 +
  567 + // 录像查询以channelId作为deviceId查询
  568 + resultHolder.put(key, uuid, deferredResultEx);
  569 +
  570 + if (!exist) {
  571 + playService.play(mediaInfo, deviceId, channelId, null, eventResult -> {
  572 + msg.setData(new HookResult(eventResult.statusCode, eventResult.msg));
  573 + resultHolder.invokeResult(msg);
  574 + }, null);
  575 + }
  576 + return result;
  577 + } else {
  578 + // 拉流代理
  579 + StreamProxyItem streamProxyByAppAndStream = streamProxyService.getStreamProxyByAppAndStream(param.getApp(), param.getStream());
  580 + if (streamProxyByAppAndStream != null && streamProxyByAppAndStream.isEnable_disable_none_reader()) {
  581 + streamProxyService.start(param.getApp(), param.getStream());
  582 + }
  583 + DeferredResult<HookResult> result = new DeferredResult<>();
  584 + result.setResult(HookResult.SUCCESS());
  585 + return result;
  586 + }
  587 + }
  588 +
  589 + /**
  590 + * 服务器启动事件，可以用于监听服务器崩溃重启；此事件对回复不敏感。
  591 + */
  592 + @ResponseBody
  593 + @PostMapping(value = "/on_server_started", produces = "application/json;charset=UTF-8")
  594 + public HookResult onServerStarted(HttpServletRequest request, @RequestBody JSONObject jsonObject) {
  595 +
  596 + jsonObject.put("ip", request.getRemoteAddr());
  597 + ZLMServerConfig zlmServerConfig = JSON.to(ZLMServerConfig.class, jsonObject);
  598 + zlmServerConfig.setIp(request.getRemoteAddr());
  599 + logger.info("[ZLM HOOK] zlm 启动 " + zlmServerConfig.getGeneralMediaServerId());
  600 + taskExecutor.execute(() -> {
  601 + List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_server_started);
  602 + if (subscribes != null && subscribes.size() > 0) {
  603 + for (ZlmHttpHookSubscribe.Event subscribe : subscribes) {
  604 + subscribe.response(null, jsonObject);
  605 + }
  606 + }
  607 + mediaServerService.zlmServerOnline(zlmServerConfig);
  608 + });
  609 +
  610 + return HookResult.SUCCESS();
  611 + }
  612 +
  613 + /**
  614 + * 发送rtp(startSendRtp)被动关闭时回调
  615 + */
  616 + @ResponseBody
  617 + @PostMapping(value = "/on_send_rtp_stopped", produces = "application/json;charset=UTF-8")
  618 + public HookResult onSendRtpStopped(HttpServletRequest request, @RequestBody OnSendRtpStoppedHookParam param) {
  619 +
  620 + logger.info("[ZLM HOOK] rtp发送关闭：{}->{}/{}", param.getMediaServerId(), param.getApp(), param.getStream());
  621 +
  622 + // 查找对应的上级推流，发送停止
  623 + if (!"rtp".equals(param.getApp())) {
  624 + return HookResult.SUCCESS();
  625 + }
  626 + taskExecutor.execute(() -> {
  627 + List<SendRtpItem> sendRtpItems = redisCatchStorage.querySendRTPServerByStream(param.getStream());
  628 + if (sendRtpItems.size() > 0) {
  629 + for (SendRtpItem sendRtpItem : sendRtpItems) {
  630 + ParentPlatform parentPlatform = storager.queryParentPlatByServerGBId(sendRtpItem.getPlatformId());
  631 + try {
  632 + commanderFroPlatform.streamByeCmd(parentPlatform, sendRtpItem.getCallId());
  633 + } catch (SipException | InvalidArgumentException | ParseException e) {
  634 + logger.error("[命令发送失败] 国标级联 发送BYE: {}", e.getMessage());
  635 + }
  636 + redisCatchStorage.deleteSendRTPServer(parentPlatform.getServerGBId(), sendRtpItem.getChannelId(),
  637 + sendRtpItem.getCallId(), sendRtpItem.getStreamId());
  638 + }
  639 + }
  640 + });
  641 +
  642 + return HookResult.SUCCESS();
  643 + }
  644 +
  645 + /**
  646 + * rtpServer收流超时
  647 + */
  648 + @ResponseBody
  649 + @PostMapping(value = "/on_rtp_server_timeout", produces = "application/json;charset=UTF-8")
  650 + public HookResult onRtpServerTimeout(HttpServletRequest request, @RequestBody OnRtpServerTimeoutHookParam param) {
  651 + logger.info("[ZLM HOOK] rtpServer收流超时：{}->{}({})", param.getMediaServerId(), param.getStream_id(), param.getSsrc());
  652 +
  653 + taskExecutor.execute(() -> {
  654 + JSONObject json = (JSONObject) JSON.toJSON(param);
  655 + List<ZlmHttpHookSubscribe.Event> subscribes = this.subscribe.getSubscribes(HookType.on_rtp_server_timeout);
  656 + if (subscribes != null && subscribes.size() > 0) {
  657 + for (ZlmHttpHookSubscribe.Event subscribe : subscribes) {
  658 + subscribe.response(null, json);
  659 + }
  660 + }
  661 + });
  662 +
  663 + return HookResult.SUCCESS();
  664 + }
  665 +
  666 + private Map<String, String> urlParamToMap(String params) {
  667 + HashMap<String, String> map = new HashMap<>();
  668 + if (ObjectUtils.isEmpty(params)) {
  669 + return map;
  670 + }
  671 + String[] paramsArray = params.split("&");
  672 + if (paramsArray.length == 0) {
  673 + return map;
  674 + }
  675 + for (String param : paramsArray) {
  676 + String[] paramArray = param.split("=");
  677 + if (paramArray.length == 2) {
  678 + map.put(paramArray[0], paramArray[1]);
  679 + }
  680 + }
  681 + return map;
  682 + }
671 683 }
... ...
src/main/java/com/genersoft/iot/vmp/media/zlm/dto/hook/HookResult.java 0 → 100644
  View file @c19ad94
  1 +package com.genersoft.iot.vmp.media.zlm.dto.hook;
  2 +
  3 +public class HookResult {
  4 +
  5 + private int code;
  6 + private String msg;
  7 +
  8 +
  9 + public HookResult() {
  10 + }
  11 +
  12 + public HookResult(int code, String msg) {
  13 + this.code = code;
  14 + this.msg = msg;
  15 + }
  16 +
  17 + public static HookResult SUCCESS(){
  18 + return new HookResult(0, "success");
  19 + }
  20 +
  21 + public int getCode() {
  22 + return code;
  23 + }
  24 +
  25 + public void setCode(int code) {
  26 + this.code = code;
  27 + }
  28 +
  29 + public String getMsg() {
  30 + return msg;
  31 + }
  32 +
  33 + public void setMsg(String msg) {
  34 + this.msg = msg;
  35 + }
  36 +}
... ...
src/main/java/com/genersoft/iot/vmp/media/zlm/dto/hook/HookResultForOnPublish.java 0 → 100644
  View file @c19ad94
  1 +package com.genersoft.iot.vmp.media.zlm.dto.hook;
  2 +
  3 +public class HookResultForOnPublish extends HookResult{
  4 +
  5 + private boolean enable_audio;
  6 + private boolean enable_mp4;
  7 + private int mp4_max_second;
  8 +
  9 + public HookResultForOnPublish() {
  10 + }
  11 +
  12 + public static HookResultForOnPublish SUCCESS(){
  13 + return new HookResultForOnPublish(0, "success");
  14 + }
  15 +
  16 + public HookResultForOnPublish(int code, String msg) {
  17 + setCode(code);
  18 + setMsg(msg);
  19 + }
  20 +
  21 + public boolean isEnable_audio() {
  22 + return enable_audio;
  23 + }
  24 +
  25 + public void setEnable_audio(boolean enable_audio) {
  26 + this.enable_audio = enable_audio;
  27 + }
  28 +
  29 + public boolean isEnable_mp4() {
  30 + return enable_mp4;
  31 + }
  32 +
  33 + public void setEnable_mp4(boolean enable_mp4) {
  34 + this.enable_mp4 = enable_mp4;
  35 + }
  36 +
  37 + public int getMp4_max_second() {
  38 + return mp4_max_second;
  39 + }
  40 +
  41 + public void setMp4_max_second(int mp4_max_second) {
  42 + this.mp4_max_second = mp4_max_second;
  43 + }
  44 +}
... ...
src/main/java/com/genersoft/iot/vmp/service/impl/GbStreamServiceImpl.java
  View file @c19ad94
... ... @@ -152,6 +152,10 @@ public class GbStreamServiceImpl implements IGbStreamService {
152 152  
153 153 @Override
154 154 public void sendCatalogMsg(GbStream gbStream, String type) {
  155 + if (gbStream == null || type == null) {
  156 + logger.warn("[发送目录订阅]类型：流信息或类型为NULL");
  157 + return;
  158 + }
155 159 List<GbStream> gbStreams = new ArrayList<>();
156 160 if (gbStream.getGbId() != null) {
157 161 gbStreams.add(gbStream);
... ...
src/main/java/com/genersoft/iot/vmp/service/impl/StreamPushServiceImpl.java
  View file @c19ad94
... ... @@ -184,7 +184,9 @@ public class StreamPushServiceImpl implements IStreamPushService {
184 184 @Override
185 185 public boolean stop(String app, String streamId) {
186 186 StreamPushItem streamPushItem = streamPushMapper.selectOne(app, streamId);
187   - gbStreamService.sendCatalogMsg(streamPushItem, CatalogEvent.DEL);
  187 + if (streamPushItem != null) {
  188 + gbStreamService.sendCatalogMsg(streamPushItem, CatalogEvent.DEL);
  189 + }
188 190  
189 191 platformGbStreamMapper.delByAppAndStream(app, streamId);
190 192 gbStreamMapper.del(app, streamId);
... ...
src/main/java/com/genersoft/iot/vmp/storager/IVideoManagerStorage.java
  View file @c19ad94
... ... @@ -2,7 +2,6 @@ package com.genersoft.iot.vmp.storager;
2 2  
3 3 import com.genersoft.iot.vmp.gb28181.bean.*;
4 4 import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem;
5   -import com.genersoft.iot.vmp.media.zlm.dto.StreamPushItem;
6 5 import com.genersoft.iot.vmp.service.bean.GPSMsgInfo;
7 6 import com.genersoft.iot.vmp.storager.dao.dto.ChannelSourceInfo;
8 7 import com.genersoft.iot.vmp.vmanager.gb28181.platform.bean.ChannelReduce;
... ... @@ -324,6 +323,8 @@ public interface IVideoManagerStorage {
324 323 */
325 324 boolean resetChannels(String deviceId, List<DeviceChannel> deviceChannelList);
326 325  
  326 + boolean updateChannels(String deviceId, List<DeviceChannel> deviceChannelList);
  327 +
327 328 /**
328 329 * 获取目录信息
329 330 * @param platformId
... ...
src/main/java/com/genersoft/iot/vmp/storager/impl/VideoManagerStorageImpl.java
  View file @c19ad94
... ... @@ -194,6 +194,119 @@ public class VideoManagerStorageImpl implements IVideoManagerStorage {
194 194  
195 195 }
196 196  
  197 +
  198 + @Override
  199 + public boolean updateChannels(String deviceId, List<DeviceChannel> deviceChannelList) {
  200 + if (CollectionUtils.isEmpty(deviceChannelList)) {
  201 + return false;
  202 + }
  203 + List<DeviceChannel> allChannels = deviceChannelMapper.queryAllChannels(deviceId);
  204 + Map<String,DeviceChannel> allChannelMap = new ConcurrentHashMap<>();
  205 + if (allChannels.size() > 0) {
  206 + for (DeviceChannel deviceChannel : allChannels) {
  207 + allChannelMap.put(deviceChannel.getChannelId(), deviceChannel);
  208 + }
  209 + }
  210 + List<DeviceChannel> addChannels = new ArrayList<>();
  211 + List<DeviceChannel> updateChannels = new ArrayList<>();
  212 +
  213 +
  214 + TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
  215 + // 数据去重
  216 + StringBuilder stringBuilder = new StringBuilder();
  217 + Map<String, Integer> subContMap = new HashMap<>();
  218 + if (deviceChannelList.size() > 0) {
  219 + // 数据去重
  220 + Set<String> gbIdSet = new HashSet<>();
  221 + for (DeviceChannel deviceChannel : deviceChannelList) {
  222 + if (!gbIdSet.contains(deviceChannel.getChannelId())) {
  223 + gbIdSet.add(deviceChannel.getChannelId());
  224 + if (allChannelMap.containsKey(deviceChannel.getChannelId())) {
  225 + deviceChannel.setStreamId(allChannelMap.get(deviceChannel.getChannelId()).getStreamId());
  226 + deviceChannel.setHasAudio(allChannelMap.get(deviceChannel.getChannelId()).isHasAudio());
  227 + updateChannels.add(deviceChannel);
  228 + }else {
  229 + addChannels.add(deviceChannel);
  230 + }
  231 + if (!ObjectUtils.isEmpty(deviceChannel.getParentId())) {
  232 + if (subContMap.get(deviceChannel.getParentId()) == null) {
  233 + subContMap.put(deviceChannel.getParentId(), 1);
  234 + }else {
  235 + Integer count = subContMap.get(deviceChannel.getParentId());
  236 + subContMap.put(deviceChannel.getParentId(), count++);
  237 + }
  238 + }
  239 + }else {
  240 + stringBuilder.append(deviceChannel.getChannelId()).append(",");
  241 + }
  242 + }
  243 + if (addChannels.size() > 0) {
  244 + for (DeviceChannel channel : addChannels) {
  245 + if (subContMap.get(channel.getChannelId()) != null){
  246 + channel.setSubCount(subContMap.get(channel.getChannelId()));
  247 + }
  248 + }
  249 + }
  250 + if (updateChannels.size() > 0) {
  251 + for (DeviceChannel channel : updateChannels) {
  252 + if (subContMap.get(channel.getChannelId()) != null){
  253 + channel.setSubCount(subContMap.get(channel.getChannelId()));
  254 + }
  255 + }
  256 + }
  257 +
  258 + }
  259 + if (stringBuilder.length() > 0) {
  260 + logger.info("[目录查询]收到的数据存在重复： {}" , stringBuilder);
  261 + }
  262 + if(CollectionUtils.isEmpty(updateChannels) && CollectionUtils.isEmpty(addChannels) ){
  263 + logger.info("通道更新，数据为空={}" , deviceChannelList);
  264 + return false;
  265 + }
  266 + try {
  267 + int limitCount = 300;
  268 + boolean result = false;
  269 + if (addChannels.size() > 0) {
  270 + if (addChannels.size() > limitCount) {
  271 + for (int i = 0; i < addChannels.size(); i += limitCount) {
  272 + int toIndex = i + limitCount;
  273 + if (i + limitCount > addChannels.size()) {
  274 + toIndex = addChannels.size();
  275 + }
  276 + result = result || deviceChannelMapper.batchAdd(addChannels.subList(i, toIndex)) < 0;
  277 + }
  278 + }else {
  279 + result = result || deviceChannelMapper.batchAdd(addChannels) < 0;
  280 + }
  281 + }
  282 + if (updateChannels.size() > 0) {
  283 + if (updateChannels.size() > limitCount) {
  284 + for (int i = 0; i < updateChannels.size(); i += limitCount) {
  285 + int toIndex = i + limitCount;
  286 + if (i + limitCount > updateChannels.size()) {
  287 + toIndex = updateChannels.size();
  288 + }
  289 + result = result || deviceChannelMapper.batchUpdate(updateChannels.subList(i, toIndex)) < 0;
  290 + }
  291 + }else {
  292 + result = result || deviceChannelMapper.batchUpdate(updateChannels) < 0;
  293 + }
  294 + }
  295 + if (result) {
  296 + //事务回滚
  297 + dataSourceTransactionManager.rollback(transactionStatus);
  298 + }else {
  299 + //手动提交
  300 + dataSourceTransactionManager.commit(transactionStatus);
  301 + }
  302 + return true;
  303 + }catch (Exception e) {
  304 + e.printStackTrace();
  305 + dataSourceTransactionManager.rollback(transactionStatus);
  306 + return false;
  307 + }
  308 + }
  309 +
197 310 @Override
198 311 public void deviceChannelOnline(String deviceId, String channelId) {
199 312 deviceChannelMapper.online(deviceId, channelId);
... ...
src/main/java/com/genersoft/iot/vmp/vmanager/user/UserController.java
  View file @c19ad94
... ... @@ -11,17 +11,13 @@ import com.genersoft.iot.vmp.utils.DateUtil;
11 11 import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
12 12 import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
13 13 import com.github.pagehelper.PageInfo;
14   -
15 14 import io.swagger.v3.oas.annotations.Operation;
16 15 import io.swagger.v3.oas.annotations.Parameter;
17 16 import io.swagger.v3.oas.annotations.tags.Tag;
18 17 import org.springframework.beans.factory.annotation.Autowired;
19   -import org.springframework.http.HttpStatus;
20   -import org.springframework.http.ResponseEntity;
21 18 import org.springframework.security.authentication.AuthenticationManager;
22 19 import org.springframework.util.DigestUtils;
23 20 import org.springframework.util.ObjectUtils;
24   -import org.springframework.util.StringUtils;
25 21 import org.springframework.web.bind.annotation.*;
26 22  
27 23 import javax.security.sasl.AuthenticationException;
... ... @@ -90,7 +86,7 @@ public class UserController {
90 86  
91 87  
92 88 @PostMapping("/add")
93   - @Operation(summary = "停止视频回放")
  89 + @Operation(summary = "添加用户")
94 90 @Parameter(name = "username", description = "用户名", required = true)
95 91 @Parameter(name = "password", description = "密码（未md5加密的密码）", required = true)
96 92 @Parameter(name = "roleId", description = "角色ID", required = true)
... ...