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 |  102 ++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 77 insertions(+), 25 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 d848cee..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,7 +39,10 @@
 @RestController
 @RequestMapping("/API")
 public class AlarmController {
+    private IVideoPointService videoPointService;
     private IVideoAlarmReportService videoAlarmReportService;
+
+    private static Dictionary<String, Integer> sessionDic = new Hashtable<>();
 
     @Autowired
     private VideoUtil videoUtil;
@@ -39,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());
     }
@@ -52,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));
     }
 
 
@@ -66,30 +106,42 @@
     @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) {
+    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, String sessionId) {
+        int index = sessionDic.get(sessionId);
         try {
+            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();
+                conn.setRequestMethod("GET");
+                conn.setConnectTimeout(10 * 1000);
+                conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
 
-            System.out.println("getImages鑾峰彇鍥剧墖鍦板潃锛�" + fileUrl + "&OSSAccessKeyId=" + OSSAccessKeyId + "&Signature=" + Signature);
-            URL url = new URL(fileUrl + "&OSSAccessKeyId=" + OSSAccessKeyId + "&Signature=" + Signature);
-            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
-            conn.setRequestMethod("GET");
-            conn.setConnectTimeout(10 * 1000);
-            conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
+                InputStream inputStream = conn.getInputStream();// 閫氳繃杈撳叆娴佽幏鍙栧浘鐗囨暟鎹�
 
-            InputStream inputStream = conn.getInputStream();// 閫氳繃杈撳叆娴佽幏鍙栧浘鐗囨暟鎹�
+                byte[] data = readInputStream(inputStream);
+                System.out.println("getImages鑾峰彇鍥剧墖鎴愬姛");
 
-
-            byte[] data = readInputStream(inputStream);
-            System.out.println("getImages鑾峰彇鍥剧墖鎴愬姛");
-
-
-            System.out.println("getImages -鍥剧墖64");
-            System.out.println(Base64.getEncoder().encodeToString(data));
-            return data;
-
+                return data;
+            }
         } catch (Exception e) {
-            System.out.println("getImages鑾峰彇鍥剧墖澶辫触锛�" + e.getMessage());
-            // 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