Procházet zdrojové kódy

新增用户分类列表不能超过20个;新增全部分类的文件数量

tsurumure před 3 měsíci
rodič
revize
26566e7104

+ 1 - 1
src/main/java/com/backendsys/modules/ai/media/service/impl/MediaTtvServiceImpl.java

@@ -136,7 +136,7 @@ public class MediaTtvServiceImpl implements MediaTtvService {
                 }
                 // ------------------------------------------------------------
 
-                // [db] 更新表记录
+                // [DB] 更新表记录
                 detail.setVideo_origin_url(response.getVideo_origin_url());
                 detail.setVideo_url(video_url);
                 detail.setObject_key(object_key);

+ 1 - 1
src/main/java/com/backendsys/modules/upload/service/SysFileCategoryService.java

@@ -10,7 +10,7 @@ import java.util.Map;
 public interface SysFileCategoryService {
 
     // 获取文件分类列表
-    List<Map<String, Object>> selectUploadFileCategoryList(SysFileCategory sysFileCategory);
+    Map<String, Object> selectUploadFileCategoryList(SysFileCategory sysFileCategory);
     // 创建文件分类
     Map<String, Object> insertFileCategory(SysFileCategory sysFileCategory);
 

+ 22 - 3
src/main/java/com/backendsys/modules/upload/service/impl/SysFileCategoryServiceImpl.java

@@ -34,8 +34,17 @@ public class SysFileCategoryServiceImpl implements SysFileCategoryService {
      * 获取文件分类列表
      */
     @Override
-    public List<Map<String, Object>> selectUploadFileCategoryList(SysFileCategory sysFileCategory) {
-        return sysFileCategoryDao.selectUploadFileCategoryList(sysFileCategory);
+    public Map<String, Object> selectUploadFileCategoryList(SysFileCategory sysFileCategory) {
+
+        // [DB] 获得文件分类列表(全部)
+        List<Map<String, Object>> list = sysFileCategoryDao.selectUploadFileCategoryList(sysFileCategory);
+        // [DB] 获得文件总数量
+        Long all_file_list_total = sysFileDao.selectCount(new LambdaQueryWrapper<SysFile>().eq(SysFile::getUser_id, sysFileCategory.getUser_id()));
+
+        Map<String, Object> resp = new LinkedHashMap<>();
+        resp.put("all_file_list_total", all_file_list_total);
+        resp.put("list", list);
+        return resp;
     }
 
     /**
@@ -43,6 +52,10 @@ public class SysFileCategoryServiceImpl implements SysFileCategoryService {
      */
     @Override
     public Map<String, Object> insertFileCategory(SysFileCategory sysFileCategory) {
+
+        Long total = sysFileCategoryDao.selectCount(new LambdaQueryWrapper<SysFileCategory>().eq(SysFileCategory::getUser_id, SecurityUtil.getUserId()));
+        if (total > 20) throw new CustException("文件分类不能超过 20 个");
+
         sysFileCategoryDao.insert(sysFileCategory);
         return Map.of("id", sysFileCategory.getId());
     }
@@ -162,7 +175,7 @@ public class SysFileCategoryServiceImpl implements SysFileCategoryService {
                         insertIdSetStr.append(Convert.toStr(it.getId()));
                     }
                 });
-                message += "新增了 Id: " + insertIdSetStr + ", ";
+                message += "新增了 Id: " + insertIdSetStr + ", ";
             }
 
             // 如果有多出的ID,则抛出异常 (过滤空值)
@@ -189,6 +202,12 @@ public class SysFileCategoryServiceImpl implements SysFileCategoryService {
             if (securityUtil.hasPermission("1.1.6")) new CustException(ResultEnum.AUTH_ROLE_ERROR.getMessage());
             // [DB] 批量删除
             sysFileCategoryDao.deleteBatchIds(deleteIds);
+            
+            
+            // [DB] 查询判断数量如果大于20,则更新失败
+            Long total = sysFileCategoryDao.selectCount(new LambdaQueryWrapper<SysFileCategory>().eq(SysFileCategory::getUser_id, user_id));
+            if (total > 20) throw new CustException("文件分类不能超过 20 个");
+
 
             String deleteIdsStr = deleteIds.stream().map(String::valueOf).collect(Collectors.joining(", "));
             message += "删除了 Id: " + deleteIdsStr + ", ";

+ 1 - 1
src/main/java/com/backendsys/modules/upload/service/impl/SysFileMultipartServiceImpl.java

@@ -272,7 +272,7 @@ public class SysFileMultipartServiceImpl implements SysFileMultipartService {
          */
         setThumbUrl(sysFileEntity, UPLOAD_THUMB_SIZE.get(), UPLOAD_THUMB_SIZE.get(), StyleEnums.THUMB_BACKGROUND.getValue());
 
-        // [db] 更新分块记录
+        // [DB] 更新分块记录
         sysFileDao.updateCompleteFile(sysFileEntity);
 
         Map<String, Object> resp = new LinkedHashMap<>();

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

@@ -81,7 +81,7 @@ public class SysFileServiceImpl extends ServiceImpl<SysFileDao, SysFile> impleme
     }
 
     /**
-     * 获取文件列表
+     * 获取文件列表 (total 有问题)
      */
     @Override
     public PageEntity selectUploadFileList(SysFile sysFile) {