Merge pull request #121 from UMC-ON/feat/companypost

fix: 동행글 추천 조회 시 엔티티 직접 참조하도록 수정
UMC-ON · Dec 16, 2024 · ef32ab4 · ef32ab4
2 parents ad7ecb6 + f454a40
commit ef32ab4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/com/on/server/domain/companyPost/application/CompanyPostService.java b/src/main/java/com/on/server/domain/companyPost/application/CompanyPostService.java
@@ -131,10 +131,11 @@ public List<CompanyPostResponseDTO> getRecentCompanyPosts() {
     public List<CompanyPostResponseDTO> getNearbyCompanyPostsByLikeTravelArea(Long companyPostId, User user) {
 
         // 여러 개의 travel area 중에서 첫 번째만 선택
-        CompanyPostResponseDTO post = getCompanyPostById(companyPostId).get(0);
+        CompanyPost companyPost = companyPostRepository.findById(companyPostId)
+                .orElseThrow(() -> new BadRequestException(ResponseCode.ROW_DOES_NOT_EXIST, "게시글을 찾을 수 없습니다. ID: " + companyPostId));
 
         // 국가만 일치해도 조회되도록 국가 부분만 추출
-        String firstCountry = post.getTravelArea().get(0).split(" ")[0];
+        String firstCountry = companyPost.getTravelArea().get(0).split(" ")[0];
 
         // 국가와 일치하는 다른 게시글을 조회, 현재 사용자가 작성한 게시글은 제외
         List<CompanyPost> nearbyPosts = companyPostRepository.findTop5NearbyCompanyPosts(firstCountry, companyPostId, user, PageRequest.of(0, 5));