Merge pull request #43 from Block-Guard/refactor/#42/gpt-api-client

[Refactor] AI 서버 통신 로직 리펙토링

Author: yeonju73

src/main/java/com/blockguard/server/domain/analysis/application/FraudAnalysisService.java

@@ -5,18 +5,12 @@
 import com.blockguard.server.domain.analysis.dto.request.GptRequest;
 import com.blockguard.server.domain.analysis.dto.response.FraudAnalysisResponse;
 import com.blockguard.server.domain.analysis.dto.response.GptResponse;
-import com.blockguard.server.global.common.codes.ErrorCode;
-import com.blockguard.server.global.exception.BusinessExceptionHandler;
+import com.blockguard.server.infra.gpt.GptApiClient;
 import com.blockguard.server.infra.ocr.NaverOcrClient;
-import lombok.AllArgsConstructor;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.beans.factory.annotation.Value;
-import org.springframework.http.*;
 import org.springframework.stereotype.Service;
 import org.springframework.util.StringUtils;
-import org.springframework.web.client.RestClientException;
-import org.springframework.web.client.RestTemplate;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.IOException;
@@ -29,10 +23,7 @@
 public class FraudAnalysisService {
 
     private final NaverOcrClient naverOcrClient;
-    private final RestTemplate restTemplate;
-
-    @Value("${ai.server.url}")
-    private String aiBaseUrl;
+    private final GptApiClient gptApiClient;
 
     public FraudAnalysisResponse fraudAnalysis(FraudAnalysisRequest fraudAnalysisRequest, List<MultipartFile> imageFiles) {
         List<String> keywords = new ArrayList<>();
@@ -44,36 +35,17 @@ public FraudAnalysisResponse fraudAnalysis(FraudAnalysisRequest fraudAnalysisReq
 
         GptRequest gptRequest = buildGptRequest(fraudAnalysisRequest, keywords, additionalDescription, imageContent);
 
-        try {
-            HttpHeaders headers = new HttpHeaders();
-            headers.setContentType(MediaType.APPLICATION_JSON);
-
-            HttpEntity<GptRequest> entity = new HttpEntity<>(gptRequest, headers);
-            ResponseEntity<GptResponse> response = restTemplate.postForEntity(
-                    aiBaseUrl,
-                    entity,
-                    GptResponse.class
-            );
-
-            if (response.getBody() == null || !response.getStatusCode().is2xxSuccessful()) {
-                log.error("AI 서버 응답 오류: status={}, body={}", response.getStatusCode(), response.getBody());
-                throw new BusinessExceptionHandler(ErrorCode.AI_SERVER_ERROR);
-            }
+        // ai 서버 호출
+        GptResponse gptResponse = gptApiClient.analyze(gptRequest);
 
-            double score = response.getBody().getScore();
-            return FraudAnalysisResponse
-                    .builder()
-                    .keywords(response.getBody().getKeywords())
-                    .score(response.getBody().getScore())
-                    .estimatedFraudType(response.getBody().getEstimatedFraudType())
-                    .explanation(response.getBody().getExplanation())
-                    .riskLevel(RiskLevel.fromScore(score).getValue())
-                    .build();
-
-        } catch (RestClientException e){
-            log.error("AI 서버 통신 실패: {}", e.getMessage());
-            throw new BusinessExceptionHandler(ErrorCode.AI_SERVER_ERROR);
-        }
+        double score = gptResponse.getScore();
+        return FraudAnalysisResponse.builder()
+                .keywords(gptResponse.getKeywords())
+                .score(gptResponse.getScore())
+                .estimatedFraudType(gptResponse.getEstimatedFraudType())
+                .explanation(gptResponse.getExplanation())
+                .riskLevel(RiskLevel.fromScore(score).getValue())
+                .build();
     }
 
     private String extractOcrText(List<MultipartFile> imageFiles) {

src/main/java/com/blockguard/server/infra/gpt/GptApiClient.java

@@ -0,0 +1,54 @@
+package com.blockguard.server.infra.gpt;
+
+import com.blockguard.server.domain.analysis.dto.request.GptRequest;
+import com.blockguard.server.domain.analysis.dto.response.GptResponse;
+import com.blockguard.server.global.common.codes.ErrorCode;
+import com.blockguard.server.global.exception.BusinessExceptionHandler;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Component;
+import org.springframework.web.client.RestClientException;
+import org.springframework.web.client.RestTemplate;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class GptApiClient {
+
+    private final RestTemplate restTemplate;
+
+    @Value("${ai.server.url}")
+    private String aiServerUrl;
+
+    public GptResponse analyze(GptRequest gptRequest) {
+        log.info("gpt server 사기 분석 api 호출");
+
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_JSON);
+
+        HttpEntity<GptRequest> entity = new HttpEntity<>(gptRequest, headers);
+
+        try {
+            ResponseEntity<GptResponse> response = restTemplate.postForEntity(
+                    aiServerUrl,
+                    entity,
+                    GptResponse.class
+            );
+
+            if (response.getBody() == null || !response.getStatusCode().is2xxSuccessful()) {
+                log.error("GPT API error: status={}, body={}", response.getStatusCode(), response.getBody());
+                throw new BusinessExceptionHandler(ErrorCode.AI_SERVER_ERROR);
+            }
+            return response.getBody();
+
+        } catch (RestClientException e){
+            log.error("GPT API communication failed: {}", e.getMessage());
+            throw new BusinessExceptionHandler(ErrorCode.AI_SERVER_ERROR);
+        }
+    }
+}