[KC-108] ai api 실행

src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java

@@ -1,6 +1,8 @@
 package com.kamco.cd.kamcoback.inference.service;
 
+import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.kamco.cd.kamcoback.common.exception.CustomApiException;
 import com.kamco.cd.kamcoback.config.resttemplate.ExternalHttpClient;
 import com.kamco.cd.kamcoback.config.resttemplate.ExternalHttpClient.ExternalCallResult;
 import com.kamco.cd.kamcoback.inference.dto.InferenceDetailDto;
@@ -23,6 +25,7 @@ import com.kamco.cd.kamcoback.postgres.core.ModelMngCoreService;
 import jakarta.validation.constraints.NotNull;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
 import java.util.UUID;
@@ -33,6 +36,7 @@ import org.springframework.beans.factory.annotation.Value;
 import org.springframework.data.domain.Page;
 import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpMethod;
+import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -93,7 +97,7 @@ public class InferenceResultService {
 
     // 추론 테이블 저장
     UUID uuid = inferenceResultCoreService.saveInferenceInfo(req);
-    this.startInference(req);
+    this.startInference(req, uuid);
 
     return uuid;
   }
@@ -165,7 +169,7 @@ public class InferenceResultService {
    *
    * @param req
    */
-  private void startInference(InferenceResultDto.RegReq req) {
+  private void startInference(InferenceResultDto.RegReq req, UUID uuid) {
 
     List<MapSheetNumDto> mapSheetNum = req.getMapSheetNum();
     List<String> mapSheetNumList = new ArrayList<>();
@@ -193,9 +197,8 @@ public class InferenceResultService {
     m2.setPred_requests_areas(predRequestsAreas);
     m3.setPred_requests_areas(predRequestsAreas);
 
-    ensureAccepted(m1);
-    //        ensureAccepted(m2);
-    //        ensureAccepted(m3);
+    Long batchId = this.ensureAccepted(m1);
+    inferenceResultCoreService.update(uuid, batchId, "IN_PROGRESS");
   }
 
   /**
@@ -203,7 +206,7 @@ public class InferenceResultService {
    *
    * @param dto
    */
-  private void ensureAccepted(InferenceSendDto dto) {
+  private Long ensureAccepted(InferenceSendDto dto) {
     log.info("dto null? {}", dto == null);
     ObjectMapper om = new ObjectMapper();
     try {
@@ -227,6 +230,28 @@ public class InferenceResultService {
 
     int status = result.statusCode();
     String body = result.body();
+
+    if (status < 200 || status >= 300) {
+      throw new CustomApiException("BAD_GATEWAY", HttpStatus.BAD_GATEWAY);
+    }
+
+    Long batchId = 0L;
+
+    try {
+      List<Map<String, Object>> list =
+          om.readValue(body, new TypeReference<List<Map<String, Object>>>() {});
+
+      Integer batchIdInt = (Integer) list.get(0).get("batch_id");
+      batchId = batchIdInt.longValue();
+      if (list.isEmpty()) {
+        throw new IllegalStateException("Inference response is empty");
+      }
+
+    } catch (Exception e) {
+      log.error(e.getMessage());
+    }
+
+    return batchId;
   }
 
   /**