java 对上传图片后缀名做校验

原创

mob649e8163f390 2024-07-17 08:30:41 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8163f390的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java实现图片上传后缀名校验

作为一名刚入行的开发者，你可能会遇到需要对上传的图片进行后缀名校验的情况。在这篇文章中，我将教你如何使用Java实现这一功能。

流程概述

首先，让我们通过一个表格来概述整个流程：

步骤	描述
1	获取上传的文件
2	获取文件的后缀名
3	定义允许的后缀名列表
4	校验后缀名是否在允许的列表中
5	根据校验结果返回相应的信息

详细实现

步骤1：获取上传的文件

假设你使用的是Spring框架，可以通过@RequestParam注解获取上传的文件：

@PostMapping("/upload")
public String handleFileUpload(@RequestParam("file") MultipartFile file) {
    // 后续代码
}

步骤2：获取文件的后缀名

使用Java的FilenameUtils工具类获取文件的后缀名：

import org.apache.commons.io.FilenameUtils;

String extension = FilenameUtils.getExtension(file.getOriginalFilename());

步骤3：定义允许的后缀名列表

定义一个包含允许上传的图片后缀名的列表：

List<String> allowedExtensions = Arrays.asList("jpg", "jpeg", "png", "gif");

步骤4：校验后缀名是否在允许的列表中

使用Java的contains方法检查后缀名是否在允许的列表中：

if (allowedExtensions.contains(extension)) {
    // 后缀名允许，进行后续处理
} else {
    // 后缀名不允许，返回错误信息
    return "不允许的文件类型";
}

步骤5：根据校验结果返回相应的信息

根据校验结果，返回相应的信息或进行后续处理：

if (allowedExtensions.contains(extension)) {
    // 后缀名允许，进行后续处理
    // 例如保存文件等
    return "文件上传成功";
} else {
    return "不允许的文件类型";
}

序列图

使用Mermaid语法展示整个流程的序列图：

sequenceDiagram
    participant User as U
    participant Controller as C
    participant Service as S

    U->>C: Upload file
    C->>S: Get file extension
    S->>S: Check if extension is allowed
    S->>C: Return result
    C->>U: Return result

关系图

使用Mermaid语法展示文件、后缀名和允许的后缀名列表之间的关系图：

erDiagram
    FILE ||--o{ EXTENSION : has
    EXTENSION ||--o{ ALLOWED_EXTENSIONS : is_allowed }
    ALLOWED_EXTENSIONS {
        int id PK
        string extension
    }