【SpringMVC学习笔记_4】文件下载上传

---

前言

利用ResponseEntity实现下载文件
利用MultipartFile对象封装上传的文件实现上传文件

一、文件下载

1. 利用HttpSession获取Servlet对象
2. 通过Servlet对象拿到文件所在的全路径
3. 利用InputStream读取文件内容
4. 将读取的文件内容保存至字节数组中
5. 创建响应头、响应状态码
6. 将字节数组、响应头、状态码存入ResponseEntity中并返回

@RequestMapping("/testDown")
    public ResponseEntity<byte[]> testResponseEntity(HttpSession session) throws IOException {
        //获取ServletContext对象
        ServletContext servletContext = session.getServletContext();
        //获取服务器中文件的真实路径
        String realPath = servletContext.getRealPath("/static/img/1.jpg");
        //创建输入流
        InputStream is = new FileInputStream(realPath);
        //创建字节数组
        byte[] bytes = new byte[is.available()];
        //将流读到字节数组中
        is.read(bytes);
        //创建HttpHeaders对象设置响应头信息
        MultiValueMap<String, String> headers = new HttpHeaders();
        //设置要下载方式以及下载文件的名字
        headers.add("Content-Disposition", "attachment;filename=1.jpg");
        //设置响应状态码
        HttpStatus statusCode = HttpStatus.OK;
        //创建ResponseEntity对象
        ResponseEntity<byte[]> responseEntity = new ResponseEntity<>(bytes, headers, statusCode);
        //关闭输入流
        is.close();
        return responseEntity;
    }

二、文件上传

1. 利用MultipartFile对象封装上传的文件对象
2. 读取文件信息，获取文件后缀
3. 利用UUID重新赋予文件一个文件名
4. 利用HttpSession获取Servlet对象，并利用Servlet对象获取文件上传位置的路径
5. 判断当前是否已经创建文件夹，没有则创建
6. 将文件上传位置的路径+UUID+文件分隔符号+后缀名作为文件全路径创建文件并上传

    @RequestMapping("/testUp")
    public String testUp(MultipartFile photo, HttpSession session) throws IOException {
        //获取文件名称
        String originalFileName = photo.getOriginalFilename();
        //获取文件后缀名称
        String suffix = originalFileName.substring(originalFileName.lastIndexOf("."));
        //对文件名称作处理，使其唯一
        String uuid = UUID.randomUUID().toString();
        //获取上传路径
        ServletContext servletContext = session.getServletContext();
        String realPath = servletContext.getRealPath("photo");
        File file = new File(realPath);
        if(!file.exists()){
            file.mkdir();
        }
        String finalFilePath = realPath + uuid + File.separator + suffix;
        //上传
        photo.transferTo(new File(finalFilePath));
        return "success";
    }