shp파일 생성 수정

This commit is contained in:
2026-01-23 14:08:05 +09:00
parent 3f4cf070cd
commit 406009e700
5 changed files with 30 additions and 20 deletions

View File

@@ -1,12 +1,11 @@
package com.kamco.cd.kamcoback.inference.service;
import com.kamco.cd.kamcoback.inference.dto.InferenceResultDto.InferenceLearnDto;
import com.kamco.cd.kamcoback.inference.dto.InferenceResultShpDto;
import com.kamco.cd.kamcoback.inference.dto.InferenceResultShpDto.CreateShpRequest;
import com.kamco.cd.kamcoback.postgres.core.InferenceResultCoreService;
import com.kamco.cd.kamcoback.postgres.core.InferenceResultShpCoreService;
import com.kamco.cd.kamcoback.scheduler.service.ShpPipelineService;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
@@ -39,18 +38,20 @@ public class InferenceResultShpService {
/**
* shp 파일 수동생성
*
* @param uid
* @param req
* @param uuid
*/
public void createShp(String uid, CreateShpRequest req) {
List<Long> batchIds = new ArrayList<>();
batchIds.add(req.getM1BatchId());
batchIds.add(req.getM2BatchId());
batchIds.add(req.getM3BatchId());
public void createShp(UUID uuid) {
String batchId = req.getM1BatchId() + "," + req.getM2BatchId() + "," + req.getM3BatchId();
InferenceLearnDto dto = inferenceResultCoreService.getInferenceUid(uuid);
if (dto == null) {
return;
}
Long m1BatchId = dto.getM1ModelBatchId();
Long m2BatchId = dto.getM2ModelBatchId();
Long m3BatchId = dto.getM3ModelBatchId();
String batchId = m1BatchId + "," + m2BatchId + "," + m3BatchId;
// shp 파일 비동기 생성
shpPipelineService.runPipeline(jarPath, datasetDir, batchId, uid);
shpPipelineService.runPipeline(jarPath, datasetDir, batchId, dto.getUid());
}
}