Merge pull request #74 from SynergyX-AI-Pattern/feat/#69_search_stock_by_image

Author: Soomxn

src/main/java/com/synergyx/trading/service/stockService/search/StockSearchQueryServiceImpl.java

@@ -1,16 +1,17 @@
 package com.synergyx.trading.service.stockService.search;
 
-//import com.synergyx.trading.dto.stockSearch.StockSearchFastApiWrapperDTO;
+import com.synergyx.trading.dto.stockSearch.StockSearchFastApiWrapperDTO;
 import com.synergyx.trading.apiPayload.code.status.ErrorStatus;
 import com.synergyx.trading.apiPayload.exception.GeneralException;
 import com.synergyx.trading.dto.stockSearch.StockSearchResponseDTO;
+import com.synergyx.trading.model.Stock;
 import com.synergyx.trading.repository.StockRepository;
 import lombok.RequiredArgsConstructor;
-//import org.springframework.http.MediaType;
-//import org.springframework.http.client.MultipartBodyBuilder;
+import org.springframework.http.MediaType;
+import org.springframework.http.client.MultipartBodyBuilder;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
-//import org.springframework.web.reactive.function.client.WebClient;
+import org.springframework.web.reactive.function.client.WebClient;
 
 import java.util.List;
 
@@ -19,7 +20,7 @@
 public class StockSearchQueryServiceImpl implements StockSearchQueryService {
 
     private final StockRepository stockRepository;
-//    private final WebClient fastApiWebClient;
+    private final WebClient fastApiWebClient;
 
     /**
      * 종목명을 기준으로 종목 리스트를 검색합니다.
@@ -62,35 +63,34 @@ public StockSearchResponseDTO searchStockByImage(MultipartFile image) {
             throw new GeneralException(ErrorStatus.INVALID_IMAGE_FILE_TYPE);
         }
 
-        // TODO: FastAPI 연결 후 주석 삭제
-//        try {
-//            MultipartBodyBuilder builder = new MultipartBodyBuilder();
-//            builder.part("image", image.getResource());
-//
-//            StockSearchFastApiWrapperDTO response = fastApiWebClient.post()
-//                    .uri("/image-search")
-//                    .contentType(MediaType.MULTIPART_FORM_DATA)
-//                    .bodyValue(builder.build())
-//                    .retrieve()
-//                    .bodyToMono(StockSearchFastApiWrapperDTO.class)
-//                    .block();
-//
-//            if (response == null || !response.isSuccess()) {
-//                throw new GeneralException(ErrorStatus.IMAGE_STOCK_NOT_FOUND);
-//            }
-//
-//            return response.getData();
-//
-//        } catch (Exception e) {
-//            throw new GeneralException(ErrorStatus.IMAGE_STOCK_NOT_FOUND);
-//        }
-//    }
-        // TODO: FastAPI 연결 후 삭제
-        // 목데이터
-        return StockSearchResponseDTO.builder()
-                .id(1L)
-                .name("삼성전자")
-                .imageUrl("https://picsum.photos/30/30")
-                .build();
+        try {
+            MultipartBodyBuilder builder = new MultipartBodyBuilder();
+            builder.part("image", image.getResource());
+
+            StockSearchFastApiWrapperDTO response = fastApiWebClient.post()
+                    .uri("/api/v1/stocks/search-by-image")
+                    .contentType(MediaType.MULTIPART_FORM_DATA)
+                    .bodyValue(builder.build())
+                    .retrieve()
+                    .bodyToMono(StockSearchFastApiWrapperDTO.class)
+                    .block();
+
+            if (response == null || !response.isSuccess()) {
+                throw new GeneralException(ErrorStatus.IMAGE_STOCK_NOT_FOUND);
+            }
+
+            Long stockId = response.getData().getId();
+            Stock stock = stockRepository.findById(stockId)
+                    .orElseThrow(() -> new GeneralException(ErrorStatus.STOCK_NOT_FOUND));
+
+            return StockSearchResponseDTO.builder()
+                    .id(stock.getId())
+                    .name(stock.getName())
+                    .imageUrl(stock.getImageUrl())
+                    .build();
+
+        } catch (Exception e) {
+            throw new GeneralException(ErrorStatus.IMAGE_STOCK_NOT_FOUND);
+        }
     }
 }