[FEAT] 투자자 성향별 종목 추천 서비스(#78)

PDA4-Phoenix · Oct 19, 2023 · 10aa2f0 · 10aa2f0
1 parent d38458a
commit 10aa2f0
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 2 deletions.
diff --git a/src/main/java/com/bull4jo/kkanbustock/stock/controller/StockController.java b/src/main/java/com/bull4jo/kkanbustock/stock/controller/StockController.java
@@ -1,5 +1,6 @@
 package com.bull4jo.kkanbustock.stock.controller;
 
+import com.bull4jo.kkanbustock.member.domain.entity.InvestorType;
 import com.bull4jo.kkanbustock.stock.domain.RecommendStockResponse;
 import com.bull4jo.kkanbustock.stock.domain.Stock;
 import com.bull4jo.kkanbustock.stock.service.StockService;
@@ -28,11 +29,22 @@ public ResponseEntity<Stock> searchStockByName(@RequestParam(value = "itmsNm") f
     }
 
     @GetMapping("/v1/recommends")
-    public ResponseEntity<Page<RecommendStockResponse>> getRecommendedStocks(@RequestParam(value = "page", defaultValue = "0") int page,
-                                     @RequestParam(value = "size", defaultValue = "15") int size) {
+    public ResponseEntity<Page<RecommendStockResponse>> getRecommendedStocks(
+            @RequestParam(value = "page", defaultValue = "0") int page,
+            @RequestParam(value = "size", defaultValue = "15") int size) {
         Pageable pageable = PageRequest.of(page, size);
         return ResponseEntity.ok(stockService.getRecommendedStocks(pageable));
     }
+
+    @GetMapping("/v1/recommends/{investorType}")
+    public ResponseEntity<Page<RecommendStockResponse>> getRecommendStocksByInvestorType(
+            @PathVariable InvestorType investorType,
+            @RequestParam(value = "page", defaultValue = "0") int page,
+            @RequestParam(value = "size", defaultValue = "15") int size) {
+        Pageable pageable = PageRequest.of(page, size);
+        return ResponseEntity.ok(stockService.getRecommendStocksByInvestorType(investorType, pageable));
+    }
+
     @GetMapping("/v1/init-stocks")
     public void setInitStock() {
         stockService.setStockRepository();

diff --git a/src/main/java/com/bull4jo/kkanbustock/stock/domain/Stock.java b/src/main/java/com/bull4jo/kkanbustock/stock/domain/Stock.java
@@ -78,6 +78,12 @@ public class Stock {
     @Column
     private long mrktTotAmt;
 
+    public Stock(RecommendStock recommendStock) {
+        this.srtnCd = recommendStock.getSrtnCd();
+        this.itmsNm = recommendStock.getItmsNm();
+        this.clpr = recommendStock.getClpr();
+    }
+
     public void update(int clpr, int mkp) {
         this.clpr = clpr;
         this.mkp = mkp;

diff --git a/src/main/java/com/bull4jo/kkanbustock/stock/repository/RecommendStockRepository.java b/src/main/java/com/bull4jo/kkanbustock/stock/repository/RecommendStockRepository.java
@@ -1,12 +1,18 @@
 package com.bull4jo.kkanbustock.stock.repository;
 
+import com.bull4jo.kkanbustock.member.domain.entity.InvestorType;
 import com.bull4jo.kkanbustock.stock.domain.RecommendStock;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Repository;
 
+import java.util.Optional;
+
 @Repository
 public interface RecommendStockRepository extends JpaRepository<RecommendStock, String> {
+    @Override
     Page<RecommendStock> findAll(Pageable pageable);
+
+    Optional<Page<RecommendStock>> findByInvestorType(InvestorType investorType, Pageable pageable);
 }
diff --git a/src/main/java/com/bull4jo/kkanbustock/stock/service/StockService.java b/src/main/java/com/bull4jo/kkanbustock/stock/service/StockService.java
@@ -23,6 +23,7 @@
 import java.util.List;
 import java.util.Optional;
 import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 // 주식 넣고, 빼고, 업데이트, 가져오고, 있는지 확인하고
 @Service
@@ -301,6 +302,18 @@ public Page<RecommendStockResponse> getRecommendedStocks(Pageable pageable) {
         return new PageImpl<>(recommends, pageable, page.getTotalElements());
     }
 
+    @Transactional
+    public Page<RecommendStockResponse> getRecommendStocksByInvestorType(InvestorType investorType, Pageable pageable) {
+        Page<RecommendStock> page = recommendStockRepository.findByInvestorType(investorType, pageable).orElseThrow(() -> new CustomException(ErrorCode.RECOMMENDED_STOCK_NOT_FOUND));
+        List<RecommendStockResponse> recommends = page
+                .getContent()
+                .stream()
+                .map(RecommendStockResponse::new)
+                .collect(Collectors.toList());
+
+        return new PageImpl<>(recommends, pageable, page.getTotalElements());
+    }
+
     @Transactional
     public String update(final String strnCd, final Stock req) {
         Stock stock = stockRepository.findById(strnCd).orElseThrow(() -> new CustomException(ErrorCode.STOCK_NOT_FOUND));