[JPA] TIP: Repository.findById(id).orElseThrow() 중복 해결하기

 

 

🧐 문제 상황

@Service
@Transactional
@RequiredArgsConstructor
public class PostService {

    private final PostRepository postRepository;
    private final MemberRepository memberRepository;

    public Long create(CreatePostCommand command) {
        Member member = memberRepository.findById(command.memberId())
                .orElseThrow(NotFoundMemberException::new);
        Post post = command.toPost(member);
        Post saved = postRepository.save(post);
        return saved.getId();
    }
}

위와 같이 JpaRepository의 findById 메서드를 사용할 때에는 보통 orElseThrow()와 함께 사용하게 됩니다.

 

그러나 이 부분이 반복되는 경우, 서비스의 private method 등으로 추출하는 경우를 종종 보았습니다.

다음과 같이 말이죠.

@Service
@Transactional
@RequiredArgsConstructor
public class PostService {

    private final PostRepository postRepository;
    private final MemberRepository memberRepository;

    public Long create(CreatePostCommand command) {
        Member member = findMemberById(command.memberId());
        Post post = command.toPost(member);
        Post saved = postRepository.save(post);
        return saved.getId();
    }
    
    private Member findMemberById(Long memberId) {
        return memberRepository.findById(command.memberId())
                .orElseThrow(NotFoundMemberException::new);
    }
}

 

그러나 위 코드를 여러 서비스에서 사용하게 된다면, 이 역시도 중복되어 위와 같은 코드를 작성해야 한다는 문제를 피할 수는 없습니다.

 

 

 

 

 

 

 

🧐 해결

public interface MemberRepository extends JpaRepository<Member, Long> {

    default Member getById(Long id) {
        return findById(id).orElseThrow(NotFoundMemberException::new);
    }
}

위와 같이 getById()라는 default 메서드를 정의함으로써 해결할 수 있습니다.

 

이는 다음과 같이 사용 가능합니다.

@Service
@Transactional
@RequiredArgsConstructor
public class PostService {

    private final PostRepository postRepository;
    private final MemberRepository memberRepository;

    public Long create(CreatePostCommand command) {
        Member member = memberRepository.getById(command.memberId());
        Post post = command.toPost(member);
        Post saved = postRepository.save(post);
        return saved.getId();
    }
}