From b8d8733ad9eeeb170a71897d1078acdbea7680f2 Mon Sep 17 00:00:00 2001 From: zhanghua <314079846@qq.com> Date: 星期二, 04 三月 2025 15:23:38 +0800 Subject: [PATCH] 优化 --- ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java | 253 +++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 247 insertions(+), 6 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 00f1924..d5a1603 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,17 +1,44 @@ package com.ycl.controller.platformApi; +import cn.hutool.json.JSONConverter; +import com.alibaba.fastjson.JSONObject; +import com.aliyun.oss.ClientBuilderConfiguration; +import com.aliyun.oss.OSS; +import com.aliyun.oss.OSSClientBuilder; +import com.aliyun.oss.OSSException; +import com.aliyun.oss.common.auth.CredentialsProviderFactory; +import com.aliyun.oss.common.auth.EnvironmentVariableCredentialsProvider; +import com.aliyun.oss.common.comm.SignVersion; +import com.aliyuncs.exceptions.ClientException; +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.dto.video.HKAlarmParam; +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 com.ycl.utils.ConstantPropertiesUtils; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; +import lombok.SneakyThrows; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.http.MediaType; import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.PostMapping; -import org.springframework.web.bind.annotation.RequestBody; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import javax.xml.bind.DatatypeConverter; +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.net.HttpURLConnection; +import java.net.URL; +import java.util.*; /** * @author admin @@ -20,19 +47,233 @@ @RestController @RequestMapping("/API") public class AlarmController { - + private IVideoPointService videoPointService; private IVideoAlarmReportService videoAlarmReportService; + + private static Dictionary<String, Integer> sessionDic = new Hashtable<>(); + + @Autowired + private VideoUtil videoUtil; @Autowired public void setVideoAlarmReportService(IVideoAlarmReportService videoAlarmReportService) { 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()); } -} + @ApiOperation("娴峰悍瑙嗛鎶ヨ鎺ㄩ�丄PI") + @PostMapping("/HK/alarmReport") + @LogSave(operationType = "鎶ヨ绠$悊", contain = "娴峰悍瑙嗛鎶ヨ鎺ㄩ��") + public CommonResult hkAlarmReport(@RequestBody @Validated HKAlarmParam alarmParam) { +// System.out.println("娴峰悍瑙嗛鎶ヨ鎺ㄩ�丄PI锛�" + JSONObject.toJSONString(alarmParam)); + videoAlarmReportService.saveFromHK(alarmParam); + return new CommonResult(ResultCode.SUCCESS.getCode(), ResultCode.SUCCESS.getMessage()); + } + + @GetMapping("/Video/{code}/{subType}") + @ApiOperation(value = "璁惧鑾峰彇鐩戞帶鍦板潃") + @SneakyThrows + public CommonResult getMonitorAddress(@PathVariable String code, @PathVariable Integer subType, + @RequestParam(required = false) String scheme) { + + 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) { +// videoUtil.getKHPresets(code); + 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)); + } + } + + + @ApiOperation(value = "鑾峰彇鍥剧墖") + @GetMapping(value = "/img", produces = {MediaType.IMAGE_JPEG_VALUE, MediaType.IMAGE_PNG_VALUE}) +// @GetMapping(value = "/img") + @ResponseBody + public byte[] getImages(HttpServletRequest request, HttpServletResponse response, + @RequestParam String fileUrl, @RequestParam(required = false) String OSSAccessKeyId, + @RequestParam(required = false) String Signature) { + if (StringUtils.isNotBlank(fileUrl)) { + String sessionId = request.getSession().getId(); + sessionDic.put(sessionId, 0); + return getBytes(fileUrl, OSSAccessKeyId, Signature, sessionId); + } else { + return null; + } + } + + @ApiOperation(value = "鑾峰彇鍥剧墖") + @GetMapping(value = "/img/base64") +// @GetMapping(value = "/img") + @ResponseBody + public String getImagesBase64(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); + byte[] bytes = getBytes(fileUrl, OSSAccessKeyId, Signature, sessionId); + String base64 = DatatypeConverter.printBase64Binary(bytes); + return base64; + } + + @GetMapping(value = "/index") + @ResponseBody + public String getImages(HttpServletRequest request, HttpServletResponse response) { + return "閬傛槍缁煎悎鎵ф硶骞冲彴"; + + } + + private URL translateOSSUrl(String fileUrl) { + OSS ossClient = null; + try { + String endpoint = ConstantPropertiesUtils.END_POINT; + String accessKeyId = ConstantPropertiesUtils.ACCESS_KEY_ID; + String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET; + String bucketName = ConstantPropertiesUtils.BUCKET_NAME; + + String[] urlArray = fileUrl.split("Expires="); + +// System.out.println("-------urlArray[0]:" + urlArray[0]); + +// System.out.println("-------PREFIX_URL:" + ConstantPropertiesUtils.PREFIX_URL); + String objectName = urlArray[0].replace(ConstantPropertiesUtils.PREFIX_URL, ""); + + objectName = objectName.replace("?", ""); +// System.out.println("-------objectName:" + objectName); + ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret); + + // 璁剧疆绛惧悕URL杩囨湡鏃堕棿锛屽崟浣嶄负姣銆傛湰绀轰緥浠ヨ缃繃鏈熸椂闂翠负1灏忔椂涓轰緥銆� + Date expiration = new Date(new Date().getTime() + 3600 * 1000L); + // 鐢熸垚浠ET鏂规硶璁块棶鐨勭鍚峌RL銆傛湰绀轰緥娌℃湁棰濆璇锋眰澶达紝鍏朵粬浜哄彲浠ョ洿鎺ラ�氳繃娴忚鍣ㄨ闂浉鍏冲唴瀹广�� + URL url = ossClient.generatePresignedUrl(bucketName, objectName, expiration); + + return url; + } catch (OSSException oe) { + System.out.println("Caught an OSSException, which means your request made it to OSS, " + + "but was rejected with an error response for some reason."); + System.out.println("Error Message:" + oe.getErrorMessage()); + System.out.println("Error Code:" + oe.getErrorCode()); + System.out.println("Request ID:" + oe.getRequestId()); + System.out.println("Host ID:" + oe.getHostId()); + } finally { + if (ossClient != null) { + ossClient.shutdown(); + } + } + return null; + } + + 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); + URL url = translateOSSUrl(fileUrl); +// fileUrl = fileUrl.replace("Expires", "v"); +//// System.out.println("getImages鑾峰彇鍥剧墖鍦板潃锛�" + fileUrl + "&OSSAccessKeyId=" + OSSAccessKeyId + "&Signature=" + Signature); +// URL url = null; +// if (StringUtils.isNotBlank(OSSAccessKeyId) && StringUtils.isNotBlank(Signature)) { +// url = new URL(fileUrl + "&OSSAccessKeyId=" + OSSAccessKeyId + "&Signature=" + Signature); +// } else { +// url = new URL(fileUrl); +// } + + 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();// 閫氳繃杈撳叆娴佽幏鍙栧浘鐗囨暟鎹� + + byte[] data = readInputStream(inputStream); +// System.out.println("getImages鑾峰彇鍥剧墖鎴愬姛"); + + return data; + } + } catch (Exception e) { + index++; + sessionDic.put(sessionId, index); + System.out.println(index + "-getImages鑾峰彇鍥剧墖澶辫触锛�" + e.getMessage()); + e.printStackTrace(); + getBytes(fileUrl, OSSAccessKeyId, Signature, sessionId); + + } + return null; + } + + private static String inputStream2Base64(InputStream is) throws Exception { + byte[] data = null; + try { + ByteArrayOutputStream swapStream = new ByteArrayOutputStream(); + byte[] buff = new byte[100]; + int rc = 0; + while ((rc = is.read(buff, 0, 100)) > 0) { + swapStream.write(buff, 0, rc); + } + data = swapStream.toByteArray(); + } catch (IOException e) { + e.printStackTrace(); + } finally { + if (is != null) { + try { + is.close(); + } catch (IOException e) { + throw new Exception("杈撳叆娴佸叧闂紓甯�"); + } + } + } + + return Base64.getEncoder().encodeToString(data); + } + + private byte[] readInputStream(InputStream inStream) throws Exception { + ByteArrayOutputStream outStream = new ByteArrayOutputStream(); + //鍒涘缓涓�涓狟uffer瀛楃涓� + byte[] buffer = new byte[1024]; + //姣忔璇诲彇鐨勫瓧绗︿覆闀垮害锛屽鏋滀负-1锛屼唬琛ㄥ叏閮ㄨ鍙栧畬姣� + int len = 0; + //浣跨敤涓�涓緭鍏ユ祦浠巄uffer閲屾妸鏁版嵁璇诲彇鍑烘潵 + while ((len = inStream.read(buffer)) != -1) { + //鐢ㄨ緭鍑烘祦寰�buffer閲屽啓鍏ユ暟鎹紝涓棿鍙傛暟浠h〃浠庡摢涓綅缃紑濮嬭锛宭en浠h〃璇诲彇鐨勯暱搴� + outStream.write(buffer, 0, len); + } + //鍏抽棴杈撳叆娴� + inStream.close(); + //鎶妎utStream閲岀殑鏁版嵁鍐欏叆鍐呭瓨 + return outStream.toByteArray(); + } +} \ No newline at end of file -- Gitblit v1.8.0