| @@ -75,7 +75,7 @@ public interface ModelsService { | |||||
| Map uploadModelsPipeline(ModelsVersion modelsVersion) throws Exception; | Map uploadModelsPipeline(ModelsVersion modelsVersion) throws Exception; | ||||
| ResponseEntity<InputStreamResource> downloadAllModelFiles(Integer modelsId, String version); | 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; | String insertModelAndVersion(ModelsVo modelsVo) throws Exception; | ||||
| @@ -150,9 +150,6 @@ public class DatasetVersionServiceImpl implements DatasetVersionService { | |||||
| } | } | ||||
| @Override | @Override | ||||
| public void checkDeclaredVersion(DatasetVersion insert) throws Exception { | public void checkDeclaredVersion(DatasetVersion insert) throws Exception { | ||||
| DatasetVersion datasetVersion = datasetVersionDao.queryByDatasetVersion(insert); | DatasetVersion datasetVersion = datasetVersionDao.queryByDatasetVersion(insert); | ||||
| if (datasetVersion != null) { | if (datasetVersion != null) { | ||||
| @@ -180,6 +177,7 @@ public class DatasetVersionServiceImpl implements DatasetVersionService { | |||||
| return "新增数据集版本失败: " + e.getMessage(); | return "新增数据集版本失败: " + e.getMessage(); | ||||
| } | } | ||||
| } | } | ||||
| private void insertPrepare(DatasetVersion datasetVersion) throws Exception { | private void insertPrepare(DatasetVersion datasetVersion) throws Exception { | ||||
| @@ -353,15 +353,16 @@ public class ModelsServiceImpl implements ModelsService { | |||||
| } | } | ||||
| @Override | @Override | ||||
| public Map<String, List<ModelsVersion>> getModelVersions(Integer modelId) throws Exception { | |||||
| public List<ModelsVersion> getModelVersions(Integer modelId) throws Exception { | |||||
| // 获取所有相同模型ID的记录 | // 获取所有相同模型ID的记录 | ||||
| List<ModelsVersion> versions = modelsVersionDao.queryByModelsId(modelId); | List<ModelsVersion> versions = modelsVersionDao.queryByModelsId(modelId); | ||||
| if (versions.isEmpty()) { | if (versions.isEmpty()) { | ||||
| throw new Exception("未找到相关模型版本记录"); | throw new Exception("未找到相关模型版本记录"); | ||||
| } | } | ||||
| // 将结果按照版本分类 | |||||
| return versions.stream().collect(Collectors.groupingBy(ModelsVersion::getVersion)); | |||||
| // // 将结果按照版本分类 | |||||
| // return versions.stream().collect(Collectors.groupingBy(ModelsVersion::getVersion)); | |||||
| return versions; | |||||
| } | } | ||||
| @Override | @Override | ||||