From 1395627c10afeede362b46f03fc25d8d8dbd91e4 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Mon, 18 Nov 2024 14:33:11 +0800
Subject: [PATCH] add GetDNSFilesByConfigUUID func and test

---
 configmanage/server/internal/dnsfile.go |  7 +++++++
 configmanage/server/service/dns.go      |  5 +++++
 configmanage/server/service/dns_test.go | 17 +++++++++++++++++
 3 files changed, 29 insertions(+)

diff --git a/configmanage/server/internal/dnsfile.go b/configmanage/server/internal/dnsfile.go
index 5c68b8a0..d865e5a0 100644
--- a/configmanage/server/internal/dnsfile.go
+++ b/configmanage/server/internal/dnsfile.go
@@ -74,3 +74,10 @@ func (df *DNSFile) UpdateByuuid() error {
 	// 将成功下发的具体某一个配置状态修改为已使用
 	return db.MySQL().Model(&DNSFile{}).Where("uuid=?", df.UUID).Update("is_index", 1).Error
 }
+
+// 根据配置uuid获取所有配置文件
+func GetDNSFilesByConfigUUID(uuid string) ([]DNSFile, error) {
+	var files []DNSFile
+	err := db.MySQL().Model(&DNSFile{}).Where("config_info_uuid=?", uuid).Find(&files).Error
+	return files, err
+}
diff --git a/configmanage/server/service/dns.go b/configmanage/server/service/dns.go
index 645af611..3a34e098 100644
--- a/configmanage/server/service/dns.go
+++ b/configmanage/server/service/dns.go
@@ -66,3 +66,8 @@ func (dc *DNSConfig) Load() error {
 func GetDNSFileByInfoUUID(uuid string, isindex interface{}) (DNSFile, error) {
 	return internal.GetDNSFileByInfoUUID(uuid, isindex)
 }
+
+// 根据配置uuid获取所有配置文件
+func GetDNSFilesByConfigUUID(uuid string) ([]DNSFile, error) {
+	return internal.GetDNSFilesByConfigUUID(uuid)
+}
diff --git a/configmanage/server/service/dns_test.go b/configmanage/server/service/dns_test.go
index 01b97521..7e353d25 100644
--- a/configmanage/server/service/dns_test.go
+++ b/configmanage/server/service/dns_test.go
@@ -51,3 +51,20 @@ func TestGetDNSFileByUUID(t *testing.T) {
 	}
 	fmt.Printf("hc: %v\n", df)
 }
+
+func TestGetDNSFilesByConfigUUID(t *testing.T) {
+	// 设置测试数据
+	testUUID := "cd1574a4-cdad-4a55-9561-9ef371456b90"
+
+	// 调用被测试的函数
+	files, err := GetDNSFilesByConfigUUID(testUUID)
+	if err != nil {
+		fmt.Printf("load dnsfile error: %s\n", err)
+		os.Exit(-1)
+	}
+	if len(files) == 0 {
+		fmt.Printf("files is empty: %s\n", err)
+		os.Exit(-1)
+	}
+	fmt.Println(len(files))
+}
-- 
Gitee