package com.genersoft.iot.vmp.service.impl;
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.genersoft.iot.vmp.common.StreamInfo;
|
import com.genersoft.iot.vmp.conf.UserSetup;
|
import com.genersoft.iot.vmp.gb28181.bean.Device;
|
import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
|
import com.genersoft.iot.vmp.gb28181.event.SipSubscribe;
|
import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager;
|
import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
|
import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
|
import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
|
import com.genersoft.iot.vmp.media.zlm.ZLMHttpHookSubscribe;
|
import com.genersoft.iot.vmp.media.zlm.ZLMRESTfulUtils;
|
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
|
import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
|
import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
|
import com.genersoft.iot.vmp.vmanager.gb28181.play.bean.PlayResult;
|
import com.genersoft.iot.vmp.service.IMediaService;
|
import com.genersoft.iot.vmp.service.IPlayService;
|
import gov.nist.javax.sip.stack.SIPDialog;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.HttpStatus;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.ResourceUtils;
|
import org.springframework.web.context.request.async.DeferredResult;
|
|
import javax.sip.ClientTransaction;
|
import javax.sip.message.Response;
|
import java.io.FileNotFoundException;
|
import java.util.UUID;
|
|
@SuppressWarnings(value = {"rawtypes", "unchecked"})
|
@Service
|
public class PlayServiceImpl implements IPlayService {
|
|
private final static Logger logger = LoggerFactory.getLogger(PlayServiceImpl.class);
|
|
@Autowired
|
private IVideoManagerStorager storager;
|
|
@Autowired
|
private SIPCommander cmder;
|
|
@Autowired
|
private IRedisCatchStorage redisCatchStorage;
|
|
@Autowired
|
private DeferredResultHolder resultHolder;
|
|
@Autowired
|
private ZLMRESTfulUtils zlmresTfulUtils;
|
|
@Autowired
|
private IMediaService mediaService;
|
|
@Autowired
|
private VideoStreamSessionManager streamSession;
|
|
@Autowired
|
private UserSetup userSetup;
|
|
|
@Override
|
public PlayResult play(String deviceId, String channelId, ZLMHttpHookSubscribe.Event hookEvent, SipSubscribe.Event errorEvent) {
|
PlayResult playResult = new PlayResult();
|
Device device = storager.queryVideoDevice(deviceId);
|
StreamInfo streamInfo = redisCatchStorage.queryPlayByDevice(deviceId, channelId);
|
playResult.setDevice(device);
|
UUID uuid = UUID.randomUUID();
|
playResult.setUuid(uuid.toString());
|
DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(userSetup.getPlayTimeout());
|
playResult.setResult(result);
|
// 录像查询以channelId作为deviceId查询
|
resultHolder.put(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid, result);
|
// 超时处理
|
result.onTimeout(()->{
|
logger.warn(String.format("设备点播超时，deviceId：%s ，channelId：%s", deviceId, channelId));
|
// 释放rtpserver
|
cmder.closeRTPServer(playResult.getDevice(), channelId);
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + playResult.getUuid());
|
WVPResult wvpResult = new WVPResult();
|
wvpResult.setCode(-1);
|
wvpResult.setMsg("Timeout");
|
msg.setData(wvpResult);
|
resultHolder.invokeResult(msg);
|
});
|
result.onCompletion(()->{
|
// 点播结束时调用截图接口
|
try {
|
String classPath = ResourceUtils.getURL("classpath:").getPath();
|
// System.out.println(classPath);
|
// 兼容打包为jar的class路径
|
if(classPath.contains("jar")) {
|
classPath = classPath.substring(0, classPath.lastIndexOf("."));
|
classPath = classPath.substring(0, classPath.lastIndexOf("/") + 1);
|
}
|
if (classPath.startsWith("file:")) {
|
classPath = classPath.substring(classPath.indexOf(":") + 1, classPath.length());
|
}
|
String path = classPath + "static/static/snap/";
|
// 兼容Windows系统路径（去除前面的“/”）
|
if(System.getProperty("os.name").contains("indows")) {
|
path = path.substring(1, path.length());
|
}
|
String fileName = deviceId + "_" + channelId + ".jpg";
|
ResponseEntity responseEntity = (ResponseEntity)result.getResult();
|
if (responseEntity != null && responseEntity.getStatusCode() == HttpStatus.OK) {
|
WVPResult wvpResult = (WVPResult)responseEntity.getBody();
|
if (wvpResult.getCode() == 0) {
|
StreamInfo streamInfoForSuccess = (StreamInfo)wvpResult.getData();
|
String streamUrl = streamInfoForSuccess.getFmp4();
|
// 请求截图
|
zlmresTfulUtils.getSnap(streamUrl, 15, 1, path, fileName);
|
}
|
}
|
System.out.println(path);
|
} catch (FileNotFoundException e) {
|
e.printStackTrace();
|
}
|
});
|
if (streamInfo == null) {
|
// 发送点播消息
|
cmder.playStreamCmd(device, channelId, (JSONObject response) -> {
|
logger.info("收到订阅消息： " + response.toJSONString());
|
onPublishHandlerForPlay(response, deviceId, channelId, uuid.toString());
|
if (hookEvent != null) {
|
hookEvent.response(response);
|
}
|
}, event -> {
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid);
|
Response response = event.getResponse();
|
cmder.closeRTPServer(playResult.getDevice(), channelId);
|
WVPResult wvpResult = new WVPResult();
|
wvpResult.setCode(-1);
|
wvpResult.setMsg(String.format("点播失败， 错误码： %s, %s", response.getStatusCode(), response.getReasonPhrase()));
|
msg.setData(wvpResult);
|
resultHolder.invokeResult(msg);
|
if (errorEvent != null) {
|
errorEvent.response(event);
|
}
|
});
|
} else {
|
String streamId = streamInfo.getStreamId();
|
if (streamId == null) {
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid);
|
WVPResult wvpResult = new WVPResult();
|
wvpResult.setCode(-1);
|
wvpResult.setMsg(String.format("点播失败， redis缓存streamId等于null"));
|
msg.setData(wvpResult);
|
resultHolder.invokeResult(msg);
|
return playResult;
|
}
|
JSONObject rtpInfo = zlmresTfulUtils.getRtpInfo(streamId);
|
if (rtpInfo != null && rtpInfo.getBoolean("exist")) {
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid);
|
|
WVPResult wvpResult = new WVPResult();
|
wvpResult.setCode(0);
|
wvpResult.setMsg("success");
|
wvpResult.setData(streamInfo);
|
msg.setData(wvpResult);
|
|
resultHolder.invokeResult(msg);
|
if (hookEvent != null) {
|
hookEvent.response(JSONObject.parseObject(JSON.toJSONString(streamInfo)));
|
}
|
} else {
|
redisCatchStorage.stopPlay(streamInfo);
|
storager.stopPlay(streamInfo.getDeviceID(), streamInfo.getChannelId());
|
cmder.playStreamCmd(device, channelId, (JSONObject response) -> {
|
logger.info("收到订阅消息： " + response.toJSONString());
|
onPublishHandlerForPlay(response, deviceId, channelId, uuid.toString());
|
}, event -> {
|
cmder.closeRTPServer(playResult.getDevice(), channelId);
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid);
|
Response response = event.getResponse();
|
|
WVPResult wvpResult = new WVPResult();
|
wvpResult.setCode(-1);
|
wvpResult.setMsg(String.format("点播失败， 错误码： %s, %s", response.getStatusCode(), response.getReasonPhrase()));
|
msg.setData(wvpResult);
|
resultHolder.invokeResult(msg);
|
});
|
}
|
}
|
|
return playResult;
|
}
|
|
@Override
|
public void onPublishHandlerForPlay(JSONObject resonse, String deviceId, String channelId, String uuid) {
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid);
|
StreamInfo streamInfo = onPublishHandler(resonse, deviceId, channelId, uuid);
|
if (streamInfo != null) {
|
DeviceChannel deviceChannel = storager.queryChannel(deviceId, channelId);
|
if (deviceChannel != null) {
|
deviceChannel.setStreamId(streamInfo.getStreamId());
|
storager.startPlay(deviceId, channelId, streamInfo.getStreamId());
|
}
|
ClientTransaction transaction = streamSession.getTransaction(deviceId, channelId);
|
SIPDialog dialog = (SIPDialog)transaction.getDialog();
|
StreamInfo.TransactionInfo transactionInfo = new StreamInfo.TransactionInfo();
|
transactionInfo.callId = dialog.getCallId().getCallId();
|
transactionInfo.localTag = dialog.getLocalTag();
|
transactionInfo.remoteTag = dialog.getRemoteTag();
|
transactionInfo.branch = dialog.getFirstTransactionInt().getBranchId();
|
streamInfo.setTransactionInfo(transactionInfo);
|
redisCatchStorage.startPlay(streamInfo);
|
msg.setData(JSON.toJSONString(streamInfo));
|
|
WVPResult wvpResult = new WVPResult();
|
wvpResult.setCode(0);
|
wvpResult.setMsg("sucess");
|
wvpResult.setData(streamInfo);
|
msg.setData(wvpResult);
|
|
resultHolder.invokeResult(msg);
|
} else {
|
logger.warn("设备预览API调用失败！");
|
msg.setData("设备预览API调用失败！");
|
resultHolder.invokeResult(msg);
|
}
|
}
|
|
@Override
|
public void onPublishHandlerForPlayBack(JSONObject resonse, String deviceId, String channelId, String uuid) {
|
RequestMessage msg = new RequestMessage();
|
msg.setId(DeferredResultHolder.CALLBACK_CMD_PlAY + uuid);
|
StreamInfo streamInfo = onPublishHandler(resonse, deviceId, channelId, uuid);
|
if (streamInfo != null) {
|
redisCatchStorage.startPlayback(streamInfo);
|
msg.setData(JSON.toJSONString(streamInfo));
|
resultHolder.invokeResult(msg);
|
} else {
|
logger.warn("设备预览API调用失败！");
|
msg.setData("设备预览API调用失败！");
|
resultHolder.invokeResult(msg);
|
}
|
}
|
|
public StreamInfo onPublishHandler(JSONObject resonse, String deviceId, String channelId, String uuid) {
|
String streamId = resonse.getString("stream");
|
JSONArray tracks = resonse.getJSONArray("tracks");
|
StreamInfo streamInfo = mediaService.getStreamInfoByAppAndStream("rtp", streamId, tracks);
|
streamInfo.setDeviceID(deviceId);
|
streamInfo.setChannelId(channelId);
|
return streamInfo;
|
}
|
|
}
|
