diff --git a/src/main/java/kr/re/etri/autoflow/controllers/KubeflowRunsNewController.java b/src/main/java/kr/re/etri/autoflow/controllers/KubeflowRunsNewController.java index 1ecaa25..adb62a6 100644 --- a/src/main/java/kr/re/etri/autoflow/controllers/KubeflowRunsNewController.java +++ b/src/main/java/kr/re/etri/autoflow/controllers/KubeflowRunsNewController.java @@ -38,7 +38,7 @@ public class KubeflowRunsNewController { @Parameter(description = "Kubeflow Run ID", example = "ad980d7f-050a-4c59-a775-94394befad40") @PathVariable("runId") String runId) { - return runRepository.findById(runId) + return runRepository.findByRunId(runId) .map(ResponseEntity::ok) .orElse(ResponseEntity.notFound().build()); } diff --git a/src/main/java/kr/re/etri/autoflow/repository/KubeflowRunRepository.java b/src/main/java/kr/re/etri/autoflow/repository/KubeflowRunRepository.java index a03597c..91cc831 100644 --- a/src/main/java/kr/re/etri/autoflow/repository/KubeflowRunRepository.java +++ b/src/main/java/kr/re/etri/autoflow/repository/KubeflowRunRepository.java @@ -4,6 +4,9 @@ import kr.re.etri.autoflow.entity.KubeflowRunEntity; import org.springframework.data.jpa.repository.JpaRepository; import org.springframework.data.jpa.repository.JpaSpecificationExecutor; +import java.util.Optional; + public interface KubeflowRunRepository extends JpaRepository, JpaSpecificationExecutor { boolean existsByRunId(String runId); + Optional findByRunId(String runId); }