后端添加显示默认头像或图片

Signed-off-by: zup <jiu3295282258@163.com>

src/main/java/com/yeqifu/sys/controller/FileController.java

@@ -2,9 +2,12 @@ package com.yeqifu.sys.controller;
 
 import cn.hutool.core.date.DateUtil;
 import com.yeqifu.sys.common.AppFileUtils;
+import com.yeqifu.sys.common.Constast;
+import org.springframework.http.HttpHeaders;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.multipart.MultipartFile;
 import org.springframework.http.HttpStatus;
@@ -28,8 +31,8 @@ public class FileController {
 
     /**
      * 图片上传
-     * @param mf
-     * @return
+     * @param mf 上传的文件
+     * @return 上传结果
      */
     @RequestMapping("uploadImage")
     public Map<String, Object> uploadImage(MultipartFile mf) {
@@ -57,7 +60,7 @@ public class FileController {
             // 得到当前日期的字符串
             String dirName = DateUtil.format(new Date(), "yyyy-MM-dd");
 
-            // 构造文件夹
+            // 构造文件夹路径
             File dirFile = new File(AppFileUtils.UPLOAD_PATH, dirName);
 
             // 判断当前文件夹是否存在
@@ -87,14 +90,18 @@ public class FileController {
         return map;
     }
 
-
     /**
      * 图片下载
      * @param path 图片路径
-     * return
+     * @return 文件内容或错误信息
      */
     @RequestMapping("showImageByPath")
-    public ResponseEntity<byte[]> downloadImage(String path) {
+    public ResponseEntity<byte[]> downloadImage(@RequestParam String path) {
+        // 如果是默认图片，直接返回
+        if (path.equals(Constast.DEFAULT_IMG_USER) || path.equals(Constast.DEFAULT_IMG_GOODS)) {
+            return getDefaultImage(path);
+        }
+
         // 规范路径，防止路径穿越
         Path normalizedPath = Paths.get(AppFileUtils.UPLOAD_PATH, path).normalize();
 
@@ -102,7 +109,6 @@ public class FileController {
         if (!normalizedPath.startsWith(AppFileUtils.UPLOAD_PATH)) {
             return ResponseEntity.status(HttpStatus.FORBIDDEN).build(); // 如果路径越界，返回403 Forbidden
         }
-
         // 检查文件是否存在
         File file = normalizedPath.toFile();
         if (!file.exists() || !file.isFile()) {
@@ -131,4 +137,21 @@ public class FileController {
         }
     }
 
+    /**
+     * 获取默认图片内容
+     */
+    private ResponseEntity<byte[]> getDefaultImage(String path) {
+        // 返回一个 302 重定向响应，重定向到默认图片的 URL
+        if (path.equals(Constast.DEFAULT_IMG_USER)) {
+            return ResponseEntity.status(HttpStatus.FOUND)
+                    .header(HttpHeaders.LOCATION, Constast.DEFAULT_IMG_USER) // 图片的 URL 路径
+                    .build();
+        } else if (path.equals(Constast.DEFAULT_IMG_GOODS)) {
+            return ResponseEntity.status(HttpStatus.FOUND)
+                    .header(HttpHeaders.LOCATION, Constast.DEFAULT_IMG_GOODS) // 图片的 URL 路径
+                    .build();
+        }
+        return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
+    }
 }
+