From 4a984efeee5ba221cfaac8edc53609f866a66c27 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Tue, 5 Nov 2024 14:46:03 +0800
Subject: [PATCH] add GetSSHFilesByCinfigUUID func

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

diff --git a/configmanage/server/internal/ssh.go b/configmanage/server/internal/ssh.go
index 1bb04b1b..2abc1230 100644
--- a/configmanage/server/internal/ssh.go
+++ b/configmanage/server/internal/ssh.go
@@ -51,3 +51,10 @@ func (sf *SSHFile) UpdateByuuid() error {
 	// 将成功下发的具体某一个配置状态修改为已使用
 	return db.MySQL().Model(&SSHFile{}).Where("uuid=?", sf.UUID).Update("is_index", 1).Error
 }
+
+// 根据配置uuid获取所有配置文件
+func GetSSHFilesByCinfigUUID(uuid string) ([]SSHFile, error) {
+	var files []SSHFile
+	err := db.MySQL().Model(&SSHFile{}).Where("config_info_uuid=?", uuid).Find(&files).Error
+	return files, err
+}
diff --git a/configmanage/server/service/ssh.go b/configmanage/server/service/ssh.go
index c52f5485..47af4574 100644
--- a/configmanage/server/service/ssh.go
+++ b/configmanage/server/service/ssh.go
@@ -155,3 +155,8 @@ func (rc *SSHConfig) Apply() (json.RawMessage, error) {
 func (sc *SSHConfig) Collect() error {
 	return nil
 }
+
+// 根据配置uuid获取所有配置文件
+func GetSSHFilesByCinfigUUID(uuid string) ([]SSHFile, error) {
+	return internal.GetSSHFilesByCinfigUUID(uuid)
+}
diff --git a/configmanage/server/service/ssh_test.go b/configmanage/server/service/ssh_test.go
index 9842a23e..f9712b91 100644
--- a/configmanage/server/service/ssh_test.go
+++ b/configmanage/server/service/ssh_test.go
@@ -51,3 +51,20 @@ func TestGetSSHFileByUUID(t *testing.T) {
 	}
 	fmt.Printf("hc: %v\n", sf)
 }
+
+func TestGetSSHFilesByCinfigUUID(t *testing.T) {
+	// 设置测试数据
+	testUUID := "5973e993-6236-4b53-9eb6-0cc23c652460"
+
+	// 调用被测试的函数
+	files, err := GetSSHFilesByCinfigUUID(testUUID)
+	if err != nil {
+		fmt.Printf("load sshfiles 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