648540858
2023-06-20 21a96ad20fd75e55d03c00af8df8adb039f0c77a
src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/play/PlayController.java
@@ -24,7 +24,9 @@
import com.genersoft.iot.vmp.service.bean.InviteErrorCode;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
import com.genersoft.iot.vmp.utils.DateUtil;
import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
import com.genersoft.iot.vmp.vmanager.bean.SnapPath;
import com.genersoft.iot.vmp.vmanager.bean.StreamContent;
import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
import io.swagger.v3.oas.annotations.Operation;
@@ -39,6 +41,7 @@
import javax.servlet.http.HttpServletRequest;
import javax.sip.InvalidArgumentException;
import javax.sip.SipException;
import java.io.File;
import java.text.ParseException;
import java.util.List;
import java.util.UUID;
@@ -337,5 +340,40 @@
      return jsonObject;
   }
   @Operation(summary = "获取截图")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "channelId", description = "通道国标编号", required = true)
   @GetMapping("/snap")
   public DeferredResult<String> getSnap(HttpServletRequest request, String deviceId, String channelId) {
      if (logger.isDebugEnabled()) {
         logger.debug("获取截图: {}/{}", deviceId, channelId);
      }
      DeferredResult<String> result = new DeferredResult<>(3 * 1000L);
      String key  = DeferredResultHolder.CALLBACK_CMD_SNAP + deviceId;
      String uuid  = UUID.randomUUID().toString();
      resultHolder.put(key, uuid,  result);
      RequestMessage message = new RequestMessage();
      message.setKey(key);
      message.setId(uuid);
      String nowForUrl = DateUtil.getNowForUrl();
      String fileName = deviceId + "_" + channelId + "_" + nowForUrl + ".jpg";
      playService.getSnap(deviceId, channelId, fileName, (code, msg, data) -> {
         if (code == InviteErrorCode.SUCCESS.getCode()) {
            File snapFile = new File((String)data);
            String fileNameForUrl = deviceId + "/" + channelId + "?mark=" + nowForUrl;
            String uri = request.getRequestURL().toString().replace(request.getRequestURI(), "/api/device/query/snap/" + fileNameForUrl);
            SnapPath snapPath = SnapPath.getInstance((String) data, snapFile.getAbsolutePath(), uri);
            message.setData(snapPath);
         }else {
            message.setData(WVPResult.fail(code, msg));
         }
         resultHolder.invokeResult(message);
      });
      return result;
   }
}