refactor(common): 优化文件上传逻辑

KKSU

2025-03-17 faaf907706c62c81c2a91092b950b30c89159afc

refactor(common): 优化文件上传逻辑

- 添加文件格式限制，仅允许上传图片
- 移除非图片文件上传的相关代码- 优化代码结构，提高可读性和维护性

 src/main/java/cc/mrbird/febs/common/utils/FileUtil.java

@@ -7,8 +7,6 @@
import net.coobird.thumbnailator.Thumbnails;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.util.FileCopyUtils;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
@@ -49,6 +47,14 @@
        // 获得文件的后缀
        String fileName = file.getOriginalFilename();
        String fileExt = fileName.substring(fileName.lastIndexOf(".") + 1);
        //限制文件格式为图片
        // 限制文件格式为图片
        if (!isImage(file.getInputStream())) {
            Map<String,Object> map = new HashMap<String,Object>();
            map.put("code",1); // 1表示失败
            map.put("msg","文件格式不支持"); // 提示消息
            return map;
        }
        String newFileName = UUID.randomUUID().toString() + "." + fileExt;
        if (isImage(file.getInputStream())) {
            Thumbnails.of(file.getInputStream())
@@ -57,19 +63,20 @@
                    // 图片质量压缩比例 从0-1，越接近1质量越好
                    .outputQuality(0.9f)
                    .toOutputStream(new FileOutputStream(savePath + newFileName));
        } else {
            File uploadedFile = new File(savePath, newFileName);
            uploadedFile.setReadable(true, false);
            uploadedFile.setExecutable(true, false);
            uploadedFile.setWritable(true, false);
            try {
                //存文件
                FileCopyUtils.copy(file.getBytes(), uploadedFile);
                file.transferTo(uploadedFile);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
//        else {
//            File uploadedFile = new File(savePath, newFileName);
//            uploadedFile.setReadable(true, false);
//            uploadedFile.setExecutable(true, false);
//            uploadedFile.setWritable(true, false);
//            try {
//                //存文件
//                FileCopyUtils.copy(file.getBytes(), uploadedFile);
//                file.transferTo(uploadedFile);
//            } catch (IOException e) {
//                e.printStackTrace();
//            }
//        }

        String visitPath = (saveUrl + newFileName);
        Map<String,Object> map = new HashMap<String,Object>();