上传文件删除功能完善

base-modules/service-file/src/main/java/com/ruoyi/file/controller/FilesController.java

@@ -24,9 +24,21 @@ public class FilesController {
         return R.ok(response);
     }
 
-    @GetMapping("/{filesUUID}")
+    @GetMapping("/download/{filesUUID}")
     public void download(@PathVariable String filesUUID, HttpServletResponse response) {
         //下载文件
         filesService.download(filesUUID, response);
     }
+
+
+    @DeleteMapping("/delete/{filesUUID}")
+    public R<Void> delete(@PathVariable String filesUUID) {
+        try {
+            filesService.deleteFile(filesUUID);
+            return R.ok();
+        } catch (Exception e) {
+            return R.fail(e.getMessage()); // 删除失败，返回错误信息
+        }
+    }
+
 }

base-modules/service-file/src/main/java/com/ruoyi/file/service/Files.java

@@ -2,14 +2,12 @@ package com.ruoyi.file.service;
 
 import com.baomidou.mybatisplus.annotation.FieldFill;
 import com.baomidou.mybatisplus.annotation.TableField;
-import com.baomidou.mybatisplus.annotation.TableLogic;
 import com.baomidou.mybatisplus.annotation.TableName;
 import com.fasterxml.jackson.annotation.JsonFormat;
 import lombok.Data;
 
 import java.io.Serializable;
 import java.time.LocalDateTime;
-import java.util.Date;
 
 @Data
 @TableName("files")
@@ -38,6 +36,5 @@ public class Files implements Serializable {
     @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
     private LocalDateTime updateTime;//更新时间
 
-    @TableLogic()
     private Integer isDelete;//是否删除（1：是 0：否）
 }

base-modules/service-file/src/main/java/com/ruoyi/file/service/FilesService.java

@@ -11,4 +11,8 @@ public interface FilesService extends IService<Files> {
 
     // 下载文件
     void download(String filesUUID, HttpServletResponse response);
+
+    // 删除文件
+    void deleteFile(String filesUUID);
+
 }

base-modules/service-file/src/main/java/com/ruoyi/file/service/FilesServiceImpl.java

@@ -1,5 +1,6 @@
 package com.ruoyi.file.service;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.file.mapper.FileSequenceMapper;
 import com.ruoyi.file.mapper.FilesMapper;
@@ -53,7 +54,6 @@ public class FilesServiceImpl extends ServiceImpl<FilesMapper, Files> implements
         String fileUuid = timestamp + "A" + String.format("%04d", monthIncrement) + "." + type;
 
         // 将相对路径转化为绝对路径
-//        String destPath = new File(filesUploadPath + "/" + yearMonth).getAbsolutePath();//绝对路径
         String destPath = filesUploadPath + "/" + yearMonth;
 
         // 新的文件地址，绝对路径+新的文件名称
@@ -81,6 +81,7 @@ public class FilesServiceImpl extends ServiceImpl<FilesMapper, Files> implements
             saveFile.setSize(size); // （单位：KB）
             saveFile.setUrl(url);
             saveFile.setEnable(true);
+            saveFile.setIsDelete(0);
             saveFile.setCreateTime(LocalDateTime.now());
             saveFile.setUpdateTime(null);
             // 保存操作
@@ -116,34 +117,85 @@ public class FilesServiceImpl extends ServiceImpl<FilesMapper, Files> implements
     @Override
     public void download(String filesUUID, HttpServletResponse response) {
         try {
-            //根据文件的唯一标识码获取文件
-            File uploadFile = new File(filesUploadPath + filesUUID);
 
-            //读取文件的字节流
-            FileInputStream fileInputStream = new FileInputStream(uploadFile);
-            //将文件写入输入流
-            InputStream inputStream = new BufferedInputStream(fileInputStream);
-
-            byte[] buffer = new byte[inputStream.available()];
-            inputStream.read(buffer);
-            inputStream.close();
-
-
-            //attachment表示以附件方式下载 inline表示在线打开 "Content-Disposition: inline; filename=文件名.png"
-            //filename表示文件的默认名称，因为网络传输只支持URL编码的相关支付，因此需要将文件名URL编码后进行传输,前端收到后需要反编码才能获取到真正的名称
-            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filesUUID, "UTF-8"));
+            // 忽略 favicon.ico 请求
+            if ("favicon.ico".equals(filesUUID)) {
+                // 直接返回，不做任何处理
+                return;
+            }
+
+            // 确保 filesUUID 是文件名，而不是完整的路径
+            // 如果 filesUUID 包含路径分隔符，需要从最后的路径分隔符开始截取文件名
+            String fileName = filesUUID.substring(filesUUID.lastIndexOf('/') + 1);
+
+            // 确保路径拼接时使用正确的分隔符
+            LocalDateTime now = LocalDateTime.now();
+            String yearMonth = now.format(DateTimeFormatter.ofPattern("yyyyMM"));
+            File uploadFile = new File(filesUploadPath + "/" + yearMonth, fileName);
+
+            // 检查文件是否存在
+            if (!uploadFile.exists()) {
+                throw new FileNotFoundException("File not found: " + uploadFile.getAbsolutePath());
+            }
+
+            // 设置响应头
+            String encodedFileName = URLEncoder.encode(fileName, "UTF-8").replaceAll("\\+", "%20");
+            response.addHeader("Content-Disposition", "attachment; filename*=UTF-8''" + encodedFileName);
             response.setContentType("application/octet-stream");
 
-            //设置输出流的格式
-            ServletOutputStream os = response.getOutputStream();
-            os.write(buffer);
+            // 使用 try-with-resources 确保流正确关闭
+            try (InputStream inputStream = new BufferedInputStream(new FileInputStream(uploadFile));
+                 ServletOutputStream os = response.getOutputStream()) {
 
-            //关闭
-            fileInputStream.close();
-            os.flush();
-            os.close();
+                byte[] buffer = new byte[1024];
+                int length;
+                while ((length = inputStream.read(buffer)) != -1) {
+                    os.write(buffer, 0, length);
+                }
+            }
         } catch (IOException e) {
             e.printStackTrace();
         }
     }
+
+    @Override
+    public void deleteFile(String filesUUID) {
+        try {
+
+            // 根据文件名查询文件信息
+            Files file = getOne(new QueryWrapper<Files>().eq("name", filesUUID));
+
+            if (file == null) {
+                throw new RuntimeException("文件不存在，无法删除");
+            }
+
+            // 删除本地文件
+            String filePath = file.getPath() + "/" + file.getName();
+            File localFile = new File(filePath);
+            if (localFile.exists()) {
+                if (!localFile.delete()) {
+                    throw new RuntimeException("文件删除失败");
+                }
+            }
+
+            // 更新数据库记录
+            Files updateFile = new Files();
+            updateFile.setId(file.getId());
+            updateFile.setFilesName(file.getFilesName());
+            updateFile.setName(file.getName());
+            updateFile.setPath(file.getPath());
+            updateFile.setType(file.getType());
+            updateFile.setSize(file.getSize());
+            updateFile.setUrl(file.getUrl());
+            updateFile.setEnable(false);
+            updateFile.setIsDelete(1); // 设置为已删除
+            updateFile.setCreateTime(file.getCreateTime()); // 保留原始的 createTime
+            updateFile.setUpdateTime(LocalDateTime.now());
+
+            // 提交更新到数据库
+            updateById(updateFile); // 调用 updateById 方法更新数据库记录
+        } catch (Exception e) {
+            throw new RuntimeException("删除文件失败：" + e.getMessage());
+        }
+    }
 }