From 4f2c8f2df1c90553123c640bb9b4dd91bff36164 Mon Sep 17 00:00:00 2001 From: sunty <1172534965@qq.com> Date: 星期四, 22 十月 2020 18:52:35 +0800 Subject: [PATCH] fix --- ClusterApi.go | 87 +++++++++++++++++++++++++++++++++---------- 1 files changed, 66 insertions(+), 21 deletions(-) diff --git a/ClusterApi.go b/ClusterApi.go index 81614f8..bdae03a 100644 --- a/ClusterApi.go +++ b/ClusterApi.go @@ -29,19 +29,35 @@ return true, nil } -//鍚姩鏈嶅姟 -func StartServer(binPath string, ip string) bool { - resultMsg := CMDSC("sh " + binPath + "/elasticsearch -d") +//鍒濆鍖栫储寮� +func InitIndex(indexPath string) (bool) { + resultMsg := CMDSC("sh " + indexPath + "/indexInit.sh") if resultMsg == "杩愯澶辫触" { return false } - bool := VerifyNodeServer(ip, "9200") - return bool + return true +} + +//鍚姩鏈嶅姟 +func StartServer(binPath string, ip string, port string) bool { + resultMsg := CMDSC("sh " + binPath + "/es_start.sh") + if resultMsg == "杩愯澶辫触" { + return false + } + return true } //鍏抽棴鏈嶅姟 -func StopServer(binPath string) { - +func StopServer(binPath string, ip string, port string) bool { + b := VerifyShortNodeServer(ip, port) + if b == true { + resultMsg := CMDSC("sh " + binPath + "/es_stop.sh") + if resultMsg == "杩愯澶辫触" { + return true + } + } + bs := VerifyShortNodeServer(ip, port) + return bs } //鏇存崲鑺傜偣瑙掕壊 @@ -65,8 +81,8 @@ return true, nil } -//楠岃瘉璇ヨ妭鐐规槸鍚﹁鍒涘缓杩� -func VerifyCreated(configPath string, ip string) (bool, error) { +//楠岃瘉璇ヨ妭鐐规槸鍚﹁浣跨敤杩� +func VerifyCreated(configPath string) (bool, error) { v := viper.New() v.SetConfigType("yml") v.SetConfigName("elasticsearch") @@ -76,13 +92,8 @@ return true, err } hosts := v.GetStringSlice("discovery.zen.ping.unicast.hosts") - for _, pick := range hosts { - if pick == ip { - return true, errors.New("璇ヨ妭鐐瑰凡缁忚鎵ц鈥滃垱寤洪泦缇も��") - } - } if len(hosts) > 1 { - return true, errors.New("璇ヨ妭鐐瑰凡缁忚鎵ц鈥滃姞鍏ラ泦缇も��") + return true, errors.New("璇ヨ妭鐐瑰凡缁忚浣跨敤") } return false, nil } @@ -135,6 +146,37 @@ return role, nil } +//鏌ヨ缁勬挱鍒楄〃 +func GetDiscoveryZenPingUnicastHosts(configPath string) ([]string, error) { + v := viper.New() + v.SetConfigType("yml") + v.SetConfigName("elasticsearch") + v.AddConfigPath(configPath) + err := v.ReadInConfig() + if err != nil { + return nil, err + } + hosts := v.GetStringSlice("discovery.zen.ping.unicast.hosts") + return hosts, 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 +206,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 { @@ -257,7 +303,6 @@ } request, err := http.NewRequest(method, url, bytes.NewBuffer(parama)) request.Header.Set("Content-type", "application/json") - if err != nil { fmt.Println("build request fail !") return nil, err -- Gitblit v1.8.0