From 190f56a4cdcff03a5635d5b0ee616af14db4b146 Mon Sep 17 00:00:00 2001
From: Chris Park <dev.chris.kr@gmail.com>
Date: Sat, 23 Dec 2023 17:43:46 +0900
Subject: [PATCH] feat: Add extra information on OpenSearch result

---
 .../domain/search/SearchServiceImpl.java      | 33 ++++++++++++++++---
 .../domain/search/dto/SearchResultDto.java    |  9 +++++
 2 files changed, 38 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/depromeet/breadmapbackend/domain/search/SearchServiceImpl.java b/src/main/java/com/depromeet/breadmapbackend/domain/search/SearchServiceImpl.java
index c30e4c1d..4311f279 100644
--- a/src/main/java/com/depromeet/breadmapbackend/domain/search/SearchServiceImpl.java
+++ b/src/main/java/com/depromeet/breadmapbackend/domain/search/SearchServiceImpl.java
@@ -1,6 +1,10 @@
 package com.depromeet.breadmapbackend.domain.search;
 
+import com.depromeet.breadmapbackend.domain.bakery.Bakery;
 import com.depromeet.breadmapbackend.domain.bakery.BakeryRepository;
+import com.depromeet.breadmapbackend.domain.flag.FlagBakery;
+import com.depromeet.breadmapbackend.domain.flag.FlagBakeryRepository;
+import com.depromeet.breadmapbackend.domain.flag.FlagRepository;
 import com.depromeet.breadmapbackend.domain.review.Review;
 import com.depromeet.breadmapbackend.domain.review.ReviewQueryRepository;
 import com.depromeet.breadmapbackend.domain.review.ReviewService;
@@ -36,6 +40,8 @@ public class SearchServiceImpl implements SearchService {
     private final UserRepository userRepository;
     private final SubwayStationRepository subwayStationRepository;
     private final ReviewQueryRepository reviewQueryRepository;
+    private final FlagBakeryRepository flagBakeryRepository;
+    private final FlagRepository flagRepository;
 
     private final ReviewService reviewService;
     private final SearchLogService searchLogService;
@@ -60,7 +66,7 @@ public List<SearchDto> searchDatabase(String oAuthId, String keyword, Double lat
 
     @Override
     public SearchResultResponse searchEngine(String oAuthId, String keyword, Double userLat, Double userLng, SearchType searchType) {
-        userRepository.findByOAuthId(oAuthId).orElseThrow(() -> new DaedongException(DaedongStatus.USER_NOT_FOUND));
+        User user = userRepository.findByOAuthId(oAuthId).orElseThrow(() -> new DaedongException(DaedongStatus.USER_NOT_FOUND));
         searchLogService.saveRecentSearchLog(oAuthId, keyword);
 
         SearchResultResponse.SearchResultResponseBuilder builder = SearchResultResponse.builder();
@@ -84,8 +90,10 @@ public SearchResultResponse searchEngine(String oAuthId, String keyword, Double
             searchResults.addAll(getSearchEngineDtoList(document.getHits(), userLat, userLng));
         }
 
-        List<BakeryReviewScoreDto> bakeriesReviews = reviewQueryRepository.getBakeriesReview(SearchEngineDto.extractBakeryIdList(searchResults));
-        List<SearchResultDto> searchResultDtos = mergeSearchEngineAndReview(searchResults, bakeriesReviews);
+        List<Long> bakeryIds = SearchEngineDto.extractBakeryIdList(searchResults);
+        List<BakeryReviewScoreDto> bakeriesReviews = reviewQueryRepository.getBakeriesReview(bakeryIds);
+
+        List<SearchResultDto> searchResultDtos = this.mergeSearchEngineAndAdditionalInfo(searchResults, user.getId(), bakeriesReviews);
 
         resultSortBySearchType(searchType, searchResultDtos);
 
@@ -94,12 +102,28 @@ public SearchResultResponse searchEngine(String oAuthId, String keyword, Double
                 .build();
     }
 
-    private static List<SearchResultDto> mergeSearchEngineAndReview(List<SearchEngineDto> searchResults, List<BakeryReviewScoreDto> bakeriesReviews) {
+    private List<SearchResultDto> mergeSearchEngineAndAdditionalInfo(List<SearchEngineDto> searchResults, Long userId, List<BakeryReviewScoreDto> bakeriesReviews) {
 
         List<SearchResultDto> searchResultDtos = new ArrayList<>();
 
         for (SearchEngineDto searchResultDto : searchResults) {
             SearchResultDto.SearchResultDtoBuilder searchResultDtoBuilder = SearchResultDto.builder();
+
+            Optional<Bakery> bakeryOptional = bakeryRepository.findById(searchResultDto.getBakeryId());
+            if(bakeryOptional.isPresent()) {
+                searchResultDtoBuilder
+                        .latitude(bakeryOptional.get().getLatitude())
+                        .longitude(bakeryOptional.get().getLongitude())
+                        .bakeryImageUrl(bakeryOptional.get().getImages());
+
+                searchResultDtoBuilder.flagCount(flagBakeryRepository.countFlagNum(bakeryOptional.get()));
+
+                Optional<FlagBakery> flagBakeryOptional = flagBakeryRepository.findByBakeryAndUserId(bakeryOptional.get(), userId);
+                if(flagBakeryOptional.isPresent()) {
+                    searchResultDtoBuilder.flagColor(flagBakeryOptional.get().getFlag().getColor().getCode());
+                }
+            }
+
             searchResultDtoBuilder
                     .bakeryId(searchResultDto.getBakeryId())
                     .bakeryName(searchResultDto.getBakeryName())
@@ -107,6 +131,7 @@ private static List<SearchResultDto> mergeSearchEngineAndReview(List<SearchEngin
                     .distance(searchResultDto.getDistance())
                     .reviewNum(0L) // init
                     .totalScore(0d); // init
+
             if (searchResultDto.getBreadId() != null) {
                 searchResultDtoBuilder
                         .breadId(searchResultDto.getBreadId())
diff --git a/src/main/java/com/depromeet/breadmapbackend/domain/search/dto/SearchResultDto.java b/src/main/java/com/depromeet/breadmapbackend/domain/search/dto/SearchResultDto.java
index 81967c0a..ba95959a 100644
--- a/src/main/java/com/depromeet/breadmapbackend/domain/search/dto/SearchResultDto.java
+++ b/src/main/java/com/depromeet/breadmapbackend/domain/search/dto/SearchResultDto.java
@@ -2,6 +2,8 @@
 
 import lombok.*;
 
+import java.util.List;
+
 @Getter
 @Builder
 @AllArgsConstructor
@@ -15,4 +17,11 @@ public class SearchResultDto {
     private Double distance;
     private Long reviewNum;
     private Double totalScore;
+
+    private Double latitude;
+    private Double longitude;
+    private List<String> bakeryImageUrl;
+
+    private Integer flagCount;
+    private String flagColor;
 }