Merge pull request #23 from fallaylist/feat/ai-recommendation-valley

feat: AI 계곡 추천 기능 구현

Author: junest66

build.gradle.kts

@@ -3,6 +3,7 @@ plugins {
     id("org.springframework.boot") version "3.5.3"
     id("io.spring.dependency-management") version "1.1.7"
 }
+val springAiVersion by extra("1.0.1")
 
 group = "fallguys"
 version = "0.0.1-SNAPSHOT"
@@ -36,6 +37,7 @@ dependencies {
     implementation("org.springdoc:springdoc-openapi-starter-webmvc-ui:2.8.9")
     implementation("software.amazon.awssdk:bom:2.32.29")
     implementation("software.amazon.awssdk:s3:2.32.29")
+    implementation("org.springframework.ai:spring-ai-starter-model-openai")
 
     compileOnly("org.projectlombok:lombok")
     developmentOnly("org.springframework.boot:spring-boot-docker-compose")
@@ -50,6 +52,11 @@ dependencies {
     testImplementation("org.springframework.security:spring-security-test")
     testRuntimeOnly("org.junit.platform:junit-platform-launcher")
 }
+dependencyManagement {
+    imports {
+        mavenBom("org.springframework.ai:spring-ai-bom:$springAiVersion")
+    }
+}
 
 tasks.withType<Test> {
     useJUnitPlatform()

compose.prod.yaml

@@ -10,15 +10,22 @@ services:
       - redis
     env_file:
       - ./.env
-
     environment:
       - SPRING_PROFILES_ACTIVE=prod
+      - SPRING_DATASOURCE_URL=jdbc:mysql://mysql:3306/fallaylist?useSSL=false&serverTimezone=UTC&allowPublicKeyRetrieval=true
+      - SPRING_DATASOURCE_USERNAME=${MYSQL_USER}
+      - SPRING_DATASOURCE_PASSWORD=${MYSQL_PASSWORD}
     restart: always
 
   mysql:
     image: 'mysql:8.0'
     env_file:
       - ./.env
+    environment:
+      - MYSQL_ROOT_PASSWORD=${MYSQL_ROOT_PASSWORD}
+      - MYSQL_DATABASE=${MYSQL_DATABASE}
+      - MYSQL_USER=${MYSQL_USER}
+      - MYSQL_PASSWORD=${MYSQL_PASSWORD}
     ports:
       - "3306:3306"
     volumes:

src/main/java/fallguys/fallaylist/adapter/in/scheduler/recommendation/RecommendationScheduler.java

@@ -0,0 +1,17 @@
+package fallguys.fallaylist.adapter.in.scheduler.recommendation;
+
+import fallguys.fallaylist.application.recommendation.provided.ValleyRecommendationRegister;
+import lombok.RequiredArgsConstructor;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+@Component
+@RequiredArgsConstructor
+public class RecommendationScheduler {
+    private final ValleyRecommendationRegister recommendationRegister;
+
+    @Scheduled(cron = "0 0 2 * * ?")
+    public void updateRecommendations() {
+        recommendationRegister.update();
+    }
+}

src/main/java/fallguys/fallaylist/adapter/out/external/ai/AiRecommendationAdapter.java

@@ -0,0 +1,71 @@
+package fallguys.fallaylist.adapter.out.external.ai;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import fallguys.fallaylist.adapter.out.external.ai.dto.AiValleyRecommendationResponse;
+import fallguys.fallaylist.application.recommendation.dto.KeywordAiPromt;
+import fallguys.fallaylist.application.recommendation.dto.ValleyAiPromt;
+import fallguys.fallaylist.application.recommendation.required.AiRecommendationPort;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.prompt.PromptTemplate;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.core.ParameterizedTypeReference;
+import org.springframework.core.io.Resource;
+import org.springframework.stereotype.Component;
+import org.springframework.util.StreamUtils;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class AiRecommendationAdapter implements AiRecommendationPort {
+
+    @Value("classpath:/prompts/system-message.st")
+    private Resource systemMessageResource;
+
+    @Value("classpath:/prompts/user-message.st")
+    private Resource userMessageResource;
+
+    private final ChatClient chatClient;
+    private final ObjectMapper objectMapper;
+
+    @Override
+    public List<AiValleyRecommendationResponse> recommendValleys(List<ValleyAiPromt> valleys,
+                                                                 List<KeywordAiPromt> keywords) {
+        try {
+            String systemContent = getContent(systemMessageResource);
+
+            String valleysJson = objectMapper.writeValueAsString(valleys);
+            String keywordsJson = objectMapper.writeValueAsString(keywords);
+
+            PromptTemplate userPromptTemplate = new PromptTemplate(userMessageResource);
+            String userContent = userPromptTemplate.render(
+                    Map.of("valleysJson", valleysJson, "keywordsJson", keywordsJson)
+            );
+
+            return chatClient.prompt()
+                    .system(systemContent)
+                    .user(userContent)
+                    .call()
+                    .entity(new ParameterizedTypeReference<List<AiValleyRecommendationResponse>>() {
+                    });
+
+        } catch (JsonProcessingException e) {
+            log.error("Failed to serialize request data to JSON", e);
+            return Collections.emptyList();
+        } catch (IOException e) {
+            log.error("Failed to read prompt resource file", e);
+            throw new RuntimeException("Failed to read prompt files", e);
+        }
+    }
+
+    private String getContent(Resource resource) throws IOException {
+        return StreamUtils.copyToString(resource.getInputStream(), StandardCharsets.UTF_8);
+    }
+}

src/main/java/fallguys/fallaylist/adapter/out/external/ai/dto/AiValleyRecommendationResponse.java

@@ -0,0 +1,4 @@
+package fallguys.fallaylist.adapter.out.external.ai.dto;
+
+public record AiValleyRecommendationResponse(Long valleyId, Long keywordId, String title, String reason) {
+}

src/main/java/fallguys/fallaylist/adapter/out/external/config/ChatClientConfig.java

@@ -0,0 +1,13 @@
+package fallguys.fallaylist.adapter.out.external.config;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class ChatClientConfig {
+    @Bean
+    public ChatClient chatClient(ChatClient.Builder chatClientBuilder) {
+        return chatClientBuilder.build();
+    }
+}

src/main/java/fallguys/fallaylist/adapter/out/persistence/recommendation/ValleyRecommendationRepositoryImpl.java

@@ -0,0 +1,35 @@
+package fallguys.fallaylist.adapter.out.persistence.recommendation;
+
+import fallguys.fallaylist.application.recommendation.required.ValleyRecommendationRepositoryCustom;
+import fallguys.fallaylist.domain.recommendation.ValleyRecommendation;
+import java.sql.PreparedStatement;
+import java.sql.Timestamp;
+import java.util.List;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.stereotype.Repository;
+
+@Slf4j
+@Repository
+@RequiredArgsConstructor
+public class ValleyRecommendationRepositoryImpl implements ValleyRecommendationRepositoryCustom {
+    private final JdbcTemplate jdbcTemplate;
+
+    @Override
+    public void bulkInsert(List<ValleyRecommendation> recommendations) {
+        String sql = "INSERT INTO valley_recommendation " +
+                "(valley_id, keyword_id, title, reason, created_at, is_active) " +
+                "VALUES (?, ?, ?, ?, ?, ?)";
+
+        jdbcTemplate.batchUpdate(sql, recommendations, recommendations.size(),
+                (PreparedStatement ps, ValleyRecommendation recommendation) -> {
+                    ps.setLong(1, recommendation.getValley().getId());
+                    ps.setLong(2, recommendation.getKeyword().getId());
+                    ps.setString(3, recommendation.getTitle());
+                    ps.setString(4, recommendation.getReason());
+                    ps.setTimestamp(5, Timestamp.valueOf(recommendation.getCreatedAt()));
+                    ps.setBoolean(6, recommendation.isActive());
+                });
+    }
+}

src/main/java/fallguys/fallaylist/adapter/security/filter/JwtAuthenticationFilter.java

@@ -9,6 +9,7 @@
 import jakarta.servlet.http.HttpServletRequest;
 import jakarta.servlet.http.HttpServletResponse;
 import java.io.IOException;
+import java.util.List;
 import lombok.RequiredArgsConstructor;
 import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
 import org.springframework.security.core.Authentication;
@@ -21,9 +22,15 @@
 public class JwtAuthenticationFilter extends OncePerRequestFilter {
     private static final String AUTHORIZATION_HEADER = "Authorization";
     private static final String BEARER_PREFIX = "Bearer ";
+    private static final List<String> EXCLUDED_PATHS = List.of("/api/token/reissue");
 
     private final TokenParser tokenParser;
 
+    @Override
+    protected boolean shouldNotFilter(HttpServletRequest request) throws ServletException {
+        return EXCLUDED_PATHS.stream().anyMatch(request.getRequestURI()::startsWith);
+    }
+
     @Override
     protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
             throws ServletException, IOException {

src/main/java/fallguys/fallaylist/adapter/security/oauth2/CustomOAuth2SuccessHandler.java

@@ -61,6 +61,8 @@ public void onAuthenticationSuccess(HttpServletRequest request, HttpServletRespo
                 redirectUrl = "http://localhost:3000/login?success=true";
             }
         }
+        log.info("referer = {}", referer);
+        log.info("redirectUrl = {}", redirectUrl);
 
         response.sendRedirect(redirectUrl);
         log.info("소셜로그인 성공 user ID: {}", memberAuthInfo.id());

src/main/java/fallguys/fallaylist/adapter/webapi/api/member/dto/MemberInfoResponse.java

@@ -6,7 +6,19 @@ public record MemberInfoResponse(String nickname, String profileImageUrl) {
     public static MemberInfoResponse of(Member member, String cloudfrontUrl) {
         return new MemberInfoResponse(
                 member.getNickname(),
-                member.getProfileImageUrl() != null ? cloudfrontUrl + member.getProfileImageUrl() : null
+                buildProfileImageUrl(member.getProfileImageUrl(), cloudfrontUrl)
         );
     }
+
+    private static String buildProfileImageUrl(String originalUrl, String cloudfrontUrl) {
+        if (originalUrl == null) {
+            return null;
+        }
+
+        if (originalUrl.contains(".kakaocdn.net")) {
+            return originalUrl;
+        }
+
+        return cloudfrontUrl + originalUrl;
+    }
 }