diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java index 293a4e10..27d34ae0 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java @@ -600,6 +600,10 @@ public class ExperimentServiceImpl implements ExperimentService { modelMetaVo.setName((String) targetModel.get("model_name")); modelMetaVo.setVersion((String) targetModel.get("model_version")); modelMetaVo.setOwner(gitLinkUsername); + + modelDependency.setRepoId(Integer.valueOf((String) targetModel.get("model_id"))); + modelDependency.setIdentifier((String) targetModel.get("model_identifier")); + modelDependency.setVersion((String) targetModel.get("model_version")); modelDependency.setModelName((String) targetModel.get("model_name")); List versionVos = new ArrayList<>(); 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 95dc35c2..07d1d97a 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 @@ -636,6 +636,7 @@ public class ModelsServiceImpl implements ModelsService { modelDependency.setParentModel(modelsVo.getParentModel()); modelDependency.setOwner(gitLinkUsername); modelDependency.setMeta(meta); + modelDependency.setState(Constant.State_valid); modelDependency1Dao.insert(modelDependency); CompletableFuture.supplyAsync(() -> { @@ -719,6 +720,9 @@ public class ModelsServiceImpl implements ModelsService { } } + ModelDependency1 buildingModel = modelDependency1Dao.getBuildingModel(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion()); + ModelMetaVo modelMetaVo1 = JSON.parseObject(buildingModel.getMeta(), ModelMetaVo.class); + //拼接生产的元数据后写入yaml文件 ModelMetaVo modelMetaVo = new ModelMetaVo(); BeanUtils.copyProperties(modelsVo, modelMetaVo); @@ -752,7 +756,6 @@ public class ModelsServiceImpl implements ModelsService { Map params = modelMetaVo.getParams(); YamlUtils.generateYamlFile(params, rootPath, "params"); - ModelDependency1 buildingModel = modelDependency1Dao.getBuildingModel(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion()); if (buildingModel != null) { //更新模型依赖 modelDependency1Dao.updateState(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion(), Constant.State_valid); @@ -765,6 +768,7 @@ public class ModelsServiceImpl implements ModelsService { modelDependency.setParentModel(modelsVo.getParentModel()); modelDependency.setOwner(gitLinkUsername); modelDependency.setMeta(meta); + modelDependency.setState(Constant.State_valid); modelDependency1Dao.insert(modelDependency); } diff --git a/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ModelDependency1DaoMapper.xml b/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ModelDependency1DaoMapper.xml index 031a2d73..69b6e8b8 100644 --- a/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ModelDependency1DaoMapper.xml +++ b/ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ModelDependency1DaoMapper.xml @@ -4,8 +4,8 @@ - insert into model_dependency1(parent_model, model_name, version, repo_id, identifier, owner, meta) - values (#{parentModel}, #{modelName}, #{version}, #{repoId}, #{identifier}, #{owner}, #{meta}) + insert into model_dependency1(parent_model, model_name, version, repo_id, identifier, owner, meta, state) + values (#{parentModel}, #{modelName}, #{version}, #{repoId}, #{identifier}, #{owner}, #{meta}, #{state})