shp 파일 생성 추가

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

@@ -3,7 +3,7 @@ package com.kamco.cd.kamcoback.inference.service;
 import com.kamco.cd.kamcoback.inference.ShpWriter;
 import com.kamco.cd.kamcoback.inference.dto.InferenceResultShpDto;
 import com.kamco.cd.kamcoback.postgres.core.InferenceResultShpCoreService;
-import com.kamco.cd.kamcoback.postgres.core.InferenceResultShpCoreService.ShpKey;
+import com.kamco.cd.kamcoback.postgres.entity.MapSheetAnalDataInferenceGeomEntity;
 import java.util.List;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -17,42 +17,66 @@ public class InferenceResultShpService {
   private final InferenceResultShpCoreService coreService;
   private final ShpWriter shpWriter;
 
+  /**
+   * inference_results -> tb_map_sheet_anal_data_inference / geom 업서트
+   */
+  @Transactional
   public void saveInferenceResultData() {
-
-    coreService.streamGrouped(
-        1000,
-        (key, entities) -> {
-
-          // Entity -> DTO
-          List<InferenceResultShpDto.Basic> dtoList =
-              entities.stream().map(InferenceResultShpDto.Basic::from).toList();
-
-          flushGroup(key, dtoList);
-        });
+    coreService.buildInferenceData();
   }
 
   /**
-   * 파일명 및 파일 경로
-   *
-   * @param key
-   * @param dtoList
+   * dataUid 단위로 재생성(덮어쓰기) - reset(inference false + geom 전부 false) - geom 엔티티 조회 -> dto 변환 - shp/geojson 생성 - 성공 geo_uid만 true - inference true
    */
-  private void flushGroup(ShpKey key, List<InferenceResultShpDto.Basic> dtoList) {
+  @Transactional
+  public void createShpFile() {
 
-    // TODO 경로 정해지면 수정해야함
+    // TODO 경로는 설정으로 빼는 게 좋음
     String baseDir = System.getProperty("user.home") + "/export";
 
-    // 파일명 stage_input1_input2_mapId
-    String baseName =
-        String.format("%d_%d_%d_%d", key.stage(), key.mapId(), key.input1(), key.input2());
+    int batchSize = 100;     // 한번에 처리할 data_uid 개수
+    int geomLimit = 500000;  // data_uid 당 최대 geom 로딩 수 (메모리/시간 보고 조절)
 
-    String shpBasePath = baseDir + "/shp/" + baseName; // 확장자 없이
-    String geoJsonPath = baseDir + "/geojson/" + baseName + ".geojson";
+    List<Long> dataUids = coreService.findPendingDataUids(batchSize);
 
-    // shp: .shp/.shx/.dbf 생성
-    shpWriter.writeShp(shpBasePath, dtoList);
+    for (Long dataUid : dataUids) {
 
-    // geojson: .geojson 생성
-    shpWriter.writeGeoJson(geoJsonPath, dtoList);
+      // 1) 덮어쓰기 시작: 리셋
+      coreService.resetForRegenerate(dataUid);
+
+      // 2) 생성 대상 조회(엔티티)
+      List<MapSheetAnalDataInferenceGeomEntity> entities =
+        coreService.loadGeomEntities(dataUid, geomLimit);
+
+      if (entities.isEmpty()) {
+        // 실패 상태(false 유지) -> 다음 배치에서 다시 덮어쓰기로 시도
+        continue;
+      }
+
+      // 3) 엔티티 -> DTO
+      List<InferenceResultShpDto.Basic> dtoList =
+        entities.stream().map(InferenceResultShpDto.Basic::from).toList();
+
+      // 4) 파일명: stage_mapSheet_compare_target (첫 row 기준)
+      MapSheetAnalDataInferenceGeomEntity first = entities.get(0);
+      String baseName =
+        String.format(
+          "%d_%d_%d_%d",
+          first.getStage(),
+          first.getMapSheetNum(),
+          first.getCompareYyyy(),
+          first.getTargetYyyy());
+
+      String shpBasePath = baseDir + "/shp/" + baseName;          // 확장자 없이
+      String geoJsonPath = baseDir + "/geojson/" + baseName + ".geojson";
+
+      // 5) 파일 생성 (예외 발생 시 성공 마킹 안 됨 -> 다음에 덮어쓰기 재시도)
+      shpWriter.writeShp(shpBasePath, dtoList);
+      shpWriter.writeGeoJson(geoJsonPath, dtoList);
+
+      // 6) 성공 마킹: geo_uid만 true
+      List<Long> geoUids = entities.stream().map(MapSheetAnalDataInferenceGeomEntity::getGeoUid).toList();
+      coreService.markSuccess(dataUid, geoUids);
+    }
   }
 }