From 7ef4892f9f24f941aca37e6b3991b808a0aca619 Mon Sep 17 00:00:00 2001 From: zhanghua <314079846@qq.com> Date: 星期五, 08 九月 2023 11:16:35 +0800 Subject: [PATCH] 优化 --- ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java | 79 +++++++++++++++++++++++++++++++-------- 1 files changed, 62 insertions(+), 17 deletions(-) diff --git a/ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java b/ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java index 276ce3a..fda52c9 100644 --- a/ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java +++ b/ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java @@ -1,10 +1,16 @@ package com.ycl.controller.platformApi; +import cn.hutool.json.JSONConverter; +import com.alibaba.fastjson.JSONObject; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.StringUtils; +import com.ycl.annotation.LogSave; import com.ycl.api.CommonResult; import com.ycl.dto.video.AlarmParam; +import com.ycl.entity.video.VideoPoint; import com.ycl.enums.common.ResultCode; import com.ycl.service.video.IVideoAlarmReportService; +import com.ycl.service.video.impl.IVideoPointService; import com.ycl.util.VideoUtil; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; @@ -14,6 +20,7 @@ import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; +import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.ByteArrayOutputStream; import java.io.IOException; @@ -21,6 +28,9 @@ import java.net.HttpURLConnection; import java.net.URL; import java.util.Base64; +import java.util.Dictionary; +import java.util.Hashtable; +import java.util.List; /** * @author admin @@ -29,9 +39,10 @@ @RestController @RequestMapping("/API") public class AlarmController { + private IVideoPointService videoPointService; private IVideoAlarmReportService videoAlarmReportService; - private static Integer index = 0; + private static Dictionary<String, Integer> sessionDic = new Hashtable<>(); @Autowired private VideoUtil videoUtil; @@ -41,9 +52,16 @@ this.videoAlarmReportService = videoAlarmReportService; } + @Autowired + public void setVideoPointService(IVideoPointService videoPointService) { + this.videoPointService = videoPointService; + } + @ApiOperation("瑙嗛鎶ヨ鎺ㄩ�丄PI") @PostMapping("/AlarmReport") + @LogSave(operationType = "鎶ヨ绠$悊", contain = "瑙嗛鎶ヨ鎺ㄩ��") public CommonResult alarmReport(@RequestBody @Validated AlarmParam alarmParam) { +// System.out.println("淇濆瓨鎶ヨ鏁版嵁锛�" + JSONObject.toJSONString(alarmParam)); videoAlarmReportService.save(alarmParam); return new CommonResult(ResultCode.SUCCESS.getCode(), ResultCode.SUCCESS.getMessage()); } @@ -54,13 +72,33 @@ public CommonResult getMonitorAddress(@PathVariable String code, @PathVariable Integer subType, @RequestParam(required = false) String scheme) { - if (subType == null) { - subType = 1; + LambdaQueryWrapper<VideoPoint> queryWrapper = new LambdaQueryWrapper<VideoPoint>() + .eq(VideoPoint::getPlatResourceId, code) + .or().eq(VideoPoint::getCode, code); + List<VideoPoint> pointList = videoPointService.list(queryWrapper); + if (pointList.size() > 0) { + VideoPoint videoPoint = pointList.get(0); + if (videoPoint.getType() == 1) { + return CommonResult.successApi(videoUtil.getKHVideo(code)); + } else { + if (subType == null) { + subType = 1; + } + if (!StringUtils.isNotBlank(scheme)) { + scheme = "RTSP"; + } + return CommonResult.successApi(videoUtil.getVideo(code, scheme, subType)); + } + } else { + + if (subType == null) { + subType = 1; + } + if (!StringUtils.isNotBlank(scheme)) { + scheme = "RTSP"; + } + return CommonResult.successApi(videoUtil.getVideo(code, scheme, subType)); } - if (!StringUtils.isNotBlank(scheme)) { - scheme = "RTSP"; - } - return CommonResult.successApi(videoUtil.getVideo(code, scheme, subType)); } @@ -68,17 +106,21 @@ @GetMapping(value = "/img", produces = {MediaType.IMAGE_JPEG_VALUE, MediaType.IMAGE_PNG_VALUE}) // @GetMapping(value = "/img") @ResponseBody - public byte[] getImages(HttpServletResponse response, @RequestParam String fileUrl, @RequestParam String OSSAccessKeyId, @RequestParam String Signature) { - - index = 0; - return getBytes(fileUrl, OSSAccessKeyId, Signature); + public byte[] getImages(HttpServletRequest request, HttpServletResponse response, + @RequestParam String fileUrl, @RequestParam(required = false) String OSSAccessKeyId, + @RequestParam(required = false) String Signature) { + String sessionId = request.getSession().getId(); + sessionDic.put(sessionId, 0); + return getBytes(fileUrl, OSSAccessKeyId, Signature, sessionId); } - private byte[] getBytes(String fileUrl, String OSSAccessKeyId, String Signature) { + private byte[] getBytes(String fileUrl, String OSSAccessKeyId, String Signature, String sessionId) { + int index = sessionDic.get(sessionId); try { - if (index < 10) { - index++; + if (sessionDic.get(sessionId) < 10) { + sessionDic.remove(sessionId); + fileUrl = fileUrl.replace("Expires", "v"); System.out.println("getImages鑾峰彇鍥剧墖鍦板潃锛�" + fileUrl + "&OSSAccessKeyId=" + OSSAccessKeyId + "&Signature=" + Signature); URL url = new URL(fileUrl + "&OSSAccessKeyId=" + OSSAccessKeyId + "&Signature=" + Signature); HttpURLConnection conn = (HttpURLConnection) url.openConnection(); @@ -94,9 +136,12 @@ return data; } } catch (Exception e) { - System.out.println(index + "getImages鑾峰彇鍥剧墖澶辫触锛�" + e.getMessage()); - getBytes(fileUrl, OSSAccessKeyId, Signature); - // e.printStackTrace(); + index++; + sessionDic.put(sessionId, index); + System.out.println(index + "-getImages鑾峰彇鍥剧墖澶辫触锛�" + e.getMessage()); + e.printStackTrace(); + getBytes(fileUrl, OSSAccessKeyId, Signature, sessionId); + } return null; } -- Gitblit v1.8.0