玩转文件上传：轻松实现Spring MVC文件上传

2023-05-20 01:17:12

让文件上传轻而易举：使用 Spring MVC 简化文件处理

1. 轻松上手：单文件上传

准备上传文件？Spring MVC 让它变得轻而易举！只需在控制器中添加一个映射方法，使用 @PostMapping 注解指定处理文件上传的请求。别忘了使用 @RequestParam("file") MultipartFile file 接收前端传递过来的文件。

在 HTML 表单中，使用 <input type="file"> 让用户选择要上传的文件。

@PostMapping("/upload")
public String upload(@RequestParam("file") MultipartFile file) {
    // 文件保存逻辑
    return "redirect:/success";
}

<form action="/upload" method="post" enctype="multipart/form-data">
    <input type="file" name="file">
    <input type="submit" value="上传">
</form>

2. 应对挑战：多文件上传

需要一次上传多个文件？<input type="file" multiple> 了解一下！使用它，用户可以轻松选择多个文件进行上传。

<form action="/upload" method="post" enctype="multipart/form-data">
    <input type="file" name="files" multiple>
    <input type="submit" value="上传">
</form>

在控制器中，我们使用 MultipartFile[] 数组来接收多个文件。

@PostMapping("/upload")
public String upload(@RequestParam("files") MultipartFile[] files) {
    // 文件保存逻辑
    return "redirect:/success";
}

3. 保障安全：限制文件大小和类型

安全至上！为了防止恶意文件或过大文件上传，我们可以限制文件大小和类型。

@PostMapping("/upload")
public String upload(@RequestParam("file") MultipartFile file) {
    if (file.getSize() > 1024 * 1024 * 5) {
        return "redirect:/error?message=文件太大";
    }

    if (!Arrays.asList("image/jpeg", "image/png", "image/gif").contains(file.getContentType())) {
        return "redirect:/error?message=文件类型错误";
    }

    // 文件保存逻辑
    return "redirect:/success";
}

4. 完善功能：文件保存和下载

上传的文件需要一个新家！我们可以将它们存储在本地磁盘或云存储中。

@PostMapping("/upload")
public String upload(@RequestParam("file") MultipartFile file) {
    // 文件保存逻辑
    Path path = Paths.get("/path/to/file", file.getOriginalFilename());
    file.transferTo(path);
    return "redirect:/success";
}

下载文件也很简单，使用 ResponseEntity 即可。

@GetMapping("/download/{filename}")
public ResponseEntity<byte[]> download(@PathVariable String filename) {
    Path path = Paths.get("/path/to/file", filename);
    byte[] data = Files.readAllBytes(path);
    HttpHeaders headers = new HttpHeaders();
    headers.set("Content-Disposition", "attachment; filename=" + filename);
    headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
    return new ResponseEntity<>(data, headers, HttpStatus.OK);
}