Browse Source

根据模型id查版本返回list

pull/7/head
西大锐 1 year ago
parent
commit
14ea5ad7a2
3 changed files with 6 additions and 7 deletions
  1. +1
    -1
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java
  2. +1
    -3
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/DatasetVersionServiceImpl.java
  3. +4
    -3
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java

+ 1
- 1
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java View File

@@ -75,7 +75,7 @@ public interface ModelsService {
Map uploadModelsPipeline(ModelsVersion modelsVersion) throws Exception;

ResponseEntity<InputStreamResource> downloadAllModelFiles(Integer modelsId, String version);
Map getModelVersions(Integer modelId) throws Exception;
List<ModelsVersion> getModelVersions(Integer modelId) throws Exception;

String insertModelAndVersion(ModelsVo modelsVo) throws Exception;



+ 1
- 3
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/DatasetVersionServiceImpl.java View File

@@ -150,9 +150,6 @@ public class DatasetVersionServiceImpl implements DatasetVersionService {
}

@Override



public void checkDeclaredVersion(DatasetVersion insert) throws Exception {
DatasetVersion datasetVersion = datasetVersionDao.queryByDatasetVersion(insert);
if (datasetVersion != null) {
@@ -180,6 +177,7 @@ public class DatasetVersionServiceImpl implements DatasetVersionService {
return "新增数据集版本失败: " + e.getMessage();
}


}

private void insertPrepare(DatasetVersion datasetVersion) throws Exception {


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

@@ -353,15 +353,16 @@ public class ModelsServiceImpl implements ModelsService {
}

@Override
public Map<String, List<ModelsVersion>> getModelVersions(Integer modelId) throws Exception {
public List<ModelsVersion> getModelVersions(Integer modelId) throws Exception {
// 获取所有相同模型ID的记录
List<ModelsVersion> versions = modelsVersionDao.queryByModelsId(modelId);

if (versions.isEmpty()) {
throw new Exception("未找到相关模型版本记录");
}
// 将结果按照版本分类
return versions.stream().collect(Collectors.groupingBy(ModelsVersion::getVersion));
// // 将结果按照版本分类
// return versions.stream().collect(Collectors.groupingBy(ModelsVersion::getVersion));
return versions;
}

@Override


Loading…
Cancel
Save