SpringBoot 上传文件校验：如何判断文件是否为图片

在 Spring Boot 项目中，上传文件是一个常见的需求。为了保证上传文件的安全性和有效性，我们需要对上传的文件进行校验。其中，判断上传的文件是否为图片是一个重要的校验步骤。

判断上传文件是否为图片

可以通过以下方式判断上传的文件是否为图片：

获取上传文件的文件名和扩展名。
判断文件的扩展名是否为图片格式，常见的图片格式包括'jpg'、'jpeg'、'png'、'bmp'、'gif'等。
如果文件扩展名为图片格式，则可以使用 Java 的 ImageIO 类来读取图片文件，如果能够成功读取图片，则说明该文件为图片。

示例代码

import java.io.IOException;
import javax.imageio.ImageIO;
import org.springframework.web.multipart.MultipartFile;

public class FileUtil {
    // 判断上传的文件是否为图片
    public static boolean isImage(MultipartFile file) throws IOException {
        String fileName = file.getOriginalFilename();
        String extension = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase();
        if (extension.equals("jpg") || extension.equals("jpeg") || extension.equals("png") || extension.equals("bmp") || extension.equals("gif")) {
            // 使用ImageIO类读取图片文件，如果能够成功读取图片，则说明该文件为图片
            return ImageIO.read(file.getInputStream()) != null;
        }
        return false;
    }
}

代码说明

isImage 方法接受一个 MultipartFile 对象作为参数，代表上传的文件。
获取上传文件的原始文件名，并提取文件扩展名。
判断文件扩展名是否为常见的图片格式。
如果文件扩展名为图片格式，则使用 ImageIO.read() 方法尝试读取文件内容。如果读取成功，则说明该文件为图片，返回 true；否则返回 false。

总结

通过以上方法，我们可以有效地判断上传的文件是否为图片。在实际应用中，您还可以根据具体需求扩展校验逻辑，例如：

校验文件大小
校验文件格式
校验文件内容
校验文件安全等。