فهرست منبع

修复编辑文件

tsurumure 3 ماه پیش
والد
کامیت
9ed188463e
1فایلهای تغییر یافته به همراه14 افزوده شده و 6 حذف شده
  1. 14 6
      src/main/java/com/backendsys/modules/upload/service/impl/SysFileServiceImpl.java

+ 14 - 6
src/main/java/com/backendsys/modules/upload/service/impl/SysFileServiceImpl.java

@@ -363,17 +363,25 @@ public class SysFileServiceImpl extends ServiceImpl<SysFileDao, SysFile> impleme
     @Override
     public Map<String, Object> updateUploadFile(SysFile sysFile) {
 
+        // 查询文件是否存在
+        LambdaQueryWrapper<SysFile> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(SysFile::getUser_id, sysFile.getUser_id());
+        wrapper.eq(SysFile::getId, sysFile.getCategory_id());
+        Boolean isFileExist = sysFileDao.exists(wrapper);
+        if (!isFileExist) throw new CustException("文件不存在");
+
         // 查询文件分类是否存在
         if (sysFile.getCategory_id() != null) {
-            LambdaQueryWrapper<SysFileCategory> wrapper = new LambdaQueryWrapper<>();
-            wrapper.eq(SysFileCategory::getUser_id, sysFile.getUser_id());
-            wrapper.eq(SysFileCategory::getId, sysFile.getCategory_id());
-            if (!sysFileCategoryDao.exists(wrapper)) {
-                throw new CustException("文件分类不存在");
-            }
+            LambdaQueryWrapper<SysFileCategory> wrapperCategory = new LambdaQueryWrapper<>();
+            wrapperCategory.eq(SysFileCategory::getUser_id, sysFile.getUser_id());
+            wrapperCategory.eq(SysFileCategory::getId, sysFile.getCategory_id());
+            Boolean isFileCategoryExist = sysFileCategoryDao.exists(wrapperCategory);
+            if (!isFileCategoryExist) throw new CustException("文件分类不存在");
         }
 
+        // [DB] 更新文件
         sysFileDao.updateFile(sysFile);
+
         return Map.of("id", sysFile.getId());
     }