Merge pull request #152 from prgrms-aibe-devcourse/refactor/expense-LLM

refactor : Prompt DB에서 관리

Author: lSNOTNULL

src/main/java/store/lastdance/domain/prompt/Prompt.java

@@ -0,0 +1,34 @@
+package store.lastdance.domain.prompt;
+
+import jakarta.persistence.*;
+import lombok.AccessLevel;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import store.lastdance.domain.common.BaseTimeEntity;
+
+@Entity
+@Table(name = "prompts")
+@Getter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+public class Prompt extends BaseTimeEntity {
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @Column(name = "prompt_type", unique = true, nullable = false) // Renamed to prompt_type
+    private String promptType;
+
+    @Column(name = "prompt_content", columnDefinition = "TEXT", nullable = false)
+    private String promptContent;
+
+    @Builder
+    public Prompt(String promptType, String promptContent) { // Renamed parameter
+        this.promptType = promptType;
+        this.promptContent = promptContent;
+    }
+
+    public void updateContent(String promptContent) {
+        this.promptContent = promptContent;
+    }
+}

src/main/java/store/lastdance/exception/ErrorCode.java

@@ -88,7 +88,8 @@ public enum ErrorCode {
     TOO_MANY_REQUESTS("요청이 너무 잦습니다. 잠시 후 다시 시도해주세요.", HttpStatus.TOO_MANY_REQUESTS),
     HISTORY_NOT_FOUND("LLM 분석 기록을 찾을 수 없습니다.", HttpStatus.NOT_FOUND )
     ,
-    INVALID_HISTORY_REQUEST("LLM 분석 기록 요청이 유효하지 않습니다.", HttpStatus.BAD_REQUEST);
+    INVALID_HISTORY_REQUEST("LLM 분석 기록 요청이 유효하지 않습니다.", HttpStatus.BAD_REQUEST), PROMPT_NOT_FOUND("프롬프트를 찾을 수 없습니다", HttpStatus.NOT_FOUND )
+    ,;
 
     private final String message;
     private final HttpStatus httpStatus;

src/main/java/store/lastdance/repository/prompt/PromptRepository.java

@@ -0,0 +1,10 @@
+package store.lastdance.repository.prompt;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+import store.lastdance.domain.prompt.Prompt;
+
+import java.util.Optional;
+
+public interface PromptRepository extends JpaRepository<Prompt, Long> {
+    Optional<Prompt> findByPromptType(String promptType);
+}

src/main/java/store/lastdance/service/admin/AdminServiceImpl.java

@@ -36,6 +36,7 @@
 import store.lastdance.exception.CustomException;
 import store.lastdance.exception.ErrorCode;
 
+import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
@@ -960,8 +961,53 @@ public ExpenseAnalyzerFeedbackStatsDTO getExpenseAnalyzerFeedbackStats(UUID user
             LocalDateTime startDate = parsePeriod(period, endDate);
             statsSummary = expenseAnalysisHistoryRepository.getFeedbackStatsSummaryBetween(startDate, endDate);
             trends = expenseAnalysisHistoryRepository.findFeedbackTrendsBetween(startDate, endDate);
-        }
 
+            // 날짜별 데이터 보간 로직 추가
+            Map<LocalDate, FeedbackTrendDTO> trendMap = trends.stream()
+                    .collect(Collectors.toMap(
+                            dto -> LocalDate.of(dto.year(), dto.month(), dto.day()),
+                            dto -> dto
+                    ));
+
+            List<FeedbackTrendDTO> interpolatedTrends = new ArrayList<>();
+            LocalDate current = startDate.toLocalDate();
+            long lastUpCount = 0L;
+            long lastDownCount = 0L;
+
+            while (!current.isAfter(endDate.toLocalDate())) {
+                FeedbackTrendDTO dailyTrend = trendMap.get(current);
+                long currentUpCountForInterpolation = lastUpCount;
+                long currentDownCountForInterpolation = lastDownCount;
+                long currentTotalCount = 0L;
+
+                if (dailyTrend != null) {
+                    currentTotalCount = dailyTrend.totalCount();
+                    // 해당 날짜에 실제 up/down 피드백이 있다면 그 값을 사용
+                    if (dailyTrend.upCount() > 0 || dailyTrend.downCount() > 0) {
+                        currentUpCountForInterpolation = dailyTrend.upCount();
+                        currentDownCountForInterpolation = dailyTrend.downCount();
+                    }
+                    // else, dailyTrend가 존재하지만 up/down이 0이면 이전 값을 유지 (currentUpCountForInterpolation, currentDownCountForInterpolation은 lastUpCount, lastDownCount 그대로)
+                }
+
+                // 현재 날짜의 DTO를 추가 (필요시 보간된 값 사용)
+                interpolatedTrends.add(new FeedbackTrendDTO(
+                        current.getYear(),
+                        current.getMonthValue(),
+                        current.getDayOfMonth(),
+                        currentTotalCount, // 실제 totalCount 사용, 없으면 0
+                        currentUpCountForInterpolation,
+                        currentDownCountForInterpolation
+                ));
+
+                // 다음 반복을 위해 lastUpCount와 lastDownCount 업데이트
+                lastUpCount = currentUpCountForInterpolation;
+                lastDownCount = currentDownCountForInterpolation;
+
+                current = current.plusDays(1);
+            }
+            trends = interpolatedTrends;
+        }
         long upCount = statsSummary.upCount() != null ? statsSummary.upCount() : 0L;
         long downCount = statsSummary.downCount() != null ? statsSummary.downCount() : 0L;
         long totalFeedbacks = upCount + downCount;

