Merge pull request #30 from Block-Guard/feat/#28/connet-fastapi-server

[Feat] FastAPI 서버 연동

Author: chaeyuuu

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

@@ -1,13 +1,22 @@
 package com.blockguard.server.domain.analysis.application;
 
+import com.blockguard.server.domain.analysis.domain.enums.RiskLevel;
 import com.blockguard.server.domain.analysis.dto.request.FraudAnalysisRequest;
 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.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;
@@ -16,10 +25,14 @@
 
 @Service
 @Slf4j
-@AllArgsConstructor
+@RequiredArgsConstructor
 public class FraudAnalysisService {
 
     private final NaverOcrClient naverOcrClient;
+    private final RestTemplate restTemplate;
+
+    @Value("${ai.server.url}")
+    private String aiBaseUrl;
 
     public FraudAnalysisResponse fraudAnalysis(FraudAnalysisRequest fraudAnalysisRequest, List<MultipartFile> imageFiles) {
         List<String> keywords = new ArrayList<>();
@@ -29,17 +42,38 @@ public FraudAnalysisResponse fraudAnalysis(FraudAnalysisRequest fraudAnalysisReq
         String imageContent = "";
         imageContent = extractOcrText(imageFiles);
 
-        // TODO: gpt 서버와 연동 예정
         GptRequest gptRequest = buildGptRequest(fraudAnalysisRequest, keywords, additionalDescription, imageContent);
 
-        return FraudAnalysisResponse
-                .builder()
-                .keywords(keywords)
-                .additionalDescription(additionalDescription)
-                .messageContent(StringUtils.hasText(fraudAnalysisRequest.getMessageContent()) ? fraudAnalysisRequest.getMessageContent() : null)
-                .imageContent(StringUtils.hasText(imageContent) ? imageContent : null)
-                .build();
+        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);
+            }
+
+            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).getName())
+                    .build();
+
+        } catch (RestClientException e){
+            log.error("AI 서버 통신 실패: {}", e.getMessage());
+            throw new BusinessExceptionHandler(ErrorCode.AI_SERVER_ERROR);
+        }
     }
 
     private String extractOcrText(List<MultipartFile> imageFiles) {

src/main/java/com/blockguard/server/domain/analysis/domain/enums/FraudType.java

@@ -1,9 +1,38 @@
 package com.blockguard.server.domain.analysis.domain.enums;
 
-// Todo: 사기 클래스 확정 후 변경
+import lombok.Getter;
+
+@Getter
 public enum FraudType {
-    PROSECUTOR_IMPERSONATION,
-    FAMILY_IMPERSONATION,
-    LOAN_FRAUD,
-    CARD_DELIVERY_IMPERSONATION
+    VOICE_PHISHING           ("보이스피싱"),
+    MESSENGER_PHISHING       ("메신저피싱"),
+    SMISHING                 ("스미싱"),
+    INSTITUTION_IMPERSONATION("기관사칭형"),
+    LOAN_FRAUD               ("대출사기형"),
+    CARD_COMPANY_IMPERSONATION("카드사 사칭형"),
+    FAMILY_IMPERSONATION      ("가족/지인사칭형"),
+    CELEBRATORY_IMPERSONATION("경조사사칭형"),
+    FIRST_BIRTHDAY_INVITE     ("돌잔치 초대장형"),
+    MOBILE_WEDDING_INVITE     ("모바일 청첩장형"),
+    OBITUARY_MESSAGE          ("부고문자형"),
+    PUBLIC_AGENCY_IMPERSONATION("공공기관 사칭형"),
+    FINE_AND_PENALTY          ("범칙금/과태료 부과형"),
+    HEALTH_INSURANCE_IMPERSONATION("건강보험공단 사칭형"),
+    POLICE_SUMMONS            ("경찰출석요구형"),
+    NATIONAL_TAX_SERVICE      ("국세청 사칭형"),
+    PART_TIME_SCAM            ("알바/부업 사기형"),
+    GOVERNMENT_GRANT_SCAM     ("정부지원금 위장형"),
+    DELIVERY_SCAM             ("택배사기형"),
+    INVESTMENT_SCAM           ("투자사기형"),
+    CRYPTOCURRENCY_SCAM       ("가상화폐 사기형"),
+    STOCK_INVESTMENT_SCAM     ("주식투자 사기형"),
+    IPO_SCAM                  ("청약 공모주 사기형"),
+    FALSE_PAYMENT_SCAM        ("허위결제사기형");
+
+    private final String korName;
+
+    FraudType(String korName) {
+        this.korName = korName;
+    }
+
 }

src/main/java/com/blockguard/server/domain/analysis/domain/enums/RiskLevel.java

@@ -1,5 +1,27 @@
 package com.blockguard.server.domain.analysis.domain.enums;
 
+import lombok.Getter;
+
+@Getter
 public enum RiskLevel {
-    Dangers, Caution, Safety
+    Dangers("위험"),
+    Caution("주의"),
+    Safety("안전");
+
+    private final String name;
+
+    RiskLevel(String name) {
+        this.name = name;
+    }
+
+    public static RiskLevel fromScore(double score) {
+        if (score <= 30) {
+            return Safety;
+        } else if (score <= 60) {
+            return Caution;
+        } else {
+            return Dangers;
+        }
+    }
+
 }

src/main/java/com/blockguard/server/domain/analysis/dto/response/FraudAnalysisResponse.java

@@ -12,9 +12,9 @@
 @AllArgsConstructor
 @NoArgsConstructor
 public class FraudAnalysisResponse {
-    // for test
-    private String messageContent;
+    private String riskLevel;
+    private Double score;
+    private String estimatedFraudType;
     private List<String> keywords;
-    private String additionalDescription;
-    private String imageContent;
+    private String explanation;
 }

src/main/java/com/blockguard/server/domain/analysis/dto/response/GptResponse.java

@@ -0,0 +1,20 @@
+package com.blockguard.server.domain.analysis.dto.response;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+import java.util.List;
+
+@Getter
+@Builder
+@AllArgsConstructor
+@NoArgsConstructor
+public class GptResponse {
+    private Double score;
+    private String estimatedFraudType;
+    private List<String> keywords;
+    private String explanation;
+}
+

src/main/java/com/blockguard/server/global/common/codes/ErrorCode.java

@@ -29,7 +29,8 @@ public enum ErrorCode {
     IMAGE_UPLOAD_LIMIT_EXCEEDED(HttpStatus.BAD_REQUEST, 4026, "이미지는 최대 2장까지만 업로드 가능합니다."),
 
     // 5000~ : server error
-    INTERNAL_SERVER_ERROR(HttpStatus.INTERNAL_SERVER_ERROR, 5000, "서버 오류가 발생했습니다.");
+    INTERNAL_SERVER_ERROR(HttpStatus.INTERNAL_SERVER_ERROR, 5000, "서버 오류가 발생했습니다."),
+    AI_SERVER_ERROR(HttpStatus.INTERNAL_SERVER_ERROR, 5001, "AI 서버와의 통신 오류가 발생했습니다.");
 
     private final HttpStatus status;
     private final int code;