guzijian / wvp-assist · Commits

Commit 2ad1001e94c75aa423da7095546b18af68c3e837

Authored by 648540858
1 parent 8679ad21

修复录像并发下载内存占用过多的问题

Inline Side-by-side
Showing 2 changed files with 198 additions and 174 deletions
src/main/java/top/panll/assist/config/WebMvcConfig.java 0 → 100644
  View file @2ad1001
  1 +package top.panll.assist.config;
  2 +
  3 +import org.springframework.beans.factory.annotation.Autowired;
  4 +import org.springframework.context.annotation.Configuration;
  5 +import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
  6 +import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
  7 +import top.panll.assist.dto.UserSettings;
  8 +
  9 +import java.io.File;
  10 +
  11 +
  12 +@Configuration
  13 +public class WebMvcConfig extends WebMvcConfigurerAdapter {
  14 +
  15 + @Autowired
  16 + private UserSettings userSettings;
  17 +
  18 + @Override
  19 + public void addResourceHandlers(ResourceHandlerRegistry registry) {
  20 + File file = new File(userSettings.getRecord());
  21 + registry.addResourceHandler("/download/**").addResourceLocations("file://" + file.getAbsolutePath() + "/");
  22 + super.addResourceHandlers(registry);
  23 + }
  24 +}
... ...
src/main/java/top/panll/assist/controller/DownloadController.java
  View file @2ad1001
