From 986dfbd1fbbd08a0cb429caf7f63f71be4e21fb0 Mon Sep 17 00:00:00 2001
From: pfwt7cxhv <1823021652@qq.com>
Date: Wed, 4 Dec 2024 23:06:01 +0800
Subject: [PATCH] FileController

---
 FileController.java | 109 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 FileController.java

diff --git a/FileController.java b/FileController.java
new file mode 100644
index 0000000..93b1739
--- /dev/null
+++ b/FileController.java
@@ -0,0 +1,109 @@
+package com.controller;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Date;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.apache.commons.io.FileUtils; 
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired; // 导入Spring的依赖注入注解
+import org.springframework.web.bind.annotation.RequestMapping; 
+import org.springframework.web.bind.annotation.RequestParam; 
+import org.springframework.web.bind.annotation.RestController; // 导入Spring的REST控制器注解
+import org.springframework.web.multipart.MultipartFile; // 导入Spring的文件上传接口
+
+import com.annotation.IgnoreAuth; // 导入自定义注解，表示忽略身份验证
+import com.baomidou.mybatisplus.mapper.EntityWrapper; // 导入MyBatis-Plus的EntityWrapper，用于构建查询条件
+import com.entity.ConfigEntity; 
+import com.entity.EIException; 
+import com.service.ConfigService; 
+import com.utils.R; 
+
+/**
+ * 上传文件映射表
+ */
+@RestController // 标记该类为REST控制器
+@RequestMapping("file") // 设置基本请求路径为"/file"
+@SuppressWarnings({"unchecked","rawtypes"}) // 忽略编译警告
+public class FileController {
+    
+    @Autowired // 自动注入ConfigService服务
+    private ConfigService configService;
+
+    /**
+     * 上传文件
+     * @param file 上传的文件
+     * @param type 文件类型标识
+     * @param request HttpServletRequest对象
+     * @return 上传结果
+     */
+    @RequestMapping("/upload") // 映射请求路径为"/upload"
+    @IgnoreAuth // 忽略身份验证
+    public R upload(@RequestParam("file") MultipartFile file, String type, HttpServletRequest request) throws Exception {
+        // 检查上传的文件是否为空
+        if (file.isEmpty()) {
+            throw new EIException("上传文件不能为空"); 
+        }
+        // 获取文件扩展名
+        String fileExt = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".") + 1);
+        String fileName = new Date().getTime() + "." + fileExt;
+        // 定义文件保存的目标路径
+        File dest = new File(request.getSession().getServletContext().getRealPath("/upload") + "/" + fileName);
+        file.transferTo(dest);
+        
+        /**
+         * 如果使用IDEA或者Eclipse重启项目后发现之前上传的文件丢失，
+         * 请将下面一行代码注释打开
+         * 并将"D:\\ssmpiv99\\src\\main\\webapp\\upload"替换为你本地项目的upload路径，
+         * 确保项目路径中不存在中文、空格等特殊字符
+         */
+        // FileUtils.copyFile(dest, new File("D:\\ssmpiv99\\src\\main\\webapp\\upload" + "/" + fileName)); /** 修改了路径后请将该行最前面的//注释去掉 **/
+        
+        // 如果type不为空且等于"1"，则更新或保存配置信息
+        if (StringUtils.isNotBlank(type) && type.equals("1")) {
+            ConfigEntity configEntity = configService.selectOne(new EntityWrapper<ConfigEntity>().eq("name", "faceFile"));
+            if (configEntity == null) {
+                configEntity = new ConfigEntity(); // 创建新的配置实体
+                configEntity.setName("faceFile"); 
+                configEntity.setValue(fileName);
+            } else {
+                configEntity.setValue(fileName); 
+            }
+            configService.insertOrUpdate(configEntity); // 保存或更新配置
+        }
+        return R.ok().put("file", fileName); // 返回上传结果，包含文件名
+    }
+    
+    /**
+     * 下载文件
+     * @param fileName 文件名
+     * @param request HttpServletRequest对象
+     * @param response HttpServletResponse对象
+     */
+    @IgnoreAuth // 忽略身份验证
+    @RequestMapping("/download") // 映射请求路径为"/download"
+    public void download(@RequestParam String fileName, HttpServletRequest request, HttpServletResponse response) {
+        try {
+            // 定义文件的完整路径
+            File file = new File(request.getSession().getServletContext().getRealPath("/upload") + "/" + fileName);
+            // 检查文件是否存在
+            if (file.exists()) {
+                response.reset(); // 重置响应
+                // 设置响应头，指示文件下载
+                response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
+                response.setHeader("Cache-Control", "no-cache");
+                response.setHeader("Access-Control-Allow-Credentials", "true");
+                response.setContentType("application/octet-stream; charset=UTF-8"); // 设置内容类型
+                // 将文件内容写入响应输出流
+                IOUtils.write(FileUtils.readFileToByteArray(file), response.getOutputStream());
+            }
+
+        } catch (IOException e) {
+            e.printStackTrace(); // 打印异常信息
+        }
+    }
+}