Merge pull request #73 from julia-heo/develop

[Test] 혜택 요약 테스트 코드 추가
EWHA-LUX · Apr 7, 2024 · a7e46ad · a7e46ad
2 parents 679eee2 + 3555c1d
commit a7e46ad
Show file tree

Hide file tree

Showing 7 changed files with 192 additions and 12 deletions.
diff --git a/src/main/java/ewha/lux/once/domain/card/controller/CardController.java b/src/main/java/ewha/lux/once/domain/card/controller/CardController.java
@@ -1,10 +1,8 @@
 package ewha.lux.once.domain.card.controller;
 
-import ewha.lux.once.domain.card.dto.CardGoalRequestDto;
-import ewha.lux.once.domain.card.dto.CardPerformanceRequestDto;
-import ewha.lux.once.domain.card.dto.CodefCardListRequestDto;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import ewha.lux.once.domain.card.dto.*;
 import ewha.lux.once.domain.card.service.CardService;
-import ewha.lux.once.domain.card.dto.MainCardRequestDto;
 import ewha.lux.once.domain.card.service.CrawlingService;
 import ewha.lux.once.global.common.CommonResponse;
 import ewha.lux.once.global.common.CustomException;
@@ -37,6 +35,33 @@ public CommonResponse<?> testtest(@AuthenticationPrincipal UserAccount user, @Pa
             return new CommonResponse<>(e.getStatus());
         }
     }
