diff --git a/configmanage/server/internal/dnsfile.go b/configmanage/server/internal/dnsfile.go index 7e74ff87d45218c70c53f1fa522bb266a086318a..a9d4ccd1050add568d63c8c89bd6e2a9ecfec692 100644 --- a/configmanage/server/internal/dnsfile.go +++ b/configmanage/server/internal/dnsfile.go @@ -58,3 +58,9 @@ func GetDNSFileByInfoUUID(uuid string, isindex interface{}) (DNSFile, error) { err := db.MySQL().Model(&DNSFile{}).Where("config_info_uuid=?", uuid).Find(&file).Error return file, err } + +func GetDNSFileByUUID(uuid string) (DNSFile, error) { + var file DNSFile + err := db.MySQL().Model(&DNSFile{}).Where("uuid=?", uuid).Find(&file).Error + return file, err +} diff --git a/configmanage/server/service/dns_test.go b/configmanage/server/service/dns_test.go index f6ba3b3f166e223a81f3e757600fd5ea4c53c39d..01b97521422e1b45c13f13bf86692d1b6f42997c 100644 --- a/configmanage/server/service/dns_test.go +++ b/configmanage/server/service/dns_test.go @@ -7,6 +7,7 @@ import ( "testing" "github.com/google/uuid" + "openeuler.org/PilotGo/configmanage-plugin/internal" ) func TestDNSConfig_Record(t *testing.T) { @@ -27,3 +28,26 @@ func TestDNSConfig_Record(t *testing.T) { os.Exit(-1) } } + +func TestDNSConfig_Load(t *testing.T) { + // 设置测试数据 + dc := &DNSConfig{ + ConfigInfoUUID: "cd1574a4-cdad-4a55-9561-9ef371456b90", + } + err := dc.Load() + if err != nil { + fmt.Printf("record error: %s\n", err) + os.Exit(-1) + } + fmt.Printf("hc: %v\n", dc) +} + +func TestGetDNSFileByUUID(t *testing.T) { + uuid := "a134a449-d635-4f08-8dd8-1e3de2a6a509" + df, err := internal.GetDNSFileByUUID(uuid) + if err != nil { + fmt.Printf("get dnsfile error: %s\n", err) + os.Exit(-1) + } + fmt.Printf("hc: %v\n", df) +} diff --git a/configmanage/server/service/ssh.go b/configmanage/server/service/ssh.go index 0f03686ff3b629dd09af6b34aca41a9e90eb4529..a0158af2e50afd3cb84ad2b57851d6d9dd095f16 100644 --- a/configmanage/server/service/ssh.go +++ b/configmanage/server/service/ssh.go @@ -85,7 +85,7 @@ func (sc *SSHConfig) Load() error { func (sc *SSHConfig) Apply() ([]NodeResult, error) { // 从数据库获取下发的信息 - sf, err := internal.GetHostFileByUUID(sc.UUID) + sf, err := internal.GetSSHFileByUUID(sc.UUID) if err != nil { return nil, err }