1   -package top.panll.assist.controller;
2   -
3   -
4   -import io.swagger.v3.oas.annotations.Operation;
5   -import io.swagger.v3.oas.annotations.Parameter;
6   -import io.swagger.v3.oas.annotations.tags.Tag;
7   -import org.apache.catalina.connector.ClientAbortException;
8   -import org.slf4j.Logger;
9   -import org.slf4j.LoggerFactory;
10   -import org.springframework.beans.factory.annotation.Autowired;
11   -import org.springframework.stereotype.Controller;
12   -import org.springframework.web.bind.annotation.*;
13   -import top.panll.assist.dto.UserSettings;
14   -import top.panll.assist.utils.PageInfo;
15   -
16   -import javax.servlet.http.HttpServletRequest;
17   -import javax.servlet.http.HttpServletResponse;
18   -import java.io.BufferedOutputStream;
19   -import java.io.File;
20   -import java.io.IOException;
21   -import java.io.RandomAccessFile;
22   -import java.nio.charset.StandardCharsets;
23   -import java.util.List;
24   -import java.util.Map;
25   -
26   -@Controller
27   -@RequestMapping("/download")
28   -public class DownloadController {
29   -
30   - private final static Logger logger = LoggerFactory.getLogger(DownloadController.class);
31   -
32   - @Autowired
33   - private UserSettings userSettings;
34   -
35   - /**
36   - * 获取app+stream列表
37   - *
38   - * @return
39   - */
40   - @GetMapping(value = "/**")
41   - @ResponseBody
42   - public void download(HttpServletRequest request, HttpServletResponse response) {
43   -
44   - String resourcePath = request.getServletPath();
45   - System.out.println(resourcePath);
46   - resourcePath = resourcePath.substring("/download".length() + 1, resourcePath.length());
47   - String record = userSettings.getRecord();
48   -// if (record.endsWith("/")) {
49   -// record = record.substring(0, record.length() - 1);
50   -// System.out.println(record);
  1 +//package top.panll.assist.controller;
  2 +//
  3 +//
  4 +//import io.swagger.v3.oas.annotations.Operation;
  5 +//import io.swagger.v3.oas.annotations.Parameter;
  6 +//import io.swagger.v3.oas.annotations.tags.Tag;
  7 +//import org.apache.catalina.connector.ClientAbortException;
  8 +//import org.slf4j.Logger;
  9 +//import org.slf4j.LoggerFactory;
  10 +//import org.springframework.beans.factory.annotation.Autowired;
  11 +//import org.springframework.stereotype.Controller;
  12 +//import org.springframework.web.bind.annotation.*;
  13 +//import top.panll.assist.dto.UserSettings;
  14 +//import top.panll.assist.utils.PageInfo;
  15 +//
  16 +//import javax.servlet.http.HttpServletRequest;
  17 +//import javax.servlet.http.HttpServletResponse;
  18 +//import java.io.BufferedOutputStream;
  19 +//import java.io.File;
  20 +//import java.io.IOException;
  21 +//import java.io.RandomAccessFile;
  22 +//import java.nio.charset.StandardCharsets;
  23 +//import java.util.List;
  24 +//import java.util.Map;
  25 +//
  26 +//@Controller
  27 +//@RequestMapping("/download")
  28 +//public class DownloadController {
  29 +//
  30 +// private final static Logger logger = LoggerFactory.getLogger(DownloadController.class);
  31 +//
  32 +// @Autowired
  33 +// private UserSettings userSettings;
  34 +//
  35 +// /**
  36 +// * 获取app+stream列表
  37 +// *
  38 +// * @return
  39 +// */
  40 +// @GetMapping(value = "/**")
  41 +// @ResponseBody
  42 +// public void download(HttpServletRequest request, HttpServletResponse response) {
  43 +//
  44 +// String resourcePath = request.getServletPath();
  45 +// System.out.println(resourcePath);
  46 +// resourcePath = resourcePath.substring("/download".length() + 1, resourcePath.length());
  47 +// String record = userSettings.getRecord();
  48 +//// if (record.endsWith("/")) {
  49 +//// record = record.substring(0, record.length() - 1);
  50 +//// System.out.println(record);
  51 +//// }
  52 +// System.out.println(record + resourcePath);
  53 +// File file = new File(record + resourcePath);
  54 +// if (!file.exists()) {
  55 +// response.setStatus(HttpServletResponse.SC_NOT_FOUND);
  56 +// return;
51 57 // }
52   - System.out.println(record + resourcePath);
53   - File file = new File(record + resourcePath);
54   - if (!file.exists()) {
55   - response.setStatus(HttpServletResponse.SC_NOT_FOUND);
56   - return;
57   - }
58   -
59   - /**
60   - * 参考实现来自： CSDN 进修的CODER SpringBoot Java实现Http方式分片下载断点续传+实现H5大视频渐进式播放
61   - * https://blog.csdn.net/lovequanquqn/article/details/104562945
62   - */
63   - String range = request.getHeader("Range");
64   - logger.info("current request rang:" + range);
65   - //开始下载位置
66   - long startByte = 0;
67   - //结束下载位置
68   - long endByte = file.length() - 1;
69   - logger.info("文件开始位置：{}，文件结束位置：{}，文件总长度：{}", startByte, endByte, file.length());
70   -
71   - //有range的话
72   - if (range != null && range.contains("bytes=") && range.contains("-")) {
73   - range = range.substring(range.lastIndexOf("=") + 1).trim();
74   - String[] ranges = range.split("-");
75   - try {
76   - //判断range的类型
77   - if (ranges.length == 1) {
78   - // 类型一：bytes=-2343,
79   - if (range.startsWith("-")) {
80   - endByte = Long.parseLong(ranges[0]);
81   - }
82   - //类型二：bytes=2343-
83   - else if (range.endsWith("-")) {
84   - startByte = Long.parseLong(ranges[0]);
85   - }
86   - }
87   - //类型三：bytes=22-2343
88   - else if (ranges.length == 2) {
89   - startByte = Long.parseLong(ranges[0]);
90   - endByte = Long.parseLong(ranges[1]);
91   - }
92   -
93   - } catch (NumberFormatException e) {
94   - startByte = 0;
95   - endByte = file.length() - 1;
96   - logger.error("Range Occur Error,Message:{}", e.getLocalizedMessage());
97   - }
98   -
99   -
100   - }
101   -
102   - // 要下载的长度
103   - long contentLength = endByte - startByte + 1;
104   - // 文件名
105   - String fileName = file.getName();
106   - // 文件类型
107   - String contentType = request.getServletContext().getMimeType(fileName);
108   -
109   - // 解决下载文件时文件名乱码问题
110   - byte[] fileNameBytes = fileName.getBytes(StandardCharsets.UTF_8);
111   - fileName = new String(fileNameBytes, 0, fileNameBytes.length, StandardCharsets.ISO_8859_1);
112   -
113   - response.setHeader("Content-Type", contentType);
114   - response.setHeader("Content-Length", String.valueOf(contentLength));
115   - //inline表示浏览器直接使用，attachment表示下载，fileName表示下载的文件名
116   - response.setHeader("Content-Disposition", "inline;filename=" + fileName);
117   - response.setContentType(contentType);
118   - if (range != null) {
119   - //各种响应头设置
120   - //支持断点续传，获取部分字节内容：
121   - response.setHeader("Accept-Ranges", "bytes");
122   - //http状态码要为206：表示获取部分内容
123   - response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
124   - // Content-Range，格式为：[要下载的开始位置]-[结束位置]/[文件总大小]
125   - response.setHeader("Content-Range", "bytes " + startByte + "-" + endByte + "/" + file.length());
126   - } else {
127   - response.setStatus(HttpServletResponse.SC_OK);
128   - }
129   -
130   -
131   - BufferedOutputStream outputStream = null;
132   - RandomAccessFile randomAccessFile = null;
133   - //已传送数据大小
134   - long transmitted = 0;
135   - try {
136   - randomAccessFile = new RandomAccessFile(file, "r");
137   -
138   - outputStream = new BufferedOutputStream(response.getOutputStream());
139   - byte[] buff = new byte[4096];
140   - int len = 0;
141   - randomAccessFile.seek(startByte);
142   - //warning：判断是否到了最后不足4096（buff的length）个byte这个逻辑（(transmitted + len) <= contentLength）要放前面
143   - //不然会会先读取randomAccessFile，造成后面读取位置出错;
144   - while ((transmitted + len) <= contentLength && (len = randomAccessFile.read(buff)) != -1) {
145   - outputStream.write(buff, 0, len);
146   - transmitted += len;
147   - }
148   - //处理不足buff.length部分
149   - if (transmitted < contentLength) {
150   - len = randomAccessFile.read(buff, 0, (int) (contentLength - transmitted));
151   - outputStream.write(buff, 0, len);
152   - transmitted += len;
153   - }
154   -
155   - outputStream.flush();
156   - response.flushBuffer();
157   - randomAccessFile.close();
158   - logger.info("下载完毕：" + startByte + "-" + endByte + "：" + transmitted);
159   - } catch (ClientAbortException e) {
160   - logger.warn("用户停止下载：" + startByte + "-" + endByte + "：" + transmitted);
161   - //捕获此异常表示拥护停止下载
162   - } catch (IOException e) {
163   - e.printStackTrace();
164   - logger.error("用户下载IO异常，Message：{}", e.getLocalizedMessage());
165   - } finally {
166   - try {
167   - if (randomAccessFile != null) {
168   - randomAccessFile.close();
169   - }
170   - } catch (IOException e) {
171   - e.printStackTrace();
172   - }
173   - }///end try
174   - }
175   -}
  58 +//
  59 +// /**
  60 +// * 参考实现来自： CSDN 进修的CODER SpringBoot Java实现Http方式分片下载断点续传+实现H5大视频渐进式播放
  61 +// * https://blog.csdn.net/lovequanquqn/article/details/104562945
  62 +// */
  63 +// String range = request.getHeader("Range");
  64 +// logger.info("current request rang:" + range);
  65 +// //开始下载位置
  66 +// long startByte = 0;
  67 +// //结束下载位置
  68 +// long endByte = file.length() - 1;
  69 +// logger.info("文件开始位置：{}，文件结束位置：{}，文件总长度：{}", startByte, endByte, file.length());
  70 +//
  71 +// //有range的话
  72 +// if (range != null && range.contains("bytes=") && range.contains("-")) {
  73 +// range = range.substring(range.lastIndexOf("=") + 1).trim();
  74 +// String[] ranges = range.split("-");
  75 +// try {
  76 +// //判断range的类型
  77 +// if (ranges.length == 1) {
  78 +// // 类型一：bytes=-2343,
  79 +// if (range.startsWith("-")) {
  80 +// endByte = Long.parseLong(ranges[0]);
  81 +// }
  82 +// //类型二：bytes=2343-
  83 +// else if (range.endsWith("-")) {
  84 +// startByte = Long.parseLong(ranges[0]);
  85 +// }
  86 +// }
  87 +// //类型三：bytes=22-2343
  88 +// else if (ranges.length == 2) {
  89 +// startByte = Long.parseLong(ranges[0]);
  90 +// endByte = Long.parseLong(ranges[1]);
  91 +// }
  92 +//
  93 +// } catch (NumberFormatException e) {
  94 +// startByte = 0;
  95 +// endByte = file.length() - 1;
  96 +// logger.error("Range Occur Error,Message:{}", e.getLocalizedMessage());
  97 +// }
  98 +//
  99 +//
  100 +// }
  101 +//
  102 +// // 要下载的长度
  103 +// long contentLength = endByte - startByte + 1;
  104 +// // 文件名
  105 +// String fileName = file.getName();
  106 +// // 文件类型
  107 +// String contentType = request.getServletContext().getMimeType(fileName);
  108 +//
  109 +// // 解决下载文件时文件名乱码问题
  110 +// byte[] fileNameBytes = fileName.getBytes(StandardCharsets.UTF_8);
  111 +// fileName = new String(fileNameBytes, 0, fileNameBytes.length, StandardCharsets.ISO_8859_1);
  112 +//
  113 +// response.setHeader("Content-Type", contentType);
  114 +// response.setHeader("Content-Length", String.valueOf(contentLength));
  115 +// //inline表示浏览器直接使用，attachment表示下载，fileName表示下载的文件名
  116 +// response.setHeader("Content-Disposition", "inline;filename=" + fileName);
  117 +// response.setContentType(contentType);
  118 +// if (range != null) {
  119 +// //各种响应头设置
  120 +// //支持断点续传，获取部分字节内容：
  121 +// response.setHeader("Accept-Ranges", "bytes");
  122 +// //http状态码要为206：表示获取部分内容
  123 +// response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
  124 +// // Content-Range，格式为：[要下载的开始位置]-[结束位置]/[文件总大小]
  125 +// response.setHeader("Content-Range", "bytes " + startByte + "-" + endByte + "/" + file.length());
  126 +// } else {
  127 +// response.setStatus(HttpServletResponse.SC_OK);
  128 +// }
  129 +//
  130 +//
  131 +// BufferedOutputStream outputStream = null;
  132 +// RandomAccessFile randomAccessFile = null;
  133 +// //已传送数据大小
  134 +// long transmitted = 0;
  135 +// try {
  136 +// randomAccessFile = new RandomAccessFile(file, "r");
  137 +//
  138 +// outputStream = new BufferedOutputStream(response.getOutputStream());
  139 +// byte[] buff = new byte[4096];
  140 +// int len = 0;
  141 +// randomAccessFile.seek(startByte);
  142 +// //warning：判断是否到了最后不足4096（buff的length）个byte这个逻辑（(transmitted + len) <= contentLength）要放前面
  143 +// //不然会会先读取randomAccessFile，造成后面读取位置出错;
  144 +// while ((transmitted + len) <= contentLength && (len = randomAccessFile.read(buff)) != -1) {
  145 +// outputStream.write(buff, 0, len);
  146 +// transmitted += len;
  147 +// }
  148 +// //处理不足buff.length部分
  149 +// if (transmitted < contentLength) {
  150 +// len = randomAccessFile.read(buff, 0, (int) (contentLength - transmitted));
  151 +// outputStream.write(buff, 0, len);
  152 +// transmitted += len;
  153 +// }
  154 +//
  155 +// outputStream.flush();
  156 +// response.flushBuffer();
  157 +// randomAccessFile.close();
  158 +// logger.info("下载完毕：" + startByte + "-" + endByte + "：" + transmitted);
  159 +// } catch (ClientAbortException e) {
  160 +// logger.warn("用户停止下载：" + startByte + "-" + endByte + "：" + transmitted);
  161 +// //捕获此异常表示拥护停止下载
  162 +// } catch (IOException e) {
  163 +// e.printStackTrace();
  164 +// logger.error("用户下载IO异常，Message：{}", e.getLocalizedMessage());
  165 +// } finally {
  166 +// try {
  167 +// if (randomAccessFile != null) {
  168 +// randomAccessFile.close();
  169 +// }
  170 +// } catch (IOException e) {
  171 +// e.printStackTrace();
  172 +// }
  173 +// }///end try
  174 +// }
  175 +//}
... ...