fileUploadAndDownload

Spring完成文件的存储和下载

文件的存储

• 使用数据库存储文件本体
• 使用数据库存储文件的位置，文件存储在文件系统上

使用数据库直接存储文件，使用LONGBLOB类型保存文件的Byte数据

文件的接收

前端使用form上传数据

ContentType : multipart/form-data 使用 @RequestPart 注解接收文件

    @PostMapping("/upload")
    public ResponseEntity<Object> uploadFile(@RequestPart MultipartFile file) throws IOException {
        String fileName = file.getOriginalFilename();
        String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);
        fileName = fileName.substring(0, fileName.lastIndexOf("."));
        fileMapper.uploadFile(fileName, suffix, file.getBytes());
        return new ResponseEntity(HttpStatusCode.valueOf(200));
    }

文件的下载

• 文件存储于数据库

使用Jdbc获取文件的LONGBLOB，在Java中使用byte[]数组接收数据

    @GetMapping("/download/{filename}")
    public void downloadFile(@PathVariable() String filename, HttpServletResponse response) throws IOException {
        filename = filename.substring(0, filename.lastIndexOf(".") == -1 ? filename.length() : filename.lastIndexOf("."));
        File file = jdbcTemplate.queryForObject("SELECT * FROM FileStorage WHERE file_name = ?", new Object[]{filename},
                (resultSet, rowNum) -> {
                    File tfile = new File();
                    tfile.setFileId(resultSet.getInt("file_id"));
                    tfile.setFileName(resultSet.getString("file_name"));
                    tfile.setFileType(resultSet.getString("file_type"));
                    tfile.setFileData(resultSet.getBytes("file_data"));
                    return tfile;
                }
        );

        byte[] fileData = file.getFileData();

        response.setContentType("application/octet-stream");
        response.setHeader("Content-Disposition", "attachment; filename=\"file"+file.getFileType()+"\"");

        if(fileData != null){
            ServletOutputStream stream = response.getOutputStream();
            stream.write(fileData);
            stream.flush();
        }
    }