From 317f9ad3d2a37efb0dc3ab8b8229f6426c33cc43 Mon Sep 17 00:00:00 2001 From: chenzhihang <709011834@qq.com> Date: Wed, 25 Sep 2024 10:42:55 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E9=AA=8C=E6=A8=A1=E5=9E=8B=E6=89=8B?= =?UTF-8?q?=E5=8A=A8=E5=AF=BC=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/impl/ModelsServiceImpl.java | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java index 193651e7..f3d631aa 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java @@ -730,8 +730,7 @@ public class ModelsServiceImpl implements ModelsService { HashMap map = new HashMap<>(); map.put("trainTask", modelsVo.getTrainTask()); - String t1 = JSON.toJSONString(map); - buildingModel = modelDependency1Dao.queryByTrainTask(t1); + buildingModel = modelDependency1Dao.queryByTrainTask(JSON.toJSONString(map)); } else { dvcUtils.moveFiles(sourcePath, modelPath); @@ -778,8 +777,20 @@ public class ModelsServiceImpl implements ModelsService { YamlUtils.generateYamlFile(params, rootPath, "params"); if (buildingModel != null) { - //更新模型依赖 - modelDependency1Dao.updateState(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion(), Constant.State_valid); + if (Constant.Source_Hand_Export.equals(modelsVo.getModelSource())) { + //保存模型依赖 + BeansUtils.copyPropertiesIgnoreNull(buildingModel, modelDependency); + modelDependency.setState(Constant.State_valid); + modelDependency.setModelName(modelsVo.getName()); + modelDependency.setVersion(modelsVo.getVersion()); + modelDependency.setRepoId(modelsVo.getId()); + modelDependency.setIdentifier(repositoryName); + modelDependency.setMeta(meta); + modelDependency1Dao.insert(modelDependency); + } else { + //更新模型依赖 + modelDependency1Dao.updateState(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion(), Constant.State_valid); + } } else { //保存模型依赖 modelDependency.setRepoId(modelsVo.getId());