Browse Source

Merge branch 'dev' of https://gitlink.org.cn/ci4s/ci4sManagement-cloud into dev

dev-active_learn
somunslotus 11 months ago
parent
commit
62d1d1f2f3
2 changed files with 11 additions and 4 deletions
  1. +4
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/RayIns.java
  2. +7
    -4
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/RayInsServiceImpl.java

+ 4
- 0
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/RayIns.java View File

@@ -9,6 +9,7 @@ import lombok.Data;


import java.util.ArrayList; import java.util.ArrayList;
import java.util.Date; import java.util.Date;
import java.util.List;
import java.util.Map; import java.util.Map;


@Data @Data
@@ -45,6 +46,9 @@ public class RayIns {


private Date finishTime; private Date finishTime;


@TableField(exist = false)
private List<Map> fileList;

@TableField(exist = false) @TableField(exist = false)
private ArrayList<Map<String, Object>> trialList; private ArrayList<Map<String, Object>> trialList;
} }


+ 7
- 4
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/RayInsServiceImpl.java View File

@@ -171,7 +171,9 @@ public class RayInsServiceImpl implements RayInsService {
if (Constant.Running.equals(rayIns.getStatus()) || Constant.Pending.equals(rayIns.getStatus())) { if (Constant.Running.equals(rayIns.getStatus()) || Constant.Pending.equals(rayIns.getStatus())) {
rayIns = queryStatusFromArgo(rayIns); rayIns = queryStatusFromArgo(rayIns);
} }
rayIns.setTrialList(getTrialList(rayIns.getResultPath()));
getTrialList(rayIns);

// rayIns.setTrialList(getTrialList(rayIns.getResultPath()));
return rayIns; return rayIns;
} }


@@ -265,12 +267,14 @@ public class RayInsServiceImpl implements RayInsService {
return rayInsDao.queryByRayInsIsNotTerminated(); return rayInsDao.queryByRayInsIsNotTerminated();
} }


public ArrayList<Map<String, Object>> getTrialList(String directoryPath) throws Exception {
public void getTrialList(RayIns rayIns) throws Exception {
// 获取指定路径下的所有文件 // 获取指定路径下的所有文件
String directoryPath = rayIns.getResultPath();
String prefix = directoryPath.substring(directoryPath.indexOf("/") + 1, directoryPath.length()) + "/"; String prefix = directoryPath.substring(directoryPath.indexOf("/") + 1, directoryPath.length()) + "/";
List<Map> maps = minioUtil.listFilesInDirectory(bucketName, prefix); List<Map> maps = minioUtil.listFilesInDirectory(bucketName, prefix);


if (!maps.isEmpty()) { if (!maps.isEmpty()) {
rayIns.setFileList(maps);
List<Map> collect = maps.stream().filter(map -> map.get("name").toString().startsWith("experiment_state")).collect(Collectors.toList()); List<Map> collect = maps.stream().filter(map -> map.get("name").toString().startsWith("experiment_state")).collect(Collectors.toList());
if (!collect.isEmpty()) { if (!collect.isEmpty()) {
Path experimentState = Paths.get(collect.get(0).get("name").toString()); Path experimentState = Paths.get(collect.get(0).get("name").toString());
@@ -298,9 +302,8 @@ public class RayInsServiceImpl implements RayInsService {


trialList.add(trial); trialList.add(trial);
} }
return trialList;
rayIns.setTrialList(trialList);
} }
} }
return null;
} }
} }

Loading…
Cancel
Save