From b0f42c19b056555308516c9c8e50069d64e93884 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Wed, 6 Nov 2024 11:04:33 +0800
Subject: [PATCH] add GetSSHDFilesByCinfigUUID func

---
 configmanage/server/internal/sshd.go    |  7 +++++++
 configmanage/server/service/ssh_test.go | 16 ++++++++++++++++
 configmanage/server/service/sshd.go     |  5 +++++
 3 files changed, 28 insertions(+)

diff --git a/configmanage/server/internal/sshd.go b/configmanage/server/internal/sshd.go
index a0a8fd26..d151ff6d 100644
--- a/configmanage/server/internal/sshd.go
+++ b/configmanage/server/internal/sshd.go
@@ -50,3 +50,10 @@ func (sdf *SSHDFile) UpdateByuuid() error {
 	// 将成功下发的具体某一个配置状态修改为已使用
 	return db.MySQL().Model(&SSHDFile{}).Where("uuid=?", sdf.UUID).Update("is_index", 1).Error
 }
+
+// 根据配置uuid获取所有配置文件
+func GetSSHDFilesByCinfigUUID(uuid string) ([]SSHDFile, error) {
+	var files []SSHDFile
+	err := db.MySQL().Model(&SSHDFile{}).Where("config_info_uuid=?", uuid).Find(&files).Error
+	return files, err
+}
diff --git a/configmanage/server/service/ssh_test.go b/configmanage/server/service/ssh_test.go
index b973c2ea..29e69bb9 100644
--- a/configmanage/server/service/ssh_test.go
+++ b/configmanage/server/service/ssh_test.go
@@ -81,3 +81,19 @@ func TestGetSSHFilesByNode(t *testing.T) {
 	}
 	fmt.Println(len(rcs))
 }
+func TestGetSSHDFilesByCinfigUUID(t *testing.T) {
+	// 设置测试数据
+	sdcUUID := "5973e993-6236-4b53-9eb6-0cc23c652460"
+
+	// 调用被测试的函数
+	files, err := GetSSHDFilesByCinfigUUID(sdcUUID)
+	if err != nil {
+		fmt.Printf("load sshdfiles 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))
+}
diff --git a/configmanage/server/service/sshd.go b/configmanage/server/service/sshd.go
index 27d48e43..38f3d380 100644
--- a/configmanage/server/service/sshd.go
+++ b/configmanage/server/service/sshd.go
@@ -157,3 +157,8 @@ func (sdc *SSHDConfig) Apply() (json.RawMessage, error) {
 func (sdc *SSHDConfig) Collect() error {
 	return nil
 }
+
+// 根据配置uuid获取所有配置文件
+func GetSSHDFilesByCinfigUUID(uuid string) ([]SSHDFile, error) {
+	return internal.GetSSHDFilesByCinfigUUID(uuid)
+}
-- 
Gitee