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