old mode 100644
new mode 100755
 |  |  | 
 |  |  | package com.genersoft.iot.vmp.vmanager.gb28181.playback; | 
 |  |  |  | 
 |  |  | import com.genersoft.iot.vmp.common.InviteInfo; | 
 |  |  | import com.genersoft.iot.vmp.common.InviteSessionType; | 
 |  |  | import com.genersoft.iot.vmp.common.StreamInfo; | 
 |  |  | import com.genersoft.iot.vmp.conf.UserSetting; | 
 |  |  | import com.genersoft.iot.vmp.conf.exception.ControllerException; | 
 |  |  | import com.genersoft.iot.vmp.conf.exception.ServiceException; | 
 |  |  | import com.genersoft.iot.vmp.conf.exception.SsrcTransactionNotFoundException; | 
 |  |  | import com.genersoft.iot.vmp.conf.security.JwtUtils; | 
 |  |  | import com.genersoft.iot.vmp.gb28181.bean.Device; | 
 |  |  | import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder; | 
 |  |  | import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage; | 
 |  |  | import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory; | 
 |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
 |  |  | import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander; | 
 |  |  | import com.genersoft.iot.vmp.service.IInviteStreamService; | 
 |  |  | import com.genersoft.iot.vmp.service.IPlayService; | 
 |  |  | import com.genersoft.iot.vmp.service.bean.InviteErrorCode; | 
 |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; | 
 |  |  | import com.genersoft.iot.vmp.vmanager.bean.ErrorCode; | 
 |  |  | import com.genersoft.iot.vmp.vmanager.bean.StreamContent; | 
 |  |  | import com.genersoft.iot.vmp.vmanager.bean.WVPResult; | 
 |  |  | import io.swagger.v3.oas.annotations.Operation; | 
 |  |  | import io.swagger.v3.oas.annotations.Parameter; | 
 |  |  | import io.swagger.v3.oas.annotations.security.SecurityRequirement; | 
 |  |  | import io.swagger.v3.oas.annotations.tags.Tag; | 
 |  |  | import org.slf4j.Logger; | 
 |  |  | import org.slf4j.LoggerFactory; | 
 |  |  | import org.springframework.beans.factory.annotation.Autowired; | 
 |  |  | import org.springframework.util.ObjectUtils; | 
 |  |  | import org.springframework.web.bind.annotation.CrossOrigin; | 
 |  |  | import org.springframework.web.bind.annotation.GetMapping; | 
 |  |  | import org.springframework.web.bind.annotation.PathVariable; | 
 |  |  | import org.springframework.web.bind.annotation.RequestMapping; | 
 |  |  | import org.springframework.web.bind.annotation.RestController; | 
 |  |  |  | 
 |  |  | import com.genersoft.iot.vmp.gb28181.bean.Device; | 
 |  |  | import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander; | 
 |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; | 
 |  |  | import org.springframework.web.context.request.async.DeferredResult; | 
 |  |  |  | 
 |  |  | import javax.servlet.http.HttpServletRequest; | 
 |  |  | import javax.sip.InvalidArgumentException; | 
 |  |  | import javax.sip.SipException; | 
 |  |  | import java.net.MalformedURLException; | 
 |  |  | import java.net.URL; | 
 |  |  | import java.text.ParseException; | 
 |  |  | import java.util.UUID; | 
 |  |  |  | 
 |  |  | 
 |  |  |    private SIPCommander cmder; | 
 |  |  |  | 
 |  |  |    @Autowired | 
 |  |  |    private ZLMRTPServerFactory zlmrtpServerFactory; | 
 |  |  |  | 
 |  |  |    @Autowired | 
 |  |  |    private IVideoManagerStorage storager; | 
 |  |  |  | 
 |  |  |    @Autowired | 
 |  |  |    private IRedisCatchStorage redisCatchStorage; | 
 |  |  |    private IInviteStreamService inviteStreamService; | 
 |  |  |  | 
 |  |  |    @Autowired | 
 |  |  |    private IPlayService playService; | 
 |  |  | 
 |  |  |    @Autowired | 
 |  |  |    private UserSetting userSetting; | 
 |  |  |  | 
 |  |  |    @Operation(summary = "开始视频回放") | 
 |  |  |    @Operation(summary = "开始视频回放", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
 |  |  |    @Parameter(name = "deviceId", description = "设备国标编号", required = true) | 
 |  |  |    @Parameter(name = "channelId", description = "通道国标编号", required = true) | 
 |  |  |    @Parameter(name = "startTime", description = "开始时间", required = true) | 
 |  |  |    @Parameter(name = "endTime", description = "结束时间", required = true) | 
 |  |  |    @GetMapping("/start/{deviceId}/{channelId}") | 
 |  |  |    public DeferredResult<WVPResult<StreamContent>> start(@PathVariable String deviceId, @PathVariable String channelId, | 
 |  |  |                                            String startTime, String endTime) { | 
 |  |  |    public DeferredResult<WVPResult<StreamContent>> start(HttpServletRequest request, @PathVariable String deviceId, @PathVariable String channelId, | 
 |  |  |                                             String startTime, String endTime) { | 
 |  |  |  | 
 |  |  |       if (logger.isDebugEnabled()) { | 
 |  |  |          logger.debug(String.format("设备回放 API调用,deviceId:%s ,channelId:%s", deviceId, channelId)); | 
 |  |  | 
 |  |  |       DeferredResult<WVPResult<StreamContent>> result = new DeferredResult<>(userSetting.getPlayTimeout().longValue()); | 
 |  |  |       resultHolder.put(key, uuid, result); | 
 |  |  |  | 
 |  |  |       WVPResult<StreamContent> wvpResult = new WVPResult<>(); | 
 |  |  |       RequestMessage requestMessage = new RequestMessage(); | 
 |  |  |       requestMessage.setKey(key); | 
 |  |  |       requestMessage.setId(uuid); | 
 |  |  |  | 
 |  |  |       RequestMessage msg = new RequestMessage(); | 
 |  |  |       msg.setKey(key); | 
 |  |  |       msg.setId(uuid); | 
 |  |  |       playService.playBack(deviceId, channelId, startTime, endTime, | 
 |  |  |             (code, msg, data)->{ | 
 |  |  |  | 
 |  |  |       playService.playBack(deviceId, channelId, startTime, endTime, null, | 
 |  |  |             playBackResult->{ | 
 |  |  |                wvpResult.setCode(playBackResult.getCode()); | 
 |  |  |                wvpResult.setMsg(playBackResult.getMsg()); | 
 |  |  |                if (playBackResult.getCode() == ErrorCode.SUCCESS.getCode()) { | 
 |  |  |                   StreamInfo streamInfo = (StreamInfo)playBackResult.getData(); | 
 |  |  |                   wvpResult.setData(new StreamContent(streamInfo)); | 
 |  |  |                WVPResult<StreamContent> wvpResult = new WVPResult<>(); | 
 |  |  |                if (code == InviteErrorCode.SUCCESS.getCode()) { | 
 |  |  |                   wvpResult.setCode(ErrorCode.SUCCESS.getCode()); | 
 |  |  |                   wvpResult.setMsg(ErrorCode.SUCCESS.getMsg()); | 
 |  |  |  | 
 |  |  |                   if (data != null) { | 
 |  |  |                      StreamInfo streamInfo = (StreamInfo)data; | 
 |  |  |                      if (userSetting.getUseSourceIpAsStreamIp()) { | 
 |  |  |                         streamInfo=streamInfo.clone();//深拷贝 | 
 |  |  |                         String host; | 
 |  |  |                         try { | 
 |  |  |                            URL url=new URL(request.getRequestURL().toString()); | 
 |  |  |                            host=url.getHost(); | 
 |  |  |                         } catch (MalformedURLException e) { | 
 |  |  |                            host=request.getLocalAddr(); | 
 |  |  |                         } | 
 |  |  |                         streamInfo.channgeStreamIp(host); | 
 |  |  |                      } | 
 |  |  |                      wvpResult.setData(new StreamContent(streamInfo)); | 
 |  |  |                   } | 
 |  |  |                }else { | 
 |  |  |                   wvpResult.setCode(code); | 
 |  |  |                   wvpResult.setMsg(msg); | 
 |  |  |                } | 
 |  |  |                msg.setData(wvpResult); | 
 |  |  |                resultHolder.invokeResult(msg); | 
 |  |  |                requestMessage.setData(wvpResult); | 
 |  |  |                resultHolder.invokeResult(requestMessage); | 
 |  |  |             }); | 
 |  |  |  | 
 |  |  |       return result; | 
 |  |  |    } | 
 |  |  |  | 
 |  |  |  | 
 |  |  |    @Operation(summary = "停止视频回放") | 
 |  |  |    @Operation(summary = "停止视频回放", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
 |  |  |    @Parameter(name = "deviceId", description = "设备国标编号", required = true) | 
 |  |  |    @Parameter(name = "channelId", description = "通道国标编号", required = true) | 
 |  |  |    @Parameter(name = "stream", description = "流ID", required = true) | 
 |  |  | 
 |  |  |    } | 
 |  |  |  | 
 |  |  |  | 
 |  |  |    @Operation(summary = "回放暂停") | 
 |  |  |    @Operation(summary = "回放暂停", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
 |  |  |    @Parameter(name = "streamId", description = "回放流ID", required = true) | 
 |  |  |    @GetMapping("/pause/{streamId}") | 
 |  |  |    public void playPause(@PathVariable String streamId) { | 
 |  |  | 
 |  |  |    } | 
 |  |  |  | 
 |  |  |  | 
 |  |  |    @Operation(summary = "回放恢复") | 
 |  |  |    @Operation(summary = "回放恢复", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
 |  |  |    @Parameter(name = "streamId", description = "回放流ID", required = true) | 
 |  |  |    @GetMapping("/resume/{streamId}") | 
 |  |  |    public void playResume(@PathVariable String streamId) { | 
 |  |  | 
 |  |  |    } | 
 |  |  |  | 
 |  |  |  | 
 |  |  |    @Operation(summary = "回放拖动播放") | 
 |  |  |    @Operation(summary = "回放拖动播放", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
 |  |  |    @Parameter(name = "streamId", description = "回放流ID", required = true) | 
 |  |  |    @Parameter(name = "seekTime", description = "拖动偏移量,单位s", required = true) | 
 |  |  |    @GetMapping("/seek/{streamId}/{seekTime}") | 
 |  |  |    public void playSeek(@PathVariable String streamId, @PathVariable long seekTime) { | 
 |  |  |       logger.info("playSeek: "+streamId+", "+seekTime); | 
 |  |  |       StreamInfo streamInfo = redisCatchStorage.queryPlayback(null, null, streamId, null); | 
 |  |  |       if (null == streamInfo) { | 
 |  |  |       InviteInfo inviteInfo = inviteStreamService.getInviteInfoByStream(InviteSessionType.PLAYBACK, streamId); | 
 |  |  |  | 
 |  |  |       if (null == inviteInfo || inviteInfo.getStreamInfo() == null) { | 
 |  |  |          logger.warn("streamId不存在!"); | 
 |  |  |          throw new ControllerException(ErrorCode.ERROR400.getCode(), "streamId不存在"); | 
 |  |  |       } | 
 |  |  |       Device device = storager.queryVideoDevice(streamInfo.getDeviceID()); | 
 |  |  |       Device device = storager.queryVideoDevice(inviteInfo.getDeviceId()); | 
 |  |  |       try { | 
 |  |  |          cmder.playSeekCmd(device, streamInfo, seekTime); | 
 |  |  |          cmder.playSeekCmd(device, inviteInfo.getStreamInfo(), seekTime); | 
 |  |  |       } catch (InvalidArgumentException | ParseException | SipException e) { | 
 |  |  |          throw new ControllerException(ErrorCode.ERROR100.getCode(), e.getMessage()); | 
 |  |  |       } | 
 |  |  |    } | 
 |  |  |  | 
 |  |  |    @Operation(summary = "回放倍速播放") | 
 |  |  |    @Operation(summary = "回放倍速播放", security = @SecurityRequirement(name = JwtUtils.HEADER)) | 
 |  |  |    @Parameter(name = "streamId", description = "回放流ID", required = true) | 
 |  |  |    @Parameter(name = "speed", description = "倍速0.25 0.5 1、2、4", required = true) | 
 |  |  |    @GetMapping("/speed/{streamId}/{speed}") | 
 |  |  |    public void playSpeed(@PathVariable String streamId, @PathVariable Double speed) { | 
 |  |  |       logger.info("playSpeed: "+streamId+", "+speed); | 
 |  |  |       StreamInfo streamInfo = redisCatchStorage.queryPlayback(null, null, streamId, null); | 
 |  |  |       if (null == streamInfo) { | 
 |  |  |       InviteInfo inviteInfo = inviteStreamService.getInviteInfoByStream(InviteSessionType.PLAYBACK, streamId); | 
 |  |  |  | 
 |  |  |       if (null == inviteInfo || inviteInfo.getStreamInfo() == null) { | 
 |  |  |          logger.warn("streamId不存在!"); | 
 |  |  |          throw new ControllerException(ErrorCode.ERROR400.getCode(), "streamId不存在"); | 
 |  |  |       } | 
 |  |  | 
 |  |  |          logger.warn("不支持的speed: " + speed); | 
 |  |  |          throw new ControllerException(ErrorCode.ERROR100.getCode(), "不支持的speed(0.25 0.5 1、2、4)"); | 
 |  |  |       } | 
 |  |  |       Device device = storager.queryVideoDevice(streamInfo.getDeviceID()); | 
 |  |  |       Device device = storager.queryVideoDevice(inviteInfo.getDeviceId()); | 
 |  |  |       try { | 
 |  |  |          cmder.playSpeedCmd(device, streamInfo, speed); | 
 |  |  |          cmder.playSpeedCmd(device, inviteInfo.getStreamInfo(), speed); | 
 |  |  |       } catch (InvalidArgumentException | ParseException | SipException e) { | 
 |  |  |          throw new ControllerException(ErrorCode.ERROR100.getCode(), e.getMessage()); | 
 |  |  |       } |