From dd88506d9e32017019af5f3788145d40df14e42f Mon Sep 17 00:00:00 2001
From: zmyYYDS <864398951@qq.com>
Date: Thu, 21 Dec 2023 11:48:40 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9C=80=E6=96=B0=E4=B8=80=E7=89=882.0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../srt/controller/MetamodelController.java   | 12 ++++---
 .../net/srt/service/MetamodelService.java     |  3 ++
 .../service/impl/MetamodelServiceImpl.java    | 34 +++++++++++++++++++
 3 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/srt-cloud-data-governance/src/main/java/net/srt/controller/MetamodelController.java b/srt-cloud-data-governance/src/main/java/net/srt/controller/MetamodelController.java
index 01dc863..3fc145a 100644
--- a/srt-cloud-data-governance/src/main/java/net/srt/controller/MetamodelController.java
+++ b/srt-cloud-data-governance/src/main/java/net/srt/controller/MetamodelController.java
@@ -10,10 +10,7 @@ import net.srt.framework.common.utils.Result;
 import net.srt.framework.common.utils.TreeNodeVo;
 import net.srt.service.MetamodelService;
 import net.srt.vo.MetamodelVO;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 
@@ -38,4 +35,11 @@ public class MetamodelController {
         MetamodelEntity entity = metamodelService.getById(id);
         return Result.ok(MetamodelConvert.INSTANCE.convert(entity));
     }
+
+    @PostMapping
+    @Operation(summary = "保存")
+    public Result<String> save(@RequestBody MetamodelVO vo){
+        metamodelService.save(vo);
+        return Result.ok();
+    }
 }
diff --git a/srt-cloud-data-governance/src/main/java/net/srt/service/MetamodelService.java b/srt-cloud-data-governance/src/main/java/net/srt/service/MetamodelService.java
index 41b376b..f33d1d6 100644
--- a/srt-cloud-data-governance/src/main/java/net/srt/service/MetamodelService.java
+++ b/srt-cloud-data-governance/src/main/java/net/srt/service/MetamodelService.java
@@ -3,9 +3,12 @@ package net.srt.service;
 import net.srt.entity.MetamodelEntity;
 import net.srt.framework.common.utils.TreeNodeVo;
 import net.srt.framework.mybatis.service.BaseService;
+import net.srt.vo.MetamodelVO;
 
 import java.util.List;
 
 public interface MetamodelService extends BaseService<MetamodelEntity> {
     List<TreeNodeVo> listTree();
+
+    void save(MetamodelVO vo);
 }
diff --git a/srt-cloud-data-governance/src/main/java/net/srt/service/impl/MetamodelServiceImpl.java b/srt-cloud-data-governance/src/main/java/net/srt/service/impl/MetamodelServiceImpl.java
index b5929a6..9ddb155 100644
--- a/srt-cloud-data-governance/src/main/java/net/srt/service/impl/MetamodelServiceImpl.java
+++ b/srt-cloud-data-governance/src/main/java/net/srt/service/impl/MetamodelServiceImpl.java
@@ -1,6 +1,7 @@
 package net.srt.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import net.srt.convert.MetamodelConvert;
 import net.srt.dao.MetamodelDao;
 import net.srt.entity.MetamodelEntity;
 import net.srt.framework.common.utils.BeanUtil;
@@ -8,7 +9,9 @@ import net.srt.framework.common.utils.BuildTreeUtils;
 import net.srt.framework.common.utils.TreeNodeVo;
 import net.srt.framework.mybatis.service.impl.BaseServiceImpl;
 import net.srt.service.MetamodelService;
+import net.srt.vo.MetamodelVO;
 import org.springframework.stereotype.Service;
+import srt.cloud.framework.dbswitch.common.util.StringUtil;
 
 import java.util.List;
 @Service
@@ -32,4 +35,35 @@ public class MetamodelServiceImpl extends BaseServiceImpl<MetamodelDao, Metamode
         });
         return BuildTreeUtils.buildTree(treeNodeVos);
     }
+
+    @Override
+    public void save(MetamodelVO vo) {
+        MetamodelEntity entity = MetamodelConvert.INSTANCE.convert(vo);
+        entity.setPath(recursionPath(entity,null));
+        entity.setProjectId(getProjectId());
+        entity.setBuiltin(0);
+        buildField(entity);
+        baseMapper.insert(entity);
+    }
+
+    private void buildField(MetamodelEntity entity) {
+        if(entity.getIfLeaf() == 0){
+            entity.setIcon("/src/assets/model.png");
+        }else {
+            entity.setIcon("/src/assets/folder.png");
+        }
+    }
+
+    private String recursionPath(MetamodelEntity metamodelEntity,String path){
+        if(StringUtil.isBlank(path)){
+            path = metamodelEntity.getName();
+        }
+        if(metamodelEntity.getParentId()!=0){
+            MetamodelEntity parent = getById(metamodelEntity.getParentId());
+            path = parent.getName() + "/" + path;
+            return recursionPath(parent,path);
+        }
+        return  path;
+    }
+
 }