fix: member info get 정보 추가 (#111)

Co-authored-by: yuuddin <[email protected]>
umc-owncast · Aug 30, 2024 · e9c1ff3 · e9c1ff3
1 parent 47792d8
commit e9c1ff3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/main/java/com/umc/owncast/domain/member/dto/MemberInfoDTO.java b/src/main/java/com/umc/owncast/domain/member/dto/MemberInfoDTO.java
@@ -16,4 +16,6 @@ public class MemberInfoDTO {
     String username;
     String nickname;
     Language language;
+    String mainCategory;
+    String subCategory;
 }
diff --git a/src/main/java/com/umc/owncast/domain/member/service/MemberService.java b/src/main/java/com/umc/owncast/domain/member/service/MemberService.java
@@ -70,11 +70,17 @@ public Boolean memberIdDuplicate(String loginId) {
     }
 
     public MemberInfoDTO getMemberInfo(Member member){
+        Optional<SubPrefer> subPrefer = subPreferRepository.findByMember(member);
+        if(subPrefer.isEmpty()){
+            throw new UserHandler(ErrorCode.SUBCATEGORY_NOT_EXIST);
+        }
         return MemberInfoDTO.builder()
                 .loginId(member.getLoginId())
                 .language(member.getLanguage())
                 .nickname(member.getNickname())
                 .username(member.getUsername())
+                .mainCategory(subPrefer.get().getSubCategory().getMainCategory().getKrSubCategory())
+                .subCategory(subPrefer.get().getSubCategory().getName())
                 .build();
     }