+
+    @GetMapping("/test/summary")
+    @ResponseBody
+    public CommonResponse<?> testSummary(@AuthenticationPrincipal UserAccount user, @RequestBody TestSummaryDto testSummaryDto) {
+        try {
+            cardService.updateBenefitSummaryTest(testSummaryDto.getPrompt(), testSummaryDto.getModel_name());
+            return new CommonResponse<>(ResponseCode.SUCCESS);
+        } catch (CustomException e) {
+            return new CommonResponse<>(e.getStatus());
+        } catch (JsonProcessingException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    @GetMapping("/test/summary/index")
+    @ResponseBody
+    public CommonResponse<?> testSummaryByIndex(@AuthenticationPrincipal UserAccount user, @RequestBody TestSummaryIndexDto testSummaryIndexDto) {
+        try {
+            cardService.updateBenefitSummaryTestByIndex(testSummaryIndexDto.getPrompt(), testSummaryIndexDto.getModel_name(), testSummaryIndexDto.getStart_index(), testSummaryIndexDto.getEnd_index());
+            return new CommonResponse<>(ResponseCode.SUCCESS);
+        } catch (CustomException e) {
+            return new CommonResponse<>(e.getStatus());
+        } catch (JsonProcessingException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
     // ============================================================
 
     // [Get] 마이월렛 조회
@@ -84,12 +109,13 @@ public CommonResponse<?> cardGoal(@AuthenticationPrincipal UserAccount user, @Re
             return new CommonResponse<>(e.getStatus());
         }
     }
+
     // [Get] CODEF 보유 카드 조회
     @GetMapping("/list")
     @ResponseBody
-    public CommonResponse<?> codefCardList (@AuthenticationPrincipal UserAccount user, @RequestBody CodefCardListRequestDto codefCardListRequestDto) {
+    public CommonResponse<?> codefCardList(@AuthenticationPrincipal UserAccount user, @RequestBody CodefCardListRequestDto codefCardListRequestDto) {
         try {
-            return new CommonResponse<>(ResponseCode.SUCCESS,cardService.getCodefCardList(user.getUsers(), codefCardListRequestDto));
+            return new CommonResponse<>(ResponseCode.SUCCESS, cardService.getCodefCardList(user.getUsers(), codefCardListRequestDto));
         } catch (CustomException e) {
             return new CommonResponse<>(e.getStatus());
         }
@@ -98,7 +124,7 @@ public CommonResponse<?> codefCardList (@AuthenticationPrincipal UserAccount use
     // [Post] 주카드 등록
     @PostMapping("/main")
     @ResponseBody
-    public CommonResponse<?> registerMainCard (@AuthenticationPrincipal UserAccount user, @RequestBody MainCardRequestDto mainCardRequestDto) {
+    public CommonResponse<?> registerMainCard(@AuthenticationPrincipal UserAccount user, @RequestBody MainCardRequestDto mainCardRequestDto) {
         try {
             cardService.postRegisterCard(user.getUsers(), mainCardRequestDto);
             return new CommonResponse<>(ResponseCode.SUCCESS);
@@ -110,7 +136,7 @@ public CommonResponse<?> registerMainCard (@AuthenticationPrincipal UserAccount
     // [Get] 주카드 실적 업데이트
     @GetMapping("/main/performance")
     @ResponseBody
-    public CommonResponse<?> registerMainCard (@AuthenticationPrincipal UserAccount user) {
+    public CommonResponse<?> registerMainCard(@AuthenticationPrincipal UserAccount user) {
         try {
             cardService.updateOwnedCardsPerformance(user.getUsers());
             return new CommonResponse<>(ResponseCode.SUCCESS);

diff --git a/src/main/java/ewha/lux/once/domain/card/dto/TestSummaryDto.java b/src/main/java/ewha/lux/once/domain/card/dto/TestSummaryDto.java
@@ -0,0 +1,15 @@
+package ewha.lux.once.domain.card.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+@Builder
+public class TestSummaryDto {
+    private String prompt;
+    private String model_name;
+}
diff --git a/src/main/java/ewha/lux/once/domain/card/dto/TestSummaryIndexDto.java b/src/main/java/ewha/lux/once/domain/card/dto/TestSummaryIndexDto.java
@@ -0,0 +1,17 @@
+package ewha.lux.once.domain.card.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+@Builder
+public class TestSummaryIndexDto {
+    private String prompt;
+    private String model_name;
+    private long start_index;
+    private long end_index;
+}
diff --git a/src/main/java/ewha/lux/once/domain/card/service/CardService.java b/src/main/java/ewha/lux/once/domain/card/service/CardService.java
@@ -216,7 +216,7 @@ public void updateOwnedCardsPerformance(Users nowUser) throws CustomException {
     }
 
     // 매주 월요일 04:00 AM 카드 혜택 정보 요약 작업
-    @Scheduled(cron = "0 0 4 ? * 1")
+//    @Scheduled(cron = "0 0 4 ? * 1")
     public void updateBenefitSummary() throws CustomException, JsonProcessingException {
 
         List<Card> cardList = cardRepository.findAll();
@@ -243,4 +243,81 @@ public void updateBenefitSummary() throws CustomException, JsonProcessingExcepti
         }
         log.info("전체 카드 혜택 요약 완료");
     }
+
+    // ** 추후 삭제해야 함 - 테스트용 ** ==================================
+    // GET /card/test/summary : 전체 카드 혜택 요약
+    public void updateBenefitSummaryTest(String prompt, String model_name) throws CustomException, JsonProcessingException {
+
+        List<Card> cardList = cardRepository.findAll();
+
+        int index = 1;
+        for (Card card : cardList) {
+            // 기존의 BenefitSummary 삭제
+            List<BenefitSummary> existingSummaries = benefitSummaryRepository.findByCard(card);
+            benefitSummaryRepository.deleteAll(existingSummaries);
+
+            log.info("[" + card.getName() + "] - 카드 혜택 요약 중... (" + index + "/" + cardList.size() + ")");
+
+            BenefitDto[] benefitJson = openaiService.gptBenefitSummaryTest(card.getBenefits(), prompt, model_name);
+
+            if (benefitJson == null) {
+                System.out.println("===========PASS=========" + card.getName());
+//                cardRepository.delete(card);
+                continue;
+            }
+            for (BenefitDto benefit : benefitJson) {
+                BenefitSummary benefitSummary = BenefitSummary.builder()
+                        .benefitField(benefit.getBenefit_field())
+                        .benefitContents(benefit.getContent())
+                        .card(card)
+                        .build();
+
+                benefitSummaryRepository.save(benefitSummary);
+            }
+            index++;
+        }
+        log.info("전체 카드 혜택 요약 완료");
+    }
+
+    /*
+        GET /card/test/summary/index : 일부 카드 혜택 요약
+        "prompt": 프롬프트 내용,
+        "model_name": open_ai model,
+        "start_index": 시작 인덱스,
+        "end_index": 종료 인덱스 (포함)
+     */
+    public void updateBenefitSummaryTestByIndex(String prompt, String model_name, long start_index, long end_index) throws CustomException, JsonProcessingException {
+
+        List<Card> cardList = cardRepository.findByIdBetween(start_index, end_index);
+
+
+        int index = 1;
+        for (Card card : cardList) {
+            // 기존의 BenefitSummary 삭제
+            List<BenefitSummary> existingSummaries = benefitSummaryRepository.findByCard(card);
+            benefitSummaryRepository.deleteAll(existingSummaries);
+
+            log.info("[" + card.getName() + " (card_id=" + card.getId() + ")] - 카드 혜택 요약 중... (" + index + "/" + cardList.size() + ")");
+
+            BenefitDto[] benefitJson = openaiService.gptBenefitSummaryTest(card.getBenefits(), prompt, model_name);
+
+            if (benefitJson == null) {
+                System.out.println("===========PASS=========" + card.getName());
+//                cardRepository.delete(card);
+                continue;
+            }
+            for (BenefitDto benefit : benefitJson) {
+                BenefitSummary benefitSummary = BenefitSummary.builder()
+                        .benefitField(benefit.getBenefit_field())
+                        .benefitContents(benefit.getContent())
+                        .card(card)
+                        .build();
+
+                benefitSummaryRepository.save(benefitSummary);
+            }
+            index++;
+        }
+        log.info("일부 카드 혜택 요약 완료");
+    }
+    // ============================================================
 }
diff --git a/src/main/java/ewha/lux/once/domain/card/service/CrawlingService.java b/src/main/java/ewha/lux/once/domain/card/service/CrawlingService.java
@@ -21,9 +21,10 @@ public class CrawlingService {
     private static final Logger LOG = LoggerFactory.getLogger(CrawlingService.class);
 
     // 매주 월요일 00:00 카드 혜택 크롤링
-    @Scheduled(cron = "0 0 0 ? * 1")
+    // ** 크롤링 자동화 테스트 -ing ** =========
+    @Scheduled(cron = "0 40 9 ? * *")
     public void cardCrawling() throws CustomException {
-        String[] cardCompanyList = {"Kookmin", "Hyundai", "Samsung", "Shinhan", "Lotte", "Hana"};
+        String[] cardCompanyList = {"Kookmin", "Hana", "Samsung", "Shinhan", "Lotte", "Hyundai"};
         for (String cardCompany : cardCompanyList){
             crawling(cardCompany);
         }
@@ -88,4 +89,4 @@ private static void executeInsertData(String firstInput, String secondInput) thr
             throw new CustomException(ResponseCode.CARD_BENEFITS_INSERT_FAIL);
         }
     }
-}
+}
diff --git a/src/main/java/ewha/lux/once/domain/home/service/OpenaiService.java b/src/main/java/ewha/lux/once/domain/home/service/OpenaiService.java
@@ -16,7 +16,9 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpStatus;
 import org.springframework.stereotype.Service;
+import org.springframework.web.client.HttpClientErrorException;
 import org.springframework.web.client.RestTemplate;
 
 import java.util.List;
@@ -110,4 +112,44 @@ public BenefitDto[] gptBenefitSummary(String benefits) throws CustomException, J
 
         return benefitJson;
     }
+    // ** 추후 삭제해야 함 - 테스트용 ** ==================================
+    public BenefitDto[] gptBenefitSummaryTest(String benefits, String prompt, String model_name) throws CustomException, JsonProcessingException {
+        try {
+            // gpt 요청 보내는 부분
+            OpenaiChatRequest request = new OpenaiChatRequest(model_name, prompt, benefits);
+            OpenaiChatResponse response = restTemplate.postForObject(apiUrl, request, OpenaiChatResponse.class);
+
+            if (response == null || response.getChoices() == null || response.getChoices().isEmpty()) {
+                throw new CustomException(ResponseCode.FAILED_TO_OPENAI);
+            }
+
+            String result = response.getChoices().get(0).getMessage().getContent();
+
+            ObjectMapper objectMapper = new ObjectMapper();
+            BenefitDto[] benefitJson = objectMapper.readValue(result, BenefitDto[].class);
+
+            return benefitJson;
+        } catch(CustomException | JsonProcessingException | HttpClientErrorException e){
+            e.printStackTrace();
+            System.out.println("===========오류==========");
+            int i=2;
+            while (i>0) {
+                try {
+                    OpenaiChatRequest request = new OpenaiChatRequest("gpt-4-turbo-preview", prompt, benefits);
+                    OpenaiChatResponse response = restTemplate.postForObject(apiUrl, request, OpenaiChatResponse.class);
+                    String result = response.getChoices().get(0).getMessage().getContent();
+
+                    ObjectMapper objectMapper = new ObjectMapper();
+                    BenefitDto[] benefitJson = objectMapper.readValue(result, BenefitDto[].class);
+                    return benefitJson;
+//                } catch (JsonProcessingException | InterruptedException| HttpClientErrorException ex) {
+                } catch (JsonProcessingException | HttpClientErrorException ex) {
+                    ex.printStackTrace();
+                    i--;
+                }
+            }
+            return null;
+        }
+    }
+    // ============================================================
 }
diff --git a/src/main/java/ewha/lux/once/global/repository/CardRepository.java b/src/main/java/ewha/lux/once/global/repository/CardRepository.java
@@ -13,5 +13,7 @@ public interface CardRepository extends JpaRepository<Card, Long> {
     List<Card> findAllByNameContains(String name);
     Optional<Card> findCardByName(String name);
     List<Card> findByNameContainingAndCardCompanyIn(String name, List<CardCompany> cardCompanies);
+
+    List<Card> findByIdBetween(long startIndex, long endIndex);
 }