优化文件上传

know_sub_business/src/main/java/com/supervision/knowsub/controller/FileController.java

@@ -1,5 +1,6 @@
 package com.supervision.knowsub.controller;
 
+import com.supervision.knowsub.exception.BusinessException;
 import com.supervision.knowsub.service.FileService;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
@@ -24,6 +25,9 @@ public class FileController {
     @PostMapping("upload")
     public String fileUpload(@RequestParam(name = "file") MultipartFile file) throws IOException {
         byte[] bytes = file.getBytes();
+        if (bytes.length > 20 * 1024 * 1024){
+            throw new BusinessException("文件过大,文件大小请不要超过20MB");
+        }
         return fileService.uploadFile(file.getOriginalFilename(), bytes);
     }
 

know_sub_business/src/main/java/com/supervision/knowsub/service/impl/KnowledgeManageServiceImpl.java

@@ -2,6 +2,7 @@ package com.supervision.knowsub.service.impl;
 
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.lang.Assert;
+import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.supervision.knowsub.domain.UserInfo;
@@ -24,6 +25,7 @@ import java.time.Instant;
 import java.time.LocalDateTime;
 import java.util.List;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.Set;
 
 @Slf4j
@@ -74,8 +76,8 @@ public class KnowledgeManageServiceImpl implements KnowledgeManageService {
         var context = new KnowledgeContext();
         context.setContext(reqVO.getContent());
         knowledgeContextService.save(context);
-
-        KnowledgeInfo baseInfo = reqVO.getKnowledgeInfo();
+        // 保存基本信息
+        KnowledgeInfo baseInfo = Optional.ofNullable(reqVO.getKnowledgeInfo()).orElse(new KnowledgeInfo());
         baseInfo.setContextId(context.getId());
         knowledgeInfoService.save(baseInfo);