♻️REFACTOR: 내 팀 정보 조회는 방장뿐만 아니라 모든 팀원들이 조회할 수 있게 리팩토링(#152) (#153)

BangFer · Jul 15, 2024 · f476941 · f476941
1 parent 6f43c23
commit f476941
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/com/capstone/BnagFer/domain/myteam/service/TeamQueryService.java b/src/main/java/com/capstone/BnagFer/domain/myteam/service/TeamQueryService.java
@@ -30,7 +30,10 @@ public GetTeamResponseDto getMyTeamById(Long teamId, User user) {
 
         Team team = teamRepository.findById(teamId).orElseThrow(() -> new TeamExceptionHandler(ErrorCode.TEAM_NOT_FOUND));
 
-        if (!user.getTeam().contains(team)) {
+        boolean isMember = team.getTeamMembers().stream()
+                .anyMatch(teamMember -> teamMember.getUser().equals(user));
+
+        if (!isMember) {
             throw new TeamExceptionHandler(ErrorCode.NO_AUTHORIZATION);
         }