diff --git a/src/main/java/xyz/playedu/api/controller/frontend/CourseController.java b/src/main/java/xyz/playedu/api/controller/frontend/CourseController.java index 67fe56b..bfe5582 100644 --- a/src/main/java/xyz/playedu/api/controller/frontend/CourseController.java +++ b/src/main/java/xyz/playedu/api/controller/frontend/CourseController.java @@ -32,6 +32,8 @@ import xyz.playedu.api.util.IpUtil; import java.util.Date; import java.util.HashMap; import java.util.List; +import java.util.Map; +import java.util.function.Function; import java.util.stream.Collectors; /** @@ -83,6 +85,20 @@ public class CourseController { List attachments = attachmentService.getAttachmentsByCourseId(course.getId()); + if (null != attachments && attachments.size() > 0) { + Map resourceMap = + resourceService + .chunks(attachments.stream().map(CourseAttachment::getRid).toList()) + .stream() + .collect(Collectors.toMap(Resource::getId, Function.identity())); + attachments.forEach( + courseAttachment -> { + Resource resource = resourceMap.get(courseAttachment.getRid()); + if(null != resource){ + courseAttachment.setExt(resource.getExtension()); + } + }); + } HashMap data = new HashMap<>(); data.put("course", course);