Autoflow
/
autoflow-server-mgmt
23
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[UPDATE] ExperimentsController createExperiment 메서드 로직 리팩토링 및 외부 호출 성공 후 DB 저장 처리 적용

Browse Source
main
bjkim 9 months ago
parent 52d4eca72c
commit cc1cfcc2d9
1 changed files with 13 additions and 49 deletions
Show Stats Download Patch File Download Diff File

62
src/main/java/kr/re/etri/autoflow/controllers/ExperimentsController.java
Unescape View File

@ -125,12 +125,14 @@ public class ExperimentsController {
// .doOnError(e -> log.error("Kubeflow experiment 등록 실패", e));
// }
@Operation(summary = "Experiment 등록 (Kubeflow + MLflow)")
@PostMapping
@Transactional
public Mono<ResponseEntity<ExperimentsEntity>> createExperiment(@RequestBody ExperimentsEntity experiment) {
ExperimentsEntity saved = experimentsService.save(experiment);
// DB에 미리 저장하지 않고, 메모리에 보관
ExperimentsEntity saved = new ExperimentsEntity();
saved.setDisplayName(experiment.getDisplayName());
saved.setDescription(experiment.getDescription());
Map<String, Object> kubeflowPayload = new HashMap<>();
kubeflowPayload.put("display_name", saved.getDisplayName());
@ -146,7 +148,6 @@ public class ExperimentsController {
.retrieve()
.bodyToMono(Map.class)
.flatMap(kubeflowResp -> {
// Kubeflow 응답 처리
if (kubeflowResp.containsKey("experiment_id")) {
saved.setKubeFlowId((String) kubeflowResp.get("experiment_id"));
}
@ -157,18 +158,7 @@ public class ExperimentsController {
.toLocalDateTime()
);
}
if (kubeflowResp.containsKey("last_run_created_at")) {
saved.setKubeflowLastRunCreatedAt(
Instant.parse((String) kubeflowResp.get("last_run_created_at"))
.atZone(ZoneId.of("Asia/Seoul"))
.toLocalDateTime()
);
}
if (kubeflowResp.containsKey("storage_state")) {
saved.setKubeflowStorageState((String) kubeflowResp.get("storage_state"));
}
experimentsService.save(saved);
log.info("Kubeflow experiment 등록 완료: {}", kubeflowResp);
// 2️⃣ MLflow 등록
@ -188,56 +178,30 @@ public class ExperimentsController {
log.info("MLflow experiment 등록 완료: {}", createResp);
String mlflowExpId = (String) createResp.get("experiment_id");
// 3️⃣ MLflow 조회
return webClientBuilder
.build()
return webClientBuilder.build()
.get()
.uri(mlflowBaseUrl + "/ajax-api/2.0/mlflow/experiments/get?experiment_id=" + mlflowExpId)
.headers(headers -> headers.setBasicAuth(mlflowUser, mlflowPassword))
.retrieve()
.bodyToMono(Map.class)
.flatMap(getResp -> {
log.info("MLflow experiment 상세 조회 완료: {}", getResp);
// 필요한 필드를 entity에 반영
.map(getResp -> {
if (getResp.containsKey("experiment")) {
Map<String, Object> exp = (Map<String, Object>) getResp.get("experiment");
saved.setMlFlowId((String) exp.get("experiment_id"));
saved.setMlflow_artifactLocation((String) exp.get("artifact_location"));
saved.setMlflowLifecycleStage((String) exp.get("lifecycle_stage"));
}
// MLflow 응답 필드 반영
if (exp.containsKey("experiment_id")) {
saved.setMlFlowId((String) exp.get("experiment_id"));
}
// DB 저장은 외부 호출이 모두 성공한 이후
ExperimentsEntity finalSaved = experimentsService.save(saved);
if (exp.containsKey("artifact_location")) {
saved.setMlflow_artifactLocation((String) exp.get("artifact_location"));
}
if (exp.containsKey("lifecycle_stage")) {
saved.setMlflowLifecycleStage((String) exp.get("lifecycle_stage"));
}
if (exp.containsKey("created_at")) {
// created_at은 timestamp(ms)로 올 수도 있으므로 Instant 변환
Object createdAtObj = exp.get("created_at");
Instant createdAtInstant = null;
if (createdAtObj instanceof Number) {
createdAtInstant = Instant.ofEpochMilli(((Number) createdAtObj).longValue());
} else if (createdAtObj instanceof String) {
createdAtInstant = Instant.parse((String) createdAtObj);
}
if (createdAtInstant != null) {
saved.setMlflowCreatedAt(
createdAtInstant.atZone(ZoneId.of("Asia/Seoul")).toLocalDateTime()
);
}
}
}
experimentsService.save(saved);
return Mono.just(ResponseEntity.ok(saved));
return ResponseEntity.ok(finalSaved);
});
});
})
.doOnError(e -> log.error("Experiment 등록 실패", e));
}
@Operation(summary = "Experiment 수정")
@PutMapping("/{id}")
public ResponseEntity<ExperimentsEntity> updateExperiment(

Write Preview
Loading…
Cancel
Save