修改模型dvc接口

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/dataset/NewDatasetFromGitController.java

@@ -138,7 +138,7 @@ public class NewDatasetFromGitController {
      * @return 单条数据
      */
 
-    @GetMapping("/downloadSinggerFile")
+    @GetMapping("/downloadSingleFile")
     @ApiOperation(value = "下载单个数据集文件", notes = "根据数据集版本表id下载单个数据集文件")
     public ResponseEntity<InputStreamResource> downloadDataset(@RequestParam("url") String url) throws Exception {
         return newDatasetService.downloadDatasetlocal(url);

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/model/NewModelFromGitController.java

@@ -12,6 +12,8 @@ import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.annotation.Resource;
+import java.util.List;
+import java.util.Map;
 
 @RestController
 @RequestMapping("newmodel")
@@ -27,6 +29,12 @@ public class NewModelFromGitController {
         return AjaxResult.success(this.modelsService.newCreateModel(modelsVo));
     }
 
+    @PostMapping("/addVersion")
+    @ApiOperation("添加版本")
+    public AjaxResult addVersion(@RequestBody ModelsVo modelsVo) throws Exception {
+        return AjaxResult.success(this.modelsService.newCreateVersion(modelsVo));
+    }
+
     @CrossOrigin(origins = "*", allowedHeaders = "*")
     @PostMapping("/upload")
     @ApiOperation(value = "上传模型", notes = "根据模型id上传模型文件，并将信息存入数据库。")
@@ -34,31 +42,16 @@ public class NewModelFromGitController {
         return AjaxResult.success(this.modelsService.uploadModelLocal(files, uuid));
     }
 
-    /**
-     * 模型打包下载
-     *
-     * @param version 模型版本
-     * @return 模型
-     */
     @GetMapping("/downloadAllFiles")
     @ApiOperation(value = "下载同一版本下所有模型，并打包")
-    public ResponseEntity<InputStreamResource> downloadAllDatasetFiles(@RequestParam("identifier") String identifier,
-                                                                       @RequestParam(value = "version") String version) throws Exception {
-        return modelsService.downloadAllModelFilesNew(identifier, version);
+    public ResponseEntity<InputStreamResource> downloadAllDatasetFiles(@RequestParam("name") String name,  @RequestParam("id") Integer id, @RequestParam("version") String version) throws Exception {
+        return modelsService.downloadAllModelFilesNew(name, id, version);
     }
 
-
-    /**
-     * 下载模型
-     *
-     * @param model_version_id ps:这里的id是model_version表的主键
-     * @return 模型
-     */
-
-    @GetMapping("/download/{model_version_id}")
+    @GetMapping("/downloadSingleFile")
     @ApiOperation(value = "下载单个模型文件", notes = "根据模型版本表id下载单个模型文件")
-    public ResponseEntity<InputStreamResource> downloadModel(@PathVariable("model_version_id") Integer model_version_id) throws Exception {
-        return modelsService.downloadModels(model_version_id);
+    public ResponseEntity<InputStreamResource> downloadDatasetlocal(@RequestParam("url") String url) throws Exception {
+        return modelsService.downloadDatasetlocal(url);
     }
 
     @GetMapping("/queryModels")
@@ -83,34 +76,38 @@ public class NewModelFromGitController {
 
     @GetMapping("/getVersionList")
     @ApiOperation(value = "获取模型分支列表")
-    public AjaxResult getVersionList(@RequestParam("identifier") String identifier, @RequestParam("owner")String owner) throws Exception {
+    public AjaxResult getVersionList(@RequestParam("identifier") String identifier, @RequestParam("owner") String owner) throws Exception {
         return AjaxResult.success(this.modelsService.getVersionList(identifier, owner));
     }
 
 
     @GetMapping("/getModelDetail")
     @ApiOperation(value = "获取模型详细信息")
-    public AjaxResult getModelDetail(@RequestParam("model_name") String modelName, @RequestParam("identifier") String identifier,@RequestParam("owner")  String owner, @RequestParam("version") String version) throws Exception {
-        return AjaxResult.success(this.modelsService.getModelDetail(modelName, identifier, owner, version));
+    public AjaxResult getModelDetail(@RequestParam("id") Integer id, @RequestParam("name") String name, @RequestParam("identifier") String identifier, @RequestParam("owner") String owner, @RequestParam("version") String version) throws Exception {
+        return AjaxResult.success(this.modelsService.getModelDetail(id, name, identifier, owner, version));
     }
 
     @GetMapping("/getModelDependencyTree")
     @ApiOperation(value = "获取模型依赖关系树")
-    public AjaxResult getModelDependencyTree(@RequestParam("repoId") Integer repoId, @RequestParam("model_name") String modelName, @RequestParam("version") String version) throws Exception {
-        return AjaxResult.success(this.modelsService.getModelDependencyTree(repoId, modelName, version));
+    public AjaxResult getModelDependencyTree(@RequestParam("id") Integer id, @RequestParam("model_name") String modelName, @RequestParam("version") String version) throws Exception {
+        return AjaxResult.success(this.modelsService.getModelDependencyTree(id, modelName, version));
     }
 
     @DeleteMapping("/delete")
     @ApiOperation(value = "删除模型")
-    public AjaxResult deleteModel(@RequestParam("repoId")Integer repoId, @RequestParam("identifier") String identifier, @RequestParam("owner") String owner) throws Exception {
-        this.modelsService.deleteModel(repoId,identifier, owner);
+    public AjaxResult deleteModel(@RequestParam("id") Integer id, @RequestParam("identifier") String identifier, @RequestParam("owner") String owner) throws Exception {
+        this.modelsService.deleteModel(id, identifier, owner);
         return AjaxResult.success();
     }
 
     @DeleteMapping("/deleteVersion")
     @ApiOperation(value = "删除模型版本")
-    public AjaxResult deleteVersion(@RequestParam("repoId")Integer repoId, @RequestParam("identifier") String identifier, @RequestParam("owner") String owner, @RequestParam("version") String version) throws Exception {
-        this.modelsService.deleteVersion(repoId, identifier, owner, version);
+    public AjaxResult deleteVersion(@RequestParam("id") Integer id, @RequestParam("identifier") String identifier, @RequestParam("owner") String owner, @RequestParam("version") String version) throws Exception {
+        List<Map<String, Object>> versionList = this.modelsService.getVersionList(identifier, owner);
+        if(versionList.size() == 1){
+            return AjaxResult.error("当前模型只有一个版本，不能删除该版本");
+        }
+        this.modelsService.deleteVersion(id, identifier, owner, version);
         return AjaxResult.success();
     }
 }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/domain/ModelDependency1.java

@@ -24,4 +24,6 @@ public class ModelDependency1 implements Serializable {
     private String owner;
 
     private Integer state;
+
+    private String meta;
 }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/mapper/ModelDependency1Dao.java

@@ -9,13 +9,13 @@ public interface ModelDependency1Dao {
 
     int insert(ModelDependency1 modelDependency1);
 
-    List<ModelDependency1> queryByModelName(@Param("modelName") String modelName);
+    List<ModelDependency1> queryModelDependency(@Param("modelName") String modelName, @Param("repoId") Integer repoId, @Param("owner") String owner);
 
     ModelDependency1 queryByModelNameAndVersion(@Param("repoId") Integer repoId, @Param("modelName") String modelName, @Param("version") String version);
 
     List<ModelDependency1> queryByParentModel(@Param("parentModel") String parentModel);
 
-    int deleteModel(@Param("repoId") Integer repoId, @Param("identifier")String identifier, @Param("owner") String owner, @Param("version") String version);
+    int deleteModel(@Param("repoId") Integer repoId, @Param("identifier") String identifier, @Param("owner") String owner, @Param("version") String version);
 
-    int deleteModelDependency(@Param("repoId") Integer repoId, @Param("identifier")String identifier, @Param("owner") String owner, @Param("version") String version);
+    int deleteModelDependency(@Param("repoId") Integer repoId, @Param("identifier") String identifier, @Param("owner") String owner, @Param("version") String version);
 }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java

@@ -70,6 +70,7 @@ public interface ModelsService {
 
     ResponseEntity<InputStreamResource> downloadModels(Integer id) throws Exception;
 
+    ResponseEntity<InputStreamResource> downloadDatasetlocal(String filePath) throws Exception;
 
     List<Map<String, String>> uploadModels(MultipartFile[] files, String uuid) throws Exception;
 
@@ -87,11 +88,13 @@ public interface ModelsService {
 
     List<Map<String, String>> exportModels(String path, String uuid) throws Exception;
 
-    CompletableFuture<String> newCreateModel(ModelsVo modelsVo) throws Exception;
+    String newCreateModel(ModelsVo modelsVo) throws Exception;
+
+    String newCreateVersion(ModelsVo modelsVo) throws Exception;
 
     List<Map<String, String>> uploadModelLocal(MultipartFile[] files, String uuid) throws Exception;
 
-    ResponseEntity<InputStreamResource> downloadAllModelFilesNew(String identifier, String version) throws IOException, Exception;
+    ResponseEntity<InputStreamResource> downloadAllModelFilesNew(String identifier, Integer id, String version) throws IOException, Exception;
 
     Page<ModelsVo> newPubilcQueryByPage(ModelsVo modelsVo, PageRequest pageRequest) throws Exception;
 
@@ -99,7 +102,7 @@ public interface ModelsService {
 
     List<Map<String, Object>> getVersionList(String identifier, String owner) throws Exception;
 
-    ModelsVo getModelDetail(String modelName, String identifier, String owner, String version) throws Exception;
+    ModelsVo getModelDetail(Integer id, String name, String identifier, String owner, String version) throws Exception;
 
     ModelDependency1TreeVo getModelDependencyTree(Integer repoId, String modelName, String version) throws Exception;
 

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java

@@ -41,7 +41,6 @@ import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.*;
-import java.util.concurrent.CompletableFuture;
 import java.util.stream.Collectors;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipOutputStream;
@@ -249,6 +248,29 @@ public class ModelsServiceImpl implements ModelsService {
         }
     }
 
+    @Override
+    public ResponseEntity<InputStreamResource> downloadDatasetlocal(String filePath) throws Exception {
+        File file = new File(filePath);
+
+        if (!file.exists()) {
+            throw new FileNotFoundException("File not found: " + filePath);
+        }
+
+        InputStreamResource resource = new InputStreamResource(new FileInputStream(file));
+
+        HttpHeaders headers = new HttpHeaders();
+        headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + file.getName());
+        headers.add(HttpHeaders.CONTENT_LENGTH, String.valueOf(file.length()));
+        headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_OCTET_STREAM_VALUE);
+
+        return ResponseEntity.ok()
+                .headers(headers)
+                .contentLength(file.length())
+                .contentType(MediaType.APPLICATION_OCTET_STREAM)
+                .body(resource);
+    }
+
+
     /**
      * 上传模型文件
      *
@@ -389,7 +411,6 @@ public class ModelsServiceImpl implements ModelsService {
         Models models = new Models();
         models.setName(modelsVo.getName());
         models.setDescription(modelsVo.getDescription());
-        models.setAvailableRange(modelsVo.getAvailableRange());
         models.setModelType(modelsVo.getModelType());
         models.setModelTag(modelsVo.getModelTag());
         Models modelsInsert = this.insert(models);
@@ -516,64 +537,206 @@ public class ModelsServiceImpl implements ModelsService {
     }
 
     @Override
-    public CompletableFuture<String> newCreateModel(ModelsVo modelsVo) {
-        return CompletableFuture.supplyAsync(() -> {
-            try {
-                LoginUser loginUser = SecurityUtils.getLoginUser();
-                String ci4sUsername = loginUser.getUsername();
-                String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
-                String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
+    public String newCreateModel(ModelsVo modelsVo) {
+        try {
+            LoginUser loginUser = SecurityUtils.getLoginUser();
+            String ci4sUsername = loginUser.getUsername();
+            String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
+            String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
+
+//            String ci4sUsername = "chenzhihang";
+//            String gitLinkUsername = "chenzhihang";
+//            String gitLinkPassword = "czh19970311";
+
+            Map<String, Object> userInfo = getUserInfo(ci4sUsername, gitLinkUsername, gitLinkPassword);
+            Integer userId = (Integer) userInfo.get("user_id");
+
+            String repositoryName = modelsVo.getIdentifier() == null ? ci4sUsername + "_model_" + DateUtils.dateTimeNow() : modelsVo.getIdentifier();
+            ModelDependency1 modelDependency = new ModelDependency1();
+            List<ModelDependency1> oldModelDependencys = modelDependency1Dao.queryModelDependency(modelsVo.getName(), null, gitLinkUsername);
+            if (oldModelDependencys != null && !oldModelDependencys.isEmpty()) {
+                if (oldModelDependencys.stream().anyMatch(oldModelDependency -> oldModelDependency.getVersion().equals(modelsVo.getVersion()))) {
+                    throw new RuntimeException("模型版本已存在，请勿重复创建");
+                }
+            }
 
-                Map<String, Object> userInfo = getUserInfo(ci4sUsername);
-                Integer userId = (Integer) userInfo.get("user_id");
+            //新建模型
+            // 拼接project
+            GitProjectVo gitProjectVo = new GitProjectVo();
+            gitProjectVo.setRepositoryName(repositoryName);
+            gitProjectVo.setName(modelsVo.getName());
+            gitProjectVo.setDescription(modelsVo.getDescription());
+            gitProjectVo.setPrivate(!modelsVo.getIsPublic());
+            gitProjectVo.setUserId(userId);
+
+            // 创建项目
+            Map project = gitService.createProject(gitProjectVo);
+            Integer gitlinIid = (Integer) project.get("id");
+            if (gitlinIid == null) {
+                throw new Exception("创建模型失败:" + project.get("message"));
+            }
+            // 创建分支
+            gitService.createBranch((String) userInfo.get("login"), repositoryName, modelsVo.getVersion(), "master");
+            // 定义标签 标签1：ci4s_model 标签2：ModelTag 标签3：ModelType
+            gitService.createTopic(gitlinIid, "ci4s_model");
+            gitService.createTopic(gitlinIid, "modeltag_" + modelsVo.getModelTag());
+            gitService.createTopic(gitlinIid, "modeltype_" + modelsVo.getModelType());
+
+            String branchName = modelsVo.getVersion();
+            String owner = (String) userInfo.get("login");
+
+            String projectUrl = gitendpoint + "/" + owner + "/" + repositoryName + ".git";
+            String sourcePath = modelsVo.getModelVersionVos().get(0).getUrl();
+            String relatePath = ci4sUsername + "/model/" + gitlinIid + "/" + modelsVo.getName();
+            String rootPath = localPath + relatePath;
+            String modelPath = rootPath + "/model";
+            String metaPath = rootPath + "/metadata";
 
-                String repositoryName = modelsVo.getRepositoryName() == null ? ci4sUsername + "_model_" + DateUtils.dateTimeNow() : modelsVo.getRepositoryName();
-                ModelDependency1 modelDependency = new ModelDependency1();
-                List<ModelDependency1> oldModelDependencys = modelDependency1Dao.queryByModelName(modelsVo.getName());
+            DVCUtils.gitClone(rootPath, projectUrl, branchName, gitLinkUsername, gitLinkPassword);
+            DVCUtils.moveFiles(sourcePath, modelPath);
+
+            //拼接生产的元数据后写入yaml文件
+            ModelMetaVo modelMetaVo = new ModelMetaVo();
+            BeanUtils.copyProperties(modelsVo, modelMetaVo);
+            modelMetaVo.setId(gitlinIid);
+            modelMetaVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login")));
+            modelMetaVo.setUpdateTime(DateUtils.getTime());
+            modelMetaVo.setUsage("<pre><code>" +
+                    "# 克隆模型配置文件与存储参数到本地\n" +
+                    "git clone -b " + branchName + " " + projectUrl + "\n" +
+                    "# 远程拉取配置文件\n" +
+                    "dvc pull\n" +
+                    "</code></pre>");
+            modelMetaVo.setIdentifier(repositoryName);
+            modelMetaVo.setOwner(owner);
+            modelMetaVo.setVersionDesc(modelMetaVo.getDescription());
+            modelMetaVo.setRelativePaths(relatePath);
+
+            Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo);
+            YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
+            String meta = metaMap.toString();
+
+            DVCUtils.dvcInit(rootPath);
+            // 配置远程S3地址
+            String s3Path = "management-platform-files/" + ci4sUsername + "/model/" + gitlinIid + "/" + repositoryName + "/" + branchName;
+            DVCUtils.dvcRemoteAdd(rootPath, s3Path);
+            DVCUtils.dvcConfigS3Credentials(rootPath, endpoint);
+            DVCUtils.dvcConfigS3Credentials2(rootPath, accessKeyId);
+            DVCUtils.dvcConfigS3Credentials3(rootPath, secretAccessKey);
+
+            // dvc 跟踪
+            DVCUtils.dvcAdd(rootPath, "model");
+            // git commit
+            DVCUtils.gitAdd(rootPath, ".");
+            DVCUtils.gitCommit(rootPath, "commit from ci4s with " + ci4sUsername);
+            DVCUtils.gitPush(rootPath, gitLinkUsername, gitLinkPassword);
+            DVCUtils.dvcPush(rootPath);
+
+            //保存模型依赖
+            modelDependency.setRepoId(gitlinIid);
+            modelDependency.setIdentifier(repositoryName);
+            modelDependency.setModelName(modelsVo.getName());
+            modelDependency.setVersion(modelsVo.getVersion());
+            modelDependency.setParentModel(modelsVo.getParentModel());
+            modelDependency.setOwner(gitLinkUsername);
+            modelDependency.setMeta(meta);
+            modelDependency1Dao.insert(modelDependency);
+            return "新增模型成功";
+        } catch (Exception e) {
+            logger.error(e.getMessage());
+            throw new RuntimeException(e);
+        }
+    }
 
-                if (oldModelDependencys != null && !oldModelDependencys.isEmpty()) {
+    @Override
+    public String newCreateVersion(ModelsVo modelsVo) throws Exception {
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        String ci4sUsername = loginUser.getUsername();
+        String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
+        String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
 
-                    if (oldModelDependencys.stream().anyMatch(oldModelDependency -> oldModelDependency.getRepoId().equals(modelsVo.getRepoId()) && oldModelDependency.getVersion().equals(modelsVo.getVersion()))) {
-                        throw new RuntimeException("模型版本已存在，请勿重复创建");
-                    }
+//        String ci4sUsername = "chenzhihang";
+//        String gitLinkUsername = "chenzhihang";
+//        String gitLinkPassword = "czh19970311";
 
-                    //新建版本
-                    commonDvc(userInfo, ci4sUsername, modelsVo.getRepoId(), modelsVo.getRepositoryName(), modelsVo, gitLinkUsername, gitLinkPassword, "CreateModelFromPipeline");
-                    modelDependency.setRepoId(modelsVo.getRepoId());
-                } else {
-                    //新建模型
-                    // 拼接project
-                    GitProjectVo gitProjectVo = new GitProjectVo();
-                    gitProjectVo.setRepositoryName(repositoryName);
-                    gitProjectVo.setName(modelsVo.getName());
-                    gitProjectVo.setDescription(modelsVo.getDescription());
-                    gitProjectVo.setPrivate(modelsVo.getAvailableRange() == 0);
-                    gitProjectVo.setUserId(userId);
-
-                    // 创建项目
-                    Map project = gitService.createProject(gitProjectVo);
-                    Integer gitlinIid = (Integer) project.get("id");
-                    modelDependency.setRepoId(gitlinIid);
-                    // 定义标签 标签1：ci4s_model 标签2：ModelTag 标签3：ModelType
-                    gitService.createTopic(gitlinIid, "ci4s_model");
-                    gitService.createTopic(gitlinIid, "modeltag_" + modelsVo.getModelTag());
-                    gitService.createTopic(gitlinIid, "modeltype_" + modelsVo.getModelType());
-                    commonDvc(userInfo, ci4sUsername, gitlinIid, repositoryName, modelsVo, gitLinkUsername, gitLinkPassword, "createModel");
-                }
+        Map<String, Object> userInfo = getUserInfo(ci4sUsername, gitLinkUsername, gitLinkPassword);
 
-                //保存模型依赖
-                modelDependency.setIdentifier(repositoryName);
-                modelDependency.setModelName(modelsVo.getName());
-                modelDependency.setVersion(modelsVo.getVersion());
-                modelDependency.setParentModel(modelsVo.getParentModel());
-                modelDependency.setOwner(ci4sUsername);
-                modelDependency1Dao.insert(modelDependency);
-                return "新增模型成功";
-            } catch (Exception e) {
-                logger.error(e.getMessage());
-                throw new RuntimeException(e);
+        String repositoryName = modelsVo.getIdentifier() == null ? ci4sUsername + "_model_" + DateUtils.dateTimeNow() : modelsVo.getIdentifier();
+        ModelDependency1 modelDependency = new ModelDependency1();
+        List<ModelDependency1> oldModelDependencys = modelDependency1Dao.queryModelDependency(modelsVo.getName(), modelsVo.getId(), gitLinkUsername);
+        if (oldModelDependencys != null && !oldModelDependencys.isEmpty()) {
+            if (oldModelDependencys.stream().anyMatch(oldModelDependency -> oldModelDependency.getVersion().equals(modelsVo.getVersion()))) {
+                throw new RuntimeException("模型版本已存在，请勿重复创建");
             }
-        });
+
+            //新建版本
+            String branchName = modelsVo.getVersion();
+            String owner = (String) userInfo.get("login");
+
+            String projectUrl = gitendpoint + "/" + owner + "/" + repositoryName + ".git";
+            String sourcePath = modelsVo.getModelVersionVos().get(0).getUrl();
+            String relatePath = ci4sUsername + "/model/" + modelsVo.getId() + "/" + modelsVo.getName();
+            String rootPath = localPath + relatePath;
+            String modelPath = rootPath + "/model";
+            String metaPath = rootPath + "/metadata";
+
+            //部分信息在前面的版本里面，从那边取过来
+            Map<String, Object> stringObjectMap = YamlUtils.loadYamlFile(metaPath + "/" + "metadata.yaml");
+            ModelMetaVo oldModelVo = ConvertUtil.convertMapToObject(stringObjectMap, ModelMetaVo.class);
+
+            // 创建本地分支
+            DVCUtils.createLocalBranchBasedOnMaster(rootPath, branchName);
+            //dvc checkout
+            DVCUtils.dvcCheckout(rootPath);
+            DVCUtils.moveFiles(sourcePath, modelPath);
+
+            //拼接生产的元数据后写入yaml文件
+            ModelMetaVo modelMetaVo = new ModelMetaVo();
+            BeanUtils.copyProperties(modelsVo, modelMetaVo);
+            modelMetaVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login")));
+            modelMetaVo.setUpdateTime(DateUtils.getTime());
+            modelMetaVo.setUsage("<pre><code>" +
+                    "# 克隆模型配置文件与存储参数到本地\n" +
+                    "git clone -b " + branchName + " " + projectUrl + "\n" +
+                    "# 远程拉取配置文件\n" +
+                    "dvc pull\n" +
+                    "</code></pre>");
+            modelMetaVo.setIdentifier(repositoryName);
+            modelMetaVo.setOwner(owner);
+            modelMetaVo.setDescription(oldModelVo.getDescription());
+            modelMetaVo.setModelTag(oldModelVo.getModelTag());
+            modelMetaVo.setModelType(oldModelVo.getModelType());
+            modelMetaVo.setRelativePaths(relatePath);
+
+            Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo);
+            YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
+            String meta = metaMap.toString();
+
+            // 配置远程S3地址
+            String s3Path = "management-platform-files/" + ci4sUsername + "/model/" + modelsVo.getId() + "/" + repositoryName + "/" + branchName;
+            DVCUtils.dvcRemoteAdd(rootPath, s3Path);
+            DVCUtils.dvcConfigS3Credentials(rootPath, endpoint);
+            DVCUtils.dvcConfigS3Credentials2(rootPath, accessKeyId);
+            DVCUtils.dvcConfigS3Credentials3(rootPath, secretAccessKey);
+            // dvc 跟踪
+            DVCUtils.dvcAdd(rootPath, "model");
+            DVCUtils.pushNewBranchToRemote(rootPath, gitLinkUsername, gitLinkPassword, branchName);
+            //dvc push 到远程S3
+            DVCUtils.dvcPush(rootPath);
+
+            //保存模型依赖
+            modelDependency.setRepoId(modelsVo.getId());
+            modelDependency.setIdentifier(repositoryName);
+            modelDependency.setModelName(modelsVo.getName());
+            modelDependency.setVersion(modelsVo.getVersion());
+            modelDependency.setParentModel(modelsVo.getParentModel());
+            modelDependency.setOwner(gitLinkUsername);
+            modelDependency.setMeta(meta);
+            modelDependency1Dao.insert(modelDependency);
+
+            return "新增模型版本成功";
+        }
+        return null;
     }
 
 
@@ -584,8 +747,9 @@ public class ModelsServiceImpl implements ModelsService {
         for (MultipartFile file : files) {
             // 构建objectName
             String username = SecurityUtils.getLoginUser().getUsername();
+//            String username = "chenzhihang";
             String fileName = file.getOriginalFilename();
-            String path = "/temp/" + username + "/models/" + uuid + "/model/";
+            String path = localPath + "/temp/" + username + "/models/" + uuid;
             long sizeInBytes = file.getSize();
             String formattedSize = FileUtil.formatFileSize(sizeInBytes);
             File targetFile = new File(path, file.getOriginalFilename());
@@ -594,10 +758,9 @@ public class ModelsServiceImpl implements ModelsService {
             // 保存文件到目标路径
             FileUtils.copyInputStreamToFile(file.getInputStream(), targetFile);
             // 返回上传文件的路径
-            String absolutePath = targetFile.getAbsolutePath();
             Map<String, String> result = new HashMap<>();
             result.put("fileName", fileName);
-            result.put("url", absolutePath); // objectName根据实际情况定义
+            result.put("url", path); // objectName根据实际情况定义
             result.put("fileSize", formattedSize);
             results.add(result);
         }
@@ -605,27 +768,12 @@ public class ModelsServiceImpl implements ModelsService {
     }
 
     @Override
-    public ResponseEntity<InputStreamResource> downloadAllModelFilesNew(String identifier, String version) throws Exception {
+    public ResponseEntity<InputStreamResource> downloadAllModelFilesNew(String identifier, Integer id, String version) throws Exception {
         // 命令行操作 git clone 项目地址
         LoginUser loginUser = SecurityUtils.getLoginUser();
         String ci4sUsername = loginUser.getUsername();
-        String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
-        String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
-        Map<String, Object> userInfo = getUserInfo(ci4sUsername);
-
-        String projectUrl = gitendpoint + "/" + userInfo.get("login") + "/" + identifier + ".git";
-        String localPath1 = localPath + ci4sUsername + "/model/" + identifier;
-        File folder = new File(localPath1);
-        if (folder.exists() && folder.isDirectory()) {
-            //切换分支
-            DVCUtils.gitCheckoutBranch(localPath1, version);
-            //pull
-            DVCUtils.gitPull(localPath1, gitLinkUsername, gitLinkPassword);
-            //dvc pull
-            DVCUtils.dvcPull(localPath1);
-        } else {
-            DVCUtils.gitClone(localPath1, projectUrl, version, gitLinkUsername, gitLinkPassword);
-        }
+
+        String localPath1 = localPath + ci4sUsername + "/model/" + id + "/" + identifier;
         // 打包 data 文件夹
         String dataFolderPath = localPath1 + "/model";
         String zipFilePath = localPath1 + "/model.zip";
@@ -658,21 +806,33 @@ public class ModelsServiceImpl implements ModelsService {
     }
 
     @Override
-    public Page<ModelsVo> newPubilcQueryByPage(ModelsVo modelsVo, PageRequest pageRequest) throws Exception {
+    public Page<ModelsVo> newPersonalQueryByPage(ModelsVo modelsVo, PageRequest pageRequest) throws Exception {
         LoginUser loginUser = SecurityUtils.getLoginUser();
         String ci4sUsername = loginUser.getUsername();
-        Map<String, Object> userInfo = getUserInfo(ci4sUsername);
+        String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
+        String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
+
+//        String ci4sUsername = "chenzhihang";
+//        String gitLinkUsername = "chenzhihang";
+//        String gitLinkPassword = "czh19970311";
+
+        Map<String, Object> userInfo = getUserInfo(ci4sUsername, gitLinkUsername, gitLinkPassword);
         String token = (String) userInfo.get("token");
 
         //拼接查询url
         String modelTagName = modelsVo.getModelTag();
         String modelTypeName = modelsVo.getModelType();
-        String search = modelsVo.getName();
         String topic_name = "ci4s_model";
         topic_name = StringUtils.isEmpty(modelTagName) ? topic_name : topic_name + ",modeltag_" + modelTypeName;
         topic_name = StringUtils.isEmpty(modelTagName) ? topic_name : topic_name + ",modeltype_" + modelTypeName;
 
-        String url = gitendpoint + "/api/users/" + userInfo.get("login") + "/projects.json?page=" + pageRequest.getPageNumber() + "&limit=" + pageRequest.getPageSize() + "&category=manage&topic_name=" + topic_name + "&search=" + search;
+        String url = gitendpoint + "/api/users/" + userInfo.get("login") + "/projects.json?page=" + pageRequest.getPageNumber() + "&limit=" + pageRequest.getPageSize() + "&category=manage&topic_name=" + topic_name;
+
+        String name = modelsVo.getName();
+        if (StringUtils.isNotEmpty(name)) {
+            url = url + "&search=" + name;
+        }
+
         String req = HttpUtils.sendGetWithToken(url, null, token);
         Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
         Integer total = (Integer) stringObjectMap.get("count");
@@ -681,22 +841,29 @@ public class ModelsServiceImpl implements ModelsService {
     }
 
     @Override
-    public Page<ModelsVo> newPersonalQueryByPage(ModelsVo modelsVo, PageRequest pageRequest) throws Exception {
+    public Page<ModelsVo> newPubilcQueryByPage(ModelsVo modelsVo, PageRequest pageRequest) throws Exception {
         LoginUser loginUser = SecurityUtils.getLoginUser();
         String ci4sUsername = loginUser.getUsername();
-        Map<String, Object> userInfo = getUserInfo(ci4sUsername);
+        String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
+        String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
+
+        Map<String, Object> userInfo = getUserInfo(ci4sUsername, gitLinkUsername, gitLinkPassword);
         Integer userId = (Integer) userInfo.get("user_id");
         String token = (String) userInfo.get("token");
 
         //拼接查询url
         String modelTagName = modelsVo.getModelTag();
         String modelTypeName = modelsVo.getModelType();
-        String search = modelsVo.getName();
         String topic_name = "ci4s_model";
         topic_name = StringUtils.isEmpty(modelTagName) ? topic_name : topic_name + ",modeltag_" + modelTagName;
         topic_name = StringUtils.isEmpty(modelTypeName) ? topic_name : topic_name + ",modeltype_" + modelTypeName;
 
-        String url = gitendpoint + "/api/projects.json?user_id=" + userId + "&page=" + pageRequest.getPageNumber() + "&limit=" + pageRequest.getPageSize() + "&sort_by=praises_count&topic_name=" + topic_name + "&search=" + search;
+        String url = gitendpoint + "/api/projects.json?user_id=" + userId + "&page=" + pageRequest.getPageNumber() + "&limit=" + pageRequest.getPageSize() + "&sort_by=praises_count&topic_name=" + topic_name;
+        String name = modelsVo.getName();
+        if (StringUtils.isNotEmpty(name)) {
+            url = url + "&search=" + name;
+        }
+
         String req = HttpUtils.sendGetWithToken(url, null, token);
         Map<String, Object> stringObjectMap = JacksonUtil.parseJSONStr2Map(req);
         Integer total = (Integer) stringObjectMap.get("total_count");
@@ -705,37 +872,37 @@ public class ModelsServiceImpl implements ModelsService {
     }
 
     @Override
-    public List<Map<String, Object>> getVersionList(String identifier, String owner) throws Exception {
+    public List<Map<String, Object>>   getVersionList(String identifier, String owner) throws Exception {
         List<Map<String, Object>> brancheList = gitService.getBrancheList(owner, identifier);
         return brancheList.stream()
-                .filter(branch -> "master".equals(branch.get("name")))
+                .filter(branch -> !"master".equals(branch.get("name")))
                 .collect(Collectors.toList());
     }
 
     @Override
-    public ModelsVo getModelDetail(String modelName, String identifier, String owner, String version) throws Exception {
+    public ModelsVo getModelDetail(Integer id, String name, String identifier, String owner, String version) throws Exception {
         LoginUser loginUser = SecurityUtils.getLoginUser();
         String ci4sUsername = loginUser.getUsername();
         String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername();
         String gitLinkPassword = loginUser.getSysUser().getGitLinkPassword();
+//        String ci4sUsername = "chenzhihang";
+//        String gitLinkUsername = "chenzhihang";
+//        String gitLinkPassword = "czh19970311";
 
         if (StringUtils.isEmpty(version)) {
-            List<Map<String, Object>> versionList = this.getVersionList(owner, identifier);
+            List<Map<String, Object>> versionList = this.getVersionList(identifier, owner);
             if (versionList.size() == 0) {
-                throw new Exception("数据集文件不存在");
+                throw new Exception("模型文件不存在");
             }
             version = (String) versionList.get(0).get("name");
         }
 
         // git pull操作，然后读取里面的文件列表，列出每个文件的大小和名称，封装成MAP
-        List<Map<String, Object>> fileDetailsAfterGitPull = DVCUtils.getFileDetailsAfterGitPull(localPath + ci4sUsername + "/model/", modelName, version, "model", gitLinkUsername, gitLinkPassword);
+        List<Map<String, Object>> fileDetailsAfterGitPull = DVCUtils.getFileDetailsAfterGitPull(localPath + ci4sUsername + "/model/" + id, name, version, "model", gitLinkUsername, gitLinkPassword);
 
-        Map<String, Object> stringObjectMap = YamlUtils.loadYamlFile(localPath + ci4sUsername + "/model/" + modelName + "/" + version + "/metadata/metadata.yaml");
-        assert stringObjectMap != null;
+        Map<String, Object> stringObjectMap = YamlUtils.loadYamlFile(localPath + ci4sUsername + "/model/" + id + "/" + name + "/metadata/metadata.yaml");
 
-        ModelMetaVo modelMetaVo = ConvertUtil.convertMapToObject(stringObjectMap, ModelMetaVo.class);
-        ModelsVo modelsVo = new ModelsVo();
-        BeanUtils.copyProperties(modelMetaVo, modelsVo);
+        ModelsVo modelsVo = ConvertUtil.convertMapToObject(stringObjectMap, ModelsVo.class);
 
         List<VersionVo> versionVos = new ArrayList<>();
         if (!fileDetailsAfterGitPull.isEmpty()) {
@@ -772,12 +939,6 @@ public class ModelsServiceImpl implements ModelsService {
 
     @Override
     public void deleteModel(Integer repoId, String identifier, String owner) throws Exception {
-        LoginUser loginUser = SecurityUtils.getLoginUser();
-        String ci4sUsername = loginUser.getUsername();
-        Map<String, Object> userInfo = getUserInfo(ci4sUsername);
-        String token = (String) userInfo.get("token");
-
-
         gitService.deleteProject(owner, identifier);
         //删除模型依赖
         modelDependency1Dao.deleteModel(repoId, identifier, owner, null);
@@ -786,72 +947,19 @@ public class ModelsServiceImpl implements ModelsService {
 
     @Override
     public void deleteVersion(Integer repoId, String identifier, String owner, String version) throws Exception {
-        LoginUser loginUser = SecurityUtils.getLoginUser();
-        String ci4sUsername = loginUser.getUsername();
-
         gitService.deleteBranch(owner, identifier, version);
         //删除模型依赖
         modelDependency1Dao.deleteModel(repoId, identifier, owner, version);
         modelDependency1Dao.deleteModelDependency(repoId, identifier, owner, version);
     }
 
-    void commonDvc(Map<String, Object> userInfo, String username, Integer gitlinIid, String repositoryName, ModelsVo modelsVo, String gitLinkUsername, String gitLinkPassword, String type) throws Exception {
-        String branchName = modelsVo.getVersion();
-
-        // 创建分支
-        gitService.createBranch((String) userInfo.get("login"), repositoryName, branchName, "master");
-
-        String projectUrl = gitendpoint + "/" + (String) userInfo.get("login") + "/" + repositoryName + ".git";
-        String url = modelsVo.getModelVersionVos().get(0).getUrl();
-        String sourcePath = url.substring(0, url.lastIndexOf("/"));
-        String rootPath = localPath + username + "/model/" + modelsVo.getName() + "/" + branchName;
-        String modelPath = rootPath + "/model";
-        String metaPath = rootPath + "/metadata";
-
-
-        if (type.equals("CreateModelFromPipeline") && FileUtil.checkDirectoryExists(rootPath)) {
-            DVCUtils.gitFetch(rootPath, gitLinkUsername, gitLinkPassword);
-            DVCUtils.gitCheckoutBranch(rootPath, branchName);
-        } else {
-            DVCUtils.gitClone(rootPath, projectUrl, branchName, gitLinkUsername, gitLinkPassword);
-        }
-
-        DVCUtils.moveFiles(sourcePath, modelPath);
-
-        //拼接生产的元数据后写入yaml文件
-        ModelMetaVo modelMetaVo = new ModelMetaVo();
-        BeanUtils.copyProperties(modelsVo, modelMetaVo);
-        YamlUtils.generateYamlFile(JsonUtils.objectToMap(modelMetaVo), metaPath, "metadata");
-
-        // dvc init 初始化
-        DVCUtils.dvcInit(rootPath);
-
-        // 配置远程S3地址
-        String s3Path = "management-platform-files/" + username + "/model/" + gitlinIid + repositoryName + "/" + branchName;
-        DVCUtils.dvcRemoteAdd(rootPath, s3Path);
-        DVCUtils.dvcConfigS3Credentials(rootPath, endpoint);
-        DVCUtils.dvcConfigS3Credentials2(rootPath, accessKeyId);
-        DVCUtils.dvcConfigS3Credentials3(rootPath, secretAccessKey);
-
-        // dvc 跟踪
-        DVCUtils.dvcAdd(rootPath, "model");
-
-        // git commit
-        DVCUtils.gitAdd(rootPath, ".");
-        DVCUtils.gitCommit(rootPath, "commit from ci4s with " + username);
-        DVCUtils.gitPush(rootPath, gitLinkUsername, gitLinkPassword);
-
-        // dvc push 到远程S3
-        DVCUtils.dvcPush(rootPath);
-    }
-
     public List<ModelsVo> convert(List<Map<String, Object>> lst) {
         if (lst != null && lst.size() > 0) {
             List<ModelsVo> newModelVos = ConvertUtil.convertListMapToObjectList(lst, ModelsVo.class);
 
             for (ModelsVo newModelVo : newModelVos) {
                 Map<String, Object> map = lst.stream()
-                        .filter(m -> m.get("repo_id").equals(newModelVo.getRepoId()))
+                        .filter(m -> m.get("id").equals(newModelVo.getId()))
                         .findFirst()
                         .orElse(null);
 
@@ -869,6 +977,9 @@ public class ModelsServiceImpl implements ModelsService {
                             }
                         });
                     }
+                    Map<String, Object> author = (Map<String, Object>) map.get("author");
+                    newModelVo.setCreateBy((String) author.get("name"));
+                    newModelVo.setOwner((String) author.get("login"));
                 }
             }
             return newModelVos;
@@ -906,9 +1017,13 @@ public class ModelsServiceImpl implements ModelsService {
         modelDependency1TreeVo.setChildModelList(childModelList);
     }
 
-    Map<String, Object> getUserInfo(String ci4sUsername) throws IOException {
+    Map<String, Object> getUserInfo(String ci4sUsername, String gitLinkUsername, String gitLinkPassword) throws IOException {
         Jedis jedis = new Jedis(redisHost);
         String userReq = jedis.get(ci4sUsername + "_gitUserInfo");
+        if (userReq == null) {
+            gitService.login(gitLinkUsername, gitLinkPassword);
+            userReq = jedis.get(ci4sUsername + "_gitUserInfo");
+        }
         Map<String, Object> userInfo = JsonUtils.jsonToMap(userReq);
 
         String token = gitService.checkoutToken();

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java

@@ -94,14 +94,16 @@ public class NewDatasetServiceImpl implements NewDatasetService {
         String projectUrl = gitendpoint + "/" + (String) userInfo.get("login") + "/"+ repositoryName + ".git";
 
         // 得到用户操作的路径
-        String url = datasetVo.getDatasetVersionVos().get(0).getUrl();
-        String localPath = localPathlocal+ ci4sUsername+"/datasets/" + gitlinIid + "/" + datasetVo.getName();
-        String sourcePath = url.substring(0, url.lastIndexOf("/"));
+        String sourcePath = datasetVo.getDatasetVersionVos().get(0).getUrl();
+        String relatePath = ci4sUsername+"/datasets/" + gitlinIid + "/" + datasetVo.getName();
+        String localPath = localPathlocal+ relatePath;
+        String datasetPath = localPath + "/dataset";
+
         // 命令行操作 git clone 项目地址
         DVCUtils.gitClone(localPath, projectUrl, branchName, gitLinkUsername, gitLinkPassword);
         String s3Path = "management-platform-files" + "/" + ci4sUsername + "/" + "datasets" + "/" + gitlinIid + "/" + repositoryName + "/" + branchName;
 
-        DVCUtils.moveFiles(sourcePath, localPath);
+        DVCUtils.moveFiles(sourcePath, datasetPath);
         // 拼接生产的元数据后写入yaml文件
         datasetVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login")));
         datasetVo.setUpdateTime(DateUtils.getTime());
@@ -129,7 +131,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
         DVCUtils.dvcConfigS3Credentials2(localPath, accessKeyId);
         DVCUtils.dvcConfigS3Credentials3(localPath, secretAccessKey);
         // dvc 跟踪
-        DVCUtils.dvcAdd(localPath, "data");
+        DVCUtils.dvcAdd(localPath, "dataset");
         // git commit
         DVCUtils.gitAdd(localPath, ".");
         DVCUtils.gitCommit(localPath, "commit from ci4s with " + loginUser.getUsername());
@@ -151,10 +153,11 @@ public class NewDatasetServiceImpl implements NewDatasetService {
         // 创建分支
         String branchName = StringUtils.isEmpty(datasetVo.getVersion()) ? "master" : datasetVo.getVersion();
         String repositoryName = datasetVo.getIdentifier();
-        String url = datasetVo.getDatasetVersionVos().get(0).getUrl();
-        String localPath = localPathlocal +  ci4sUsername +"/datasets/"+ datasetVo.getId() + "/" + datasetVo.getName();
-        String sourcePath = url.substring(0, url.lastIndexOf("/"));
-        String projectUrl = gitendpoint + "/" + (String) userInfo.get("login") + "/" + repositoryName + ".git";
+        String sourcePath = datasetVo.getDatasetVersionVos().get(0).getUrl();
+        String relatePath = ci4sUsername +"/datasets/"+ datasetVo.getId() + "/" + datasetVo.getName();
+        String localPath = localPathlocal + relatePath;
+        String datasetPath = localPath + "/dataset";
+        String projectUrl = gitendpoint + "/" + userInfo.get("login") + "/" + repositoryName + ".git";
 
         //部分信息在前面的版本里面，从那边取过来
         Map<String, Object> stringObjectMap = YamlUtils.loadYamlFile(localPath + "/" + "dataset.yaml");
@@ -169,7 +172,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
         // 准备数据
         String s3Path = "management-platform-files" + "/" + ci4sUsername + "/" + "datasets" + "/" + datasetVo.getId() + "/" + repositoryName + "/" + branchName;
 
-        DVCUtils.moveFiles(sourcePath, localPath);
+        DVCUtils.moveFiles(sourcePath, datasetPath);
         // 拼接生产的元数据后写入yaml文件
         datasetVo.setCreateBy(String.valueOf(StringUtils.isNotEmpty((String) userInfo.get("nickname")) ? userInfo.get("nickname") : userInfo.get("login")));
         datasetVo.setUpdateTime(DateUtils.getTime());
@@ -180,7 +183,6 @@ public class NewDatasetServiceImpl implements NewDatasetService {
                 "dvc pull\n" +
                 "</code></pre>");
         datasetVo.setIdentifier(repositoryName);
-        datasetVo.setId(datasetVo.getId());
         datasetVo.setOwner((String) userInfo.get("login"));
         if (StringUtils.isEmpty(datasetVo.getDatasetSource())){
             datasetVo.setDatasetSource("用户上传");
@@ -197,7 +199,7 @@ public class NewDatasetServiceImpl implements NewDatasetService {
         DVCUtils.dvcConfigS3Credentials2(localPath, accessKeyId);
         DVCUtils.dvcConfigS3Credentials3(localPath, secretAccessKey);
         // dvc 跟踪
-        DVCUtils.dvcAdd(localPath, "data");
+        DVCUtils.dvcAdd(localPath, "dataset");
         DVCUtils.pushNewBranchToRemote(localPath,gitLinkUsername,gitLinkPassword,branchName);
         //dvc push 到远程S3
         DVCUtils.dvcPush(localPath);
@@ -297,10 +299,6 @@ public class NewDatasetServiceImpl implements NewDatasetService {
 
     @Override
     public List<Map<String, Object>> getVersionList(String repo, String owner) throws Exception {
-        Jedis jedis = new Jedis(redisHost);
-        LoginUser loginUser = SecurityUtils.getLoginUser();
-        String ci4sUsername = loginUser.getUsername();
-        String token = jedis.get(ci4sUsername+"_gitToken");
         List<Map<String, Object>> brancheList = gitService.getBrancheList(owner, repo);
         return brancheList.stream()
                 .filter(branch -> !"master".equals(branch.get("name")))
@@ -333,16 +331,15 @@ public class NewDatasetServiceImpl implements NewDatasetService {
             // 构建objectName
             String username = SecurityUtils.getLoginUser().getUsername();
             String fileName = file.getOriginalFilename();
-            String path = localPathlocal + "temp" + "/" + username + "/" + "datasets" + "/" + uuid + "/" + "data" + "/" + fileName;
+            String path = localPathlocal + "temp" + "/" + username + "/" + "datasets" + "/" + uuid;
             long sizeInBytes = file.getSize();
             String formattedSize = FileUtil.formatFileSize(sizeInBytes);
-            File targetFile = new File(path);
+            File targetFile = new File(path, file.getOriginalFilename());
             // 确保目录存在
             targetFile.getParentFile().mkdirs();
             // 保存文件到目标路径
             FileUtils.copyInputStreamToFile(file.getInputStream(), targetFile);
             // 返回上传文件的路径
-            String absolutePath = targetFile.getAbsolutePath();
             Map<String, String> result = new HashMap<>();
             result.put("fileName", fileName);
             result.put("url", path); // objectName根据实际情况定义

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/TensorBoardServiceImpl.java

@@ -59,9 +59,6 @@ public class TensorBoardServiceImpl implements TensorBoardService {
 
     @Override
     public String runTensorBoard(FrameLogPathVo frameLogPathVo) throws Exception {
-//        if (StringUtils.isEmpty(frameLogPathVo.getPath())||StringUtils.isEmpty(frameLogPathVo.getPvcName())){
-//            throw new Exception("存储路径或存储为空");
-//        }
         if (StringUtils.isEmpty(frameLogPathVo.getPath())) {
             throw new Exception("存储路径为空");
         }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/ConvertUtil.java

@@ -100,14 +100,13 @@ public class ConvertUtil {
 
     private static boolean hasProperty(Class<?> clazz, String propertyName) {
         try {
+            Field field = clazz.getDeclaredField(propertyName);
+            return field != null;
+        } catch (NoSuchFieldException e) {
             Class<?> superClazz = clazz.getSuperclass();
-            if(superClazz != Object.class){
+            if (superClazz != Object.class) {
                 return hasProperty(superClazz, propertyName);
-            } else {
-                Field field = clazz.getDeclaredField(propertyName);
-                return field != null;
             }
-        } catch (NoSuchFieldException e) {
             return false;
         }
     }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/DVCUtils.java

@@ -45,7 +45,7 @@ public class DVCUtils {
 
     public static void moveFiles(String sourcePath, String targetPath) throws Exception {
         Path sourceDir = Paths.get(sourcePath);
-        Path targetDir = Paths.get(targetPath).resolve(sourceDir.getFileName());
+        Path targetDir = Paths.get(targetPath);
 
         if (!Files.exists(targetDir)) {
             Files.createDirectories(targetDir);

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/YamlUtils.java

@@ -16,6 +16,15 @@ public class YamlUtils {
      */
     public static void generateYamlFile(Map<String, Object> data, String path, String fileName) {
         Yaml yaml = new Yaml();
+
+        File directory = new File(path);
+        if (!directory.exists()) {
+            boolean isCreated = directory.mkdirs();
+            if (!isCreated) {
+                throw new RuntimeException("创建路径失败: " + path);
+            }
+        }
+
         String fullPath = path + "/" + fileName + ".yaml";
 
         try (FileWriter writer = new FileWriter(fullPath)) {

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java

@@ -13,6 +13,8 @@ import java.util.List;
 @Data
 @JsonNaming(PropertyNamingStrategy.SnakeCaseStrategy.class)
 public class ModelMetaVo implements Serializable {
+    private Integer id;
+
     @ApiModelProperty(name = "模型名称")
     private String name;
 
@@ -20,13 +22,13 @@ public class ModelMetaVo implements Serializable {
     private String version;
 
     @ApiModelProperty(name = "版本描述")
-    private String versionDescription;
+    private String versionDesc;
 
     @ApiModelProperty(value = "创建人")
     private String createBy;
 
     @ApiModelProperty(value = "更新时间")
-    private Date updateTime;
+    private String updateTime;
 
     @ApiModelProperty(value = "训练镜像")
     private String image;
@@ -62,6 +64,21 @@ public class ModelMetaVo implements Serializable {
     private String description;
 
     @ApiModelProperty(value = "示例用法")
-    private String examples;
+    private String usage;
+
+    @ApiModelProperty(name = "time_ago", value = "更新时间")
+    private String timeAgo;
+
+    @ApiModelProperty(name = "owner", value = "项目所有者")
+    private String owner;
 
+    @ApiModelProperty(name = "identifier", value = "模型仓库名称")
+    private String identifier;
+
+    @ApiModelProperty(name = "is_public", value = "是否公开1公开，0私有")
+    private Boolean isPublic;
+
+    @ApiModelProperty(name = "relative_paths",value = "文件相对路径")
+    private String relativePaths;
 }
+

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelsVo.java

@@ -12,34 +12,13 @@ import java.util.List;
 @Data
 public class ModelsVo extends ModelMetaVo implements Serializable {
 
-    private Integer id;
-
-    /**
-     * 模型可见范围
-     */
-    @ApiModelProperty(name = "available_range")
-    private int availableRange;
-
-    /**
-     * 状态
-     */
-    @ApiModelProperty(name = "status")
+    @ApiModelProperty(name = "status", value = "状态")
     private Integer status;
 
-    @ApiModelProperty(name = "models_version_vos")
+    @ApiModelProperty(name = "model_version_vos")
     private List<VersionVo> modelVersionVos;
 
-    /**
-     * 模型仓库名称
-     */
-    @ApiModelProperty(name = "repository_name")
-    private String repositoryName;
-
-    @ApiModelProperty(name = "repo_id")
-    private Integer repoId;
-
-    /**
-     * 父模型
-     */
+    @ApiModelProperty(name = "parent_model", value = "父模型")
     private String parentModel;
+
 }

ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ModelDependency1DaoMapper.xml

@@ -4,15 +4,24 @@
 
 
     <insert id="insert" keyProperty="id" useGeneratedKeys="true">
-        insert into model_dependency1(parent_model, model_name, version, repo_id, identifier, owner)
-        values (#{parentModel}, #{modelName}, #{version}, #{repoId}, #{identifier}, #{owner})
+        insert into model_dependency1(parent_model, model_name, version, repo_id, identifier, owner, meta)
+        values (#{parentModel}, #{modelName}, #{version}, #{repoId}, #{identifier}, #{owner}, #{meta})
     </insert>
 
-    <select id="queryByModelName" resultType="com.ruoyi.platform.domain.ModelDependency1">
+    <select id="queryModelDependency" resultType="com.ruoyi.platform.domain.ModelDependency1">
         select *
         from model_dependency1
-        where model_name = #{modelName}
-          and state = 1
+        where
+        state = 1
+        <if test="modelName != null and modelName != ''">
+            and model_name = #{modelName}
+        </if>
+        <if test="repoId != null and repoId != ''">
+            and repo_id = #{repoId}
+        </if>
+        <if test="owner != null and owner != ''">
+            and owner = #{owner}
+        </if>
     </select>
 
     <select id="queryByModelNameAndVersion" resultType="com.ruoyi.platform.domain.ModelDependency1">
@@ -47,6 +56,6 @@
         <if test="version != null and version != ''">
             ':', #{version},
         </if>
-            '%')
+        '%')
     </update>
 </mapper>