From bdd1b1bbb9ec5abef61e517030e50748c4747389 Mon Sep 17 00:00:00 2001
From: wsw <wsw@winning.com.cn>
Date: Sun, 30 Jul 2023 15:36:24 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AD=A6=E5=91=98=E7=AB=AF--=E8=AF=BE=E7=A8=8B?=
 =?UTF-8?q?=E9=99=84=E4=BB=B6=E5=88=97=E8=A1=A8=E6=9F=A5=E8=AF=A2--?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0ext=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/frontend/CourseController.java    | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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<CourseAttachment> attachments =
                 attachmentService.getAttachmentsByCourseId(course.getId());
+        if (null != attachments && attachments.size() > 0) {
+            Map<Integer, Resource> 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<String, Object> data = new HashMap<>();
         data.put("course", course);