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