[hotfix] 클래스 정보 조회 오류 수정

ITA-OneByte · Jan 14, 2025 · 5376de5 · 5376de5
1 parent f793733
commit 5376de5
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/src/main/java/classfit/example/classfit/category/controller/ClassInfoController.java b/src/main/java/classfit/example/classfit/category/controller/ClassInfoController.java
@@ -1,8 +1,10 @@
 package classfit.example.classfit.category.controller;
 
+import classfit.example.classfit.auth.annotation.AuthMember;
 import classfit.example.classfit.category.dto.response.ClassInfoResponse;
 import classfit.example.classfit.category.service.ClassInfoService;
 import classfit.example.classfit.common.ApiResponse;
+import classfit.example.classfit.member.domain.Member;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.RequiredArgsConstructor;
@@ -21,8 +23,8 @@ public class ClassInfoController {
 
     @GetMapping("/class-info")
     @Operation(summary = "클래스 정보 조회", description = "클래스 정보 조회 api 입니다.")
-    public ApiResponse<List<ClassInfoResponse>> getClassInfo() {
-        List<ClassInfoResponse> categories = classInfoService.getClasses();
+    public ApiResponse<List<ClassInfoResponse>> getClassInfo(@AuthMember Member member) {
+        List<ClassInfoResponse> categories = classInfoService.getClasses(member);
         return ApiResponse.success(categories, 200, "SUCCESS");
     }
 }
diff --git a/src/main/java/classfit/example/classfit/category/repository/MainClassRepository.java b/src/main/java/classfit/example/classfit/category/repository/MainClassRepository.java
@@ -2,15 +2,14 @@
 
 import classfit.example.classfit.academy.domain.Academy;
 import classfit.example.classfit.category.domain.MainClass;
-import classfit.example.classfit.member.domain.Member;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
 
 @Repository
 public interface MainClassRepository extends JpaRepository<MainClass, Long> {
-    List<MainClass> findAllByOrderByMainClassNameAsc();
+    List<MainClass> findAllByAcademyOrderByMainClassNameAsc(Academy academy);
 
     List<MainClass> findByAcademy(Academy academy);
 

diff --git a/src/main/java/classfit/example/classfit/category/service/ClassInfoService.java b/src/main/java/classfit/example/classfit/category/service/ClassInfoService.java
@@ -1,10 +1,12 @@
 package classfit.example.classfit.category.service;
 
+import classfit.example.classfit.academy.domain.Academy;
 import classfit.example.classfit.category.domain.MainClass;
 import classfit.example.classfit.category.domain.SubClass;
 import classfit.example.classfit.category.dto.response.ClassInfoResponse;
 import classfit.example.classfit.category.dto.response.SubClassResponse;
 import classfit.example.classfit.category.repository.MainClassRepository;
+import classfit.example.classfit.member.domain.Member;
 import jakarta.transaction.Transactional;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -18,8 +20,10 @@ public class ClassInfoService {
     private final MainClassRepository mainClassRepository;
 
     @Transactional
-    public List<ClassInfoResponse> getClasses() {
-        List<MainClass> mainClasses = mainClassRepository.findAllByOrderByMainClassNameAsc();
+    public List<ClassInfoResponse> getClasses(Member member) {
+        Academy academy = member.getAcademy();
+
+        List<MainClass> mainClasses = mainClassRepository.findAllByAcademyOrderByMainClassNameAsc(academy);
         return mainClasses.stream()
             .map(this::mapToClassInfoResponse)
             .collect(Collectors.toList());