From e8a0f05f99f33fa05085536541da6bc27e66a644 Mon Sep 17 00:00:00 2001 From: zhanghua <314079846@qq.com> Date: 星期五, 14 四月 2023 14:57:55 +0800 Subject: [PATCH] bug修改 --- ycl-platform/src/main/java/com/ycl/controller/platformApi/AlarmController.java | 93 ++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 80 insertions(+), 13 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 536a3b1..35b62d9 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 @@ -14,11 +14,16 @@ 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; import java.io.InputStream; -import java.io.OutputStream; import java.net.HttpURLConnection; import java.net.URL; +import java.util.Base64; +import java.util.Dictionary; +import java.util.Hashtable; /** * @author admin @@ -27,9 +32,9 @@ @RestController @RequestMapping("/API") public class AlarmController { - - private IVideoAlarmReportService videoAlarmReportService; + + private static Dictionary<String, Integer> sessionDic = new Hashtable<>(); @Autowired private VideoUtil videoUtil; @@ -63,25 +68,87 @@ @ApiOperation(value = "鑾峰彇鍥剧墖") - @GetMapping(value = "/img", produces = MediaType.IMAGE_PNG_VALUE) + @GetMapping(value = "/img", produces = {MediaType.IMAGE_JPEG_VALUE, MediaType.IMAGE_PNG_VALUE}) +// @GetMapping(value = "/img") @ResponseBody - public byte[] getImages(HttpServletResponse response, @RequestParam String fileUrl) { + 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 { - URL url = new URL(fileUrl); - HttpURLConnection conn = (HttpURLConnection) url.openConnection(); - conn.setRequestMethod("GET"); - conn.setConnectTimeout(5 * 1000); - InputStream inputStream = conn.getInputStream();// 閫氳繃杈撳叆娴佽幏鍙栧浘鐗囨暟鎹� + if (sessionDic.get(sessionId) < 10) { + sessionDic.remove(sessionId); + 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)"); - byte[] bytes = new byte[inputStream.available()]; - inputStream.read(bytes, 0, inputStream.available()); + InputStream inputStream = conn.getInputStream();// 閫氳繃杈撳叆娴佽幏鍙栧浘鐗囨暟鎹� - return bytes; + 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