java使用文件流上下传，防止内存溢出

背景
最近在做公司的文件服务，遇到有关文件上传下载的内存溢出问题。导致内存溢出的原因是，上传下载时，将文件流转成了字节数组，导致程序大量使用内存。后来改成文件流方式避免了内存的溢出。

接口实现

上传接口：

    @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;
        }
    }

你可能感兴趣的文章：

• 利用Tomcat等中间件访问/下载大视频文件，启用文件缓存
• java上传图片并生成多种图片规格