From faaf907706c62c81c2a91092b950b30c89159afc Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Mon, 17 Mar 2025 17:37:52 +0800
Subject: [PATCH] refactor(common): 优化文件上传逻辑
---
src/main/java/cc/mrbird/febs/common/utils/FileUtil.java | 74 +++++++++++++++++++++++++++++++++++++
1 files changed, 74 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/common/utils/FileUtil.java b/src/main/java/cc/mrbird/febs/common/utils/FileUtil.java
index 69a5d5d..15d7932 100644
--- a/src/main/java/cc/mrbird/febs/common/utils/FileUtil.java
+++ b/src/main/java/cc/mrbird/febs/common/utils/FileUtil.java
@@ -1,16 +1,23 @@
package cc.mrbird.febs.common.utils;
import cc.mrbird.febs.common.entity.FebsConstant;
+import cn.hutool.core.lang.UUID;
import com.google.common.base.Preconditions;
import lombok.extern.slf4j.Slf4j;
+import net.coobird.thumbnailator.Thumbnails;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;
+import org.springframework.web.multipart.MultipartFile;
+import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletResponse;
+import java.awt.image.BufferedImage;
import java.io.*;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Map;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
@@ -25,6 +32,73 @@
private static final int BUFFER = 1024 * 8;
/**
+ * 文件上传服务器本地路径
+ */
+ public static Map<String,Object> fileUploadEsc(MultipartFile file,String resourceUrl,String resourcePath) throws IOException {
+ // 文件保存目录路径
+ String savePath = resourcePath;
+ // 文件保存目录URL
+ String saveUrl = resourceUrl;
+ // 检查目录
+ File uploadDir = new File(savePath);
+ if (!uploadDir.isDirectory()) {
+ uploadDir.mkdir();
+ }
+ // 获得文件的后缀
+ 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())
+ // 图片大小（长宽）压缩比例 从0-1，1表示原图
+ .scale(1f)
+ // 图片质量压缩比例 从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();
+// }
+// }
+
+ String visitPath = (saveUrl + newFileName);
+ Map<String,Object> map = new HashMap<String,Object>();
+ Map<String,Object> map2 = new HashMap<String,Object>();
+ map2.put("src",visitPath);//图片url
+ map2.put("title",newFileName);//图片名称，这个会显示在输入框里
+ map.put("code",0);//0表示成功，1失败
+ map.put("msg","上传成功");//提示消息
+ map.put("data",map2);
+ return map;
+ }
+ public static boolean isImage(InputStream inputStream) {
+ BufferedImage read = null;
+ try {
+ read = ImageIO.read(inputStream);
+ } catch (IOException e) {
+ return false;
+ }
+ return read != null;
+ }
+
+ /**
* 压缩文件或目录
*
* @param fromPath 待压缩文件或路径
--
Gitblit v1.9.1