背景
最近在做公司的文件服务,遇到有关文件上传下载的内存溢出问题。导致内存溢出的原因是,上传下载时,将文件流转成了字节数组,导致程序大量使用内存。后来改成文件流方式避免了内存的溢出。
接口实现
上传接口:
@RequestMapping("uploadFile")
public Boolean uploadFile(@RequestParam("file") MultipartFile file) {
String rootPath = System.getProperty("user.dir");
String filePath = rootPath + "/files/test.mp4";
try {
File newFile = new File(filePath);
if (!newFile.exists()) {
newFile.getParentFile().mkdirs();
newFile.createNewFile();
}
// 会覆盖原文件
file.transferTo(newFile);
return true;
} catch (IOException e) {
e.printStackTrace();
return false;
}
}
下载接口:
@RequestMapping("downloadFile")
public void downloadFile(HttpServletRequest request, HttpServletResponse response) {
String rootPath = System.getProperty("user.dir");
String filePath = rootPath + "/files/test.mp4";
File file = new File(filePath);
InputStream is = null;
if (file.exists()) {
is = new FileInputStream(file);
}
if (is != null) {
String fileName = filePath.substring(filePath.lastIndexOf('/') + 1);
// 设置下载文件名
String userAgent = request.getHeader("User-Agent");
userAgent = userAgent == null ? "" : userAgent.toLowerCase();
// 针对IE或者以IE为内核的浏览器:
if (userAgent.contains("msie") || userAgent.contains("trident")) {
fileName = URLEncoder.encode(fileName, "UTF-8");
fileName = fileName.replace("+", " ");
} else {
fileName = new String(fileName.getBytes(), "iso-8859-1");
}
response.reset();
response.setContentType("application/octet-stream; charset=UTF-8");
response.setHeader("Content-Disposition", "attachment;filename=\"" + fileName + "\"");
long length = is.transferTo(response.getOutputStream());
response.addHeader("Content-Length", "" + length);
is.close();
} else {
response.setContentType("text/plain;charset=UTF-8");
response.getWriter().write("未能找到资源文件");
response.setStatus(404);
return;
}
}
你可能感兴趣的文章: