From 60fb4e16faf7a66ce6c8b1890cf6450bdec98342 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Fri, 1 Nov 2024 15:40:11 +0800
Subject: [PATCH] add tags

---
 configmanage/server/internal/confignode.go |  9 ++++++
 configmanage/server/main.go                |  1 +
 configmanage/server/service/tags.go        | 33 ++++++++++++++++++++++
 3 files changed, 43 insertions(+)
 create mode 100644 configmanage/server/service/tags.go

diff --git a/configmanage/server/internal/confignode.go b/configmanage/server/internal/confignode.go
index 366ff698..68c56b70 100644
--- a/configmanage/server/internal/confignode.go
+++ b/configmanage/server/internal/confignode.go
@@ -19,6 +19,15 @@ func GetConfigNodesByUUID(uuid string) ([]string, error) {
 	return nodes, err
 }
 
+func IsExistNode(uuid string) bool {
+	var cns []ConfigNode
+	err := db.MySQL().Model(&ConfigNode{}).Where("node_id=?", uuid).Find(&cns).Error
+	if err != nil || len(cns) == 0 {
+		return false
+	}
+	return true
+}
+
 type ConfigBatch struct {
 	ID             int        `gorm:"primary_key;AUTO_INCREMENT"`
 	ConfigInfo     ConfigInfo `gorm:"Foreignkey:ConfigInfoUUID"`
diff --git a/configmanage/server/main.go b/configmanage/server/main.go
index 815c6a17..01bb884c 100644
--- a/configmanage/server/main.go
+++ b/configmanage/server/main.go
@@ -50,6 +50,7 @@ func main() {
 	}
 
 	global.GlobalClient = client.DefaultClient(global.Init(config.Config().ConfigPlugin))
+	service.GetTags()
 	service.AddExtentions()
 	service.AddPermission()
 
diff --git a/configmanage/server/service/tags.go b/configmanage/server/service/tags.go
new file mode 100644
index 00000000..8f4a28e8
--- /dev/null
+++ b/configmanage/server/service/tags.go
@@ -0,0 +1,33 @@
+package service
+
+import (
+	"gitee.com/openeuler/PilotGo/sdk/common"
+	"openeuler.org/PilotGo/configmanage-plugin/global"
+	"openeuler.org/PilotGo/configmanage-plugin/internal"
+)
+
+func GetTags() {
+	tag_cb := func(uuids []string) []common.Tag {
+		var tags []common.Tag
+		for _, uuid := range uuids {
+			ok := internal.IsExistNode(uuid)
+			if ok {
+				tag := common.Tag{
+					UUID: uuid,
+					Type: common.TypeOk,
+					Data: "configmanage",
+				}
+				tags = append(tags, tag)
+			} else {
+				tag := common.Tag{
+					UUID: uuid,
+					Type: common.TypeError,
+					Data: "",
+				}
+				tags = append(tags, tag)
+			}
+		}
+		return tags
+	}
+	global.GlobalClient.OnGetTags(tag_cb)
+}
-- 
Gitee