src/main/java/store/lastdance/service/expense/ExpenseAnalyzerImpl.java

@@ -11,6 +11,7 @@
 import store.lastdance.dto.gemini.GeminiResponseDTO;
 import store.lastdance.exception.CustomException;
 import store.lastdance.exception.ErrorCode;
+import store.lastdance.service.prompt.PromptService; // Import PromptService
 
 import java.util.List;
 import java.util.regex.Matcher;
@@ -29,16 +30,21 @@
 @Slf4j
 public class ExpenseAnalyzerImpl implements ExpenseAnalyzer {
 
+    // Regex Pattern for JSON block parsing (kept as static final for now)
+    private static final Pattern JSON_BLOCK_PATTERN = Pattern.compile("```json\\s*([\\s\\S]+?)\\s*```|```\\s*([\\s\\S]+?)\\s*```", Pattern.CASE_INSENSITIVE);
+
     private final ObjectMapper objectMapper;
     private final WebClient webClient;
+    private final PromptService promptService; // Inject PromptService
 
-    public ExpenseAnalyzerImpl(ObjectMapper objectMapper, @Value("${GOOGLE_GEMINI_KEY}") String apiKey
+    public ExpenseAnalyzerImpl(ObjectMapper objectMapper, @Value("${GOOGLE_GEMINI_KEY}") String apiKey, PromptService promptService
     ) {
         this.objectMapper = objectMapper;
         this.webClient = WebClient.builder()
                 .baseUrl("https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key=" + apiKey)
                 .defaultHeader("Content-Type", "application/json")
                 .build();
+        this.promptService = promptService; // Assign injected service
     }
 
     @Override
@@ -56,47 +62,8 @@ public AnalyzeExpenseResponseDTO.Suggestion analyzerExpenseData(String expenseJs
     }
 
     private String createPrompt(String expenseJson) {
-        String systemInstruction = """
-                당신은 재정관리 전문가입니다.
-                사용자의 지출 데이터를 기반으로, 불필요한 지출을 줄일 수 있는 가장 효과적인 절약 팁 하나를 제안해주세요.
-                답변은 오직 *JSON 형식*에 맞춰 개선 제안 하나만 포함해야 합니다.
-                """;
-        String userPrompt = "나의 지출 내역은 다음과 같다: ```json\n" + expenseJson + "\n```\n\n" +
-                            "이 데이터를 바탕으로 가장 효과적인 개선 제안 하나를 다음 형식에 맞춰 제공해주세요.";
-        String formatInstruction = """
-                ***Format***
-                반드시 첫 줄부터 아래 포맷만 출력하고, 안내 문구나 예시 등은 출력하지 마세요.
-                - "title" : [개선 제안의 제목]
-                - "description" : [구체적인 설명, Markdown 형식으로 작성]
-                - "effect" : [예상되는 효과]
-                - "difficulty" : [쉬움/보통/어려움 중 하나]
-
-                ***Format Example***
-                {
-                "title" : "자동 저축 설정"
-                "description" : **Markdown 형식으로 작성**
-                 "
-                 지출 내역을 분석한 결과, '그룹' 유형의 지출이 상당히 많습니다. 특히 '식비', '유흥', '쇼핑' 카테고리
-                 에서 그룹 지출이 빈번하게 발생하고 있습니다.
-                 ### 제안:
-
-                - 지출 항목 분석: 그룹 내에서 어떤 항목에 가장 많은 지출이 발생하는지 파악합니다. (예: 식비, 엔터테인먼트, 쇼핑 등)
-                - 예산 설정: 각 항목별로 합리적인 월별 예산을 설정합니다. 예산을 초과하지 않도록 그룹 구성원들과 함께 노력합니다.
-                - 대안 모색: 더 저렴한 대안을 찾아봅니다. (예: 외식 대신 직접 요리, 저렴한 엔터테인먼트 활동 찾기)
-                - 정기적인 검토: 매달 예산 대비 실제 지출을 검토하고, 필요에 따라 예산을 조정합니다.
-                "
-                "effect" : "연간 목표 달성률 40% 향상"
-                "difficulty" : "쉬움"
-                }
-                ************
-                위 형식을 반드시 준수하여 JSON 객체만 응답하세요.
-                """;
-
-        String finalPrompt = systemInstruction + "\n\n" +
-                             userPrompt + "\n\n" +
-                             formatInstruction;
-
-        return finalPrompt;
+        String combinedPromptTemplate = promptService.getPromptContent("LLM_EXPENSE_ANALYSIS_PROMPT");
+        return String.format(combinedPromptTemplate, expenseJson);
     }
     private GeminiRequestDTO createRequestJson(String prompt) {
         GeminiRequestDTO dto = new GeminiRequestDTO(List.of(new GeminiRequestDTO.Content(List.of(new GeminiRequestDTO.Part(prompt)))));
@@ -114,8 +81,7 @@ private AnalyzeExpenseResponseDTO.Suggestion parseSuggestionResponse(GeminiRespo
             log.info("LLM이 생성한 텍스트 : {}", rawText);
 
             String jsonText = null;
-            Pattern p = Pattern.compile("```json\\s*([\\s\\S]+?)\\s*```|```\\s*([\\s\\S]+?)\\s*```", Pattern.CASE_INSENSITIVE);
-            Matcher m = p.matcher(rawText);
+            Matcher m = JSON_BLOCK_PATTERN.matcher(rawText); // Use the static final Pattern
             if(m.find()) {
                 jsonText = m.group(1) != null ? m.group(1) : m.group(2);
             } else if (rawText.startsWith("{") && rawText.endsWith("}")) {

src/main/java/store/lastdance/service/prompt/PromptService.java

@@ -0,0 +1,24 @@
+package store.lastdance.service.prompt;
+
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.cache.annotation.Cacheable;
+import org.springframework.stereotype.Service;
+import store.lastdance.domain.prompt.Prompt;
+import store.lastdance.exception.CustomException;
+import store.lastdance.exception.ErrorCode;
+import store.lastdance.repository.prompt.PromptRepository;
+
+@Service
+@RequiredArgsConstructor
+@Slf4j
+public class PromptService {
+    private final PromptRepository promptRepository;
+
+    @Cacheable(value = "prompts", key = "#promptType")
+    public String getPromptContent(String promptType) {
+        return promptRepository.findByPromptType(promptType)
+                .map(Prompt::getPromptContent)
+                .orElseThrow(() -> new CustomException(ErrorCode.PROMPT_NOT_FOUND));
+    }
+}