feat: 课程新增或编辑时可以上传多个附件

lichong-a · Sep 21, 2024 · 9d9f219 · 9d9f219
1 parent 2d4b383
commit 9d9f219
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.../main/java/org/funcode/portal/server/module/ielts/course/domain/vo/CourseAddOrEditVo.java b/.../main/java/org/funcode/portal/server/module/ielts/course/domain/vo/CourseAddOrEditVo.java
@@ -14,6 +14,7 @@
 import org.springframework.stereotype.Component;
 
 import java.math.BigDecimal;
+import java.util.List;
 
 /**
  * @author 李冲
@@ -46,5 +47,7 @@ public class CourseAddOrEditVo {
     private Long courseMediaStorageId;
     @Schema(description = "课程简介文件ID")
     private Long courseDescriptionStorageId;
+    @Schema(description = "课程附件ID列表")
+    private List<Long> courseAttachmentStorageIds;
 
 }
diff --git a/...in/java/org/funcode/portal/server/module/ielts/course/service/impl/CourseServiceImpl.java b/...in/java/org/funcode/portal/server/module/ielts/course/service/impl/CourseServiceImpl.java
@@ -21,6 +21,10 @@
 import org.springframework.data.jpa.domain.Specification;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.CollectionUtils;
+
+import java.util.HashSet;
+import java.util.List;
 
 /**
  * @author 李冲
@@ -84,6 +88,7 @@ private Course transAddOrEditVoToCourse(CourseAddOrEditVo courseAddOrEditVo) {
         Storage mediaStorage = storageRepository.getReferenceById(mediaStorageId);
         Storage coverStorage = storageRepository.getReferenceById(courseAddOrEditVo.getCourseCoverStorageId());
         Storage descriptionStorage = storageRepository.getReferenceById(courseAddOrEditVo.getCourseDescriptionStorageId());
+        List<Storage> courseAttachmentStorages = storageRepository.findAllById(courseAddOrEditVo.getCourseAttachmentStorageIds());
         return Course.builder()
                 .id(courseAddOrEditVo.getId())
                 .title(courseAddOrEditVo.getTitle())
@@ -92,6 +97,7 @@ private Course transAddOrEditVoToCourse(CourseAddOrEditVo courseAddOrEditVo) {
                 .courseMediaStorage(mediaStorage)
                 .courseCoverStorage(coverStorage)
                 .courseDescriptionStorage(descriptionStorage)
+                .courseAttachmentStorages(CollectionUtils.isEmpty(courseAttachmentStorages) ? new HashSet<>() : new HashSet<>(courseAttachmentStorages))
                 .build();
     }
 }