From 4300502182c646927e77639d534472a0b3f68b6c Mon Sep 17 00:00:00 2001 From: wubijie Date: Wed, 13 Nov 2024 10:32:47 +0800 Subject: [PATCH] add update ssh comfig --- .../server/controller/configinstance.go | 45 +++++++++++++++++++ configmanage/server/service/ssh.go | 4 ++ 2 files changed, 49 insertions(+) diff --git a/configmanage/server/controller/configinstance.go b/configmanage/server/controller/configinstance.go index 1c89baf8..0998059b 100644 --- a/configmanage/server/controller/configinstance.go +++ b/configmanage/server/controller/configinstance.go @@ -590,6 +590,51 @@ func UpdateConfigHandler(c *gin.Context) { response.Success(c, nil, "update host config success") case global.SSH: + // 解析参数 + sshconfig := &service.SSHConfig{} + if err := json.Unmarshal(query.Data, sshconfig); err != nil { + logger.Error("failed to parse sshconfig parameter: %s", err.Error()) + response.Fail(c, "failed to parse sshconfig parameter:", err.Error()) + return + } + + file := common.File{} + if err := json.Unmarshal([]byte(sshconfig.Content), &file); err != nil { + logger.Error("failed to parse file parameter: %s", err.Error()) + response.Fail(c, "failed to parse file parameter:", err.Error()) + return + } + file.Content = base64.StdEncoding.EncodeToString([]byte(file.Content)) + + // 查询此配置的内容 + sf, err := service.GetSSHFileByInfoUUID(ci.UUID, nil) + if err != nil { + logger.Error("failed to get sshconfig parameter: %s", err.Error()) + response.Fail(c, "failed to get sshconfig parameter:", err.Error()) + return + } + + // 更新参数 + sshconfig.UUID = sf.UUID + sshconfig.ConfigInfoUUID = ci.UUID + sshconfig.IsActive = false + sshconfig.Content, err = json.Marshal(file) + if err != nil { + logger.Error("Error encoding JSON:: %s", err.Error()) + response.Fail(c, "Error encoding JSON:", err.Error()) + return + } + + // 将参数添加到数据库 + err = sshconfig.Record() + if err != nil { + logger.Error("failed to update sshconfig: %s", err.Error()) + response.Fail(c, "failed to update sshconfig:", err.Error()) + return + } + + logger.Debug("update sshconfig success") + response.Success(c, nil, "update sshconfig success") case global.SSHD: diff --git a/configmanage/server/service/ssh.go b/configmanage/server/service/ssh.go index 7fc7133d..105194b1 100644 --- a/configmanage/server/service/ssh.go +++ b/configmanage/server/service/ssh.go @@ -229,6 +229,10 @@ func (sc *SSHConfig) Collect() error { return nil } +func GetSSHFileByInfoUUID(uuid string, isindex interface{}) (SSHFile, error) { + return internal.GetSSHFileByInfoUUID(uuid, isindex) +} + // 根据配置uuid获取所有配置文件 func GetSSHFilesByCinfigUUID(uuid string) ([]SSHFile, error) { return internal.GetSSHFilesByCinfigUUID(uuid) -- Gitee