Merge pull request #23 from enjoy-hack/Daeun

Daeun

Author: DaeunSon

src/main/java/com/example/enjoy/controller/FavoriteCourseController.java

@@ -17,13 +17,17 @@
 public class FavoriteCourseController {
     private final FavoriteCourseService favoriteCourseService;
 
-    @Operation(summary = "즐겨찾는 과목 추가", description = "학생의 즐겨찾는 과목을 추가합니다.")
+    @Operation(summary = "즐겨찾는 과목 추가", description = "학생의 즐겨찾는 과목을 최대 4개까지 한 번에 추가합니다.")
     @PostMapping("/add")
-    public ResponseEntity<Void> addFavoriteCourse(
-            @RequestParam String courseName,
+    public ResponseEntity<Void> addFavoriteCourses(
+            @RequestParam List<String> courseNames,
             @RequestParam String studentId) {
 
-        favoriteCourseService.addFavoriteCourse(studentId, courseName);
+        if (courseNames == null || courseNames.isEmpty() || courseNames.size() > 4) {
+            return ResponseEntity.badRequest().build();
+        }
+
+        favoriteCourseService.addFavoriteCourses(studentId, courseNames);
         return ResponseEntity.ok().build();
     }
 

src/main/java/com/example/enjoy/service/FavoriteCourseService.java

@@ -21,22 +21,22 @@ public class FavoriteCourseService {
     private final UserRepository userRepository;
 
     @Transactional
-    public void addFavoriteCourse(String studentId, String courseName) {
+    public void addFavoriteCourses(String studentId, List<String> courseNames) {
         User user = userRepository.findByStudentId(studentId)
                 .orElseThrow(() -> new RuntimeException("사용자를 찾을 수 없습니다."));
 
+        for (String courseName : courseNames) {
+            TrackCourse trackCourse = trackCourseRepository.findByCourseName(courseName)
+                    .orElseThrow(() -> new RuntimeException("과목을 찾을 수 없습니다."));
 
-        TrackCourse trackCourse = trackCourseRepository.findByCourseName(courseName)
-                .orElseThrow(() -> new RuntimeException("과목을 찾을 수 없습니다."));
+            boolean alreadyExists = favoriteCourseRepository
+                    .findByUserAndCourseName(user, courseName)
+                    .isPresent();
 
-        // 이미 즐겨찾기한 과목인지 확인
-        boolean alreadyExists = favoriteCourseRepository
-                .findByUserAndCourseName(user, courseName)
-                .isPresent();
-
-        if (!alreadyExists) {
-            FavoriteCourse favoriteCourse = new FavoriteCourse(user, trackCourse.getCourseName());
-            favoriteCourseRepository.save(favoriteCourse);
+            if (!alreadyExists) {
+                FavoriteCourse favoriteCourse = new FavoriteCourse(user, trackCourse.getCourseName());
+                favoriteCourseRepository.save(favoriteCourse);
+            }
         }
     }
 

src/main/java/com/example/enjoy/service/TrackService.java

@@ -217,4 +217,4 @@ private double calculateFavoriteScore(TrackProgressDto track, Set<String> favori
     public List<Track> getAllTracks() {
         return trackRepository.findAll();
     }
-}
+}