From c5301a614f367854d9548179fd4a4b6b9964bf34 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期一, 19 十月 2020 15:46:09 +0800
Subject: [PATCH] add set hosts

---
 ClusterApi.go |   32 ++++++++++++++++++++++++++------
 1 files changed, 26 insertions(+), 6 deletions(-)

diff --git a/ClusterApi.go b/ClusterApi.go
index 81614f8..cea0aef 100644
--- a/ClusterApi.go
+++ b/ClusterApi.go
@@ -65,7 +65,7 @@
 	return true, nil
 }
 
-//楠岃瘉璇ヨ妭鐐规槸鍚﹁鍒涘缓杩�
+//楠岃瘉璇ヨ妭鐐规槸鍚﹁浣跨敤杩�
 func VerifyCreated(configPath string, ip string) (bool, error) {
 	v := viper.New()
 	v.SetConfigType("yml")
@@ -135,6 +135,22 @@
 	return role, nil
 }
 
+func SetDiscoveryZenPingUnicastHosts(configPath string, hosts []string) (bool, error) {
+	v := viper.New()
+	v.SetConfigType("yml")
+	v.SetConfigName("elasticsearch")
+	v.AddConfigPath(configPath)
+	err := v.ReadInConfig()
+	if err != nil {
+		return false, err
+	}
+	v.Set("discovery.zen.ping.unicast.hosts", hosts)
+	if err := v.WriteConfig(); err != nil {
+		return false, err
+	}
+	return true,nil
+}
+
 //鏇存柊缁勬挱鍒楄〃
 func UpdateDiscoveryZenPingUnicastHosts(configPath string, oldIp string, newIp string) (bool, error) {
 	v := viper.New()
@@ -164,18 +180,22 @@
 }
 
 //娣诲姞缁勬挱鎴愬憳
-func AddDiscoveryZenPingUnicastHosts(configPath string, ip string) []string {
+func AddDiscoveryZenPingUnicastHosts(configPath string, ip string) (bool, error) {
 	v := viper.New()
 	v.SetConfigType("yml")
 	v.SetConfigName("elasticsearch")
 	v.AddConfigPath(configPath)
-	err := v.ReadInConfig()
-	if err != nil {
-
+	errR := v.ReadInConfig()
+	if errR != nil {
+		return false, errR
 	}
 	hosts := v.GetStringSlice("discovery.zen.ping.unicast.hosts")
 	hosts = append(hosts, ip)
-	return hosts
+	v.Set("discovery.zen.ping.unicast.hosts", hosts)
+	if errW := v.WriteConfig(); errW != nil {
+		return false, errW
+	}
+	return true, nil
 }
 
 type NodeInfo struct {

--
Gitblit v1.8.0