From c80fe5df67af15b0749f2870ca6312fa5b35dd2b Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 18 Apr 2024 09:37:23 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E7=BD=91=E5=85=83=E4=B8=BB=E6=9C=BA?=
 =?UTF-8?q?=E8=AE=A4=E8=AF=81=E6=A8=A1=E5=BC=8F=E6=B8=85=E9=99=A4=E4=B8=8D?=
 =?UTF-8?q?=E5=BF=85=E8=A6=81=E7=9A=84=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../repository/ne_host.impl.go                | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/modules/network_element/repository/ne_host.impl.go b/src/modules/network_element/repository/ne_host.impl.go
index 156b5709..b1effdc1 100644
--- a/src/modules/network_element/repository/ne_host.impl.go
+++ b/src/modules/network_element/repository/ne_host.impl.go
@@ -294,6 +294,20 @@ func (r *NeHostImpl) Insert(neHost model.NeHost) string {
 		params["create_time"] = time.Now().UnixMilli()
 	}
 
+	// 根据认证模式清除不必要的信息
+	if neHost.AuthMode == "0" {
+		params["private_key"] = ""
+		params["pass_phrase"] = ""
+	}
+	if neHost.AuthMode == "1" {
+		params["password"] = ""
+	}
+	if neHost.AuthMode == "2" {
+		params["password"] = ""
+		params["private_key"] = ""
+		params["pass_phrase"] = ""
+	}
+
 	// 构建执行语句
 	keys, placeholder, values := repo.KeyPlaceholderValueByInsert(params)
 	sql := "insert into ne_host (" + strings.Join(keys, ",") + ")values(" + placeholder + ")"
@@ -376,6 +390,20 @@ func (r *NeHostImpl) Update(neHost model.NeHost) int64 {
 		params["update_time"] = time.Now().UnixMilli()
 	}
 
+	// 根据认证模式清除不必要的信息
+	if neHost.AuthMode == "0" {
+		params["private_key"] = ""
+		params["pass_phrase"] = ""
+	}
+	if neHost.AuthMode == "1" {
+		params["password"] = ""
+	}
+	if neHost.AuthMode == "2" {
+		params["password"] = ""
+		params["private_key"] = ""
+		params["pass_phrase"] = ""
+	}
+
 	// 构建执行语句
 	keys, values := repo.KeyValueByUpdate(params)
 	sql := "update ne_host set " + strings.Join(keys, ",") + " where host_id = ?"