From f689591ade2457d5fa707353c0b5959e6e486589 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期四, 16 四月 2020 15:54:34 +0800
Subject: [PATCH] fix verification switch
---
main.go | 101 +++++---------------------------------------------
1 files changed, 11 insertions(+), 90 deletions(-)
diff --git a/main.go b/main.go
index 5388689..921592d 100644
--- a/main.go
+++ b/main.go
@@ -1,97 +1,18 @@
package main
import (
- "bytes"
- "fmt"
- "io/ioutil"
- "net/http"
- "os/exec"
- "time"
+ "flag"
+ "swfs/config"
+ "swfs/router"
)
+var env = flag.String("pro", "pro", "read storage info")
+
+func init() {
+ config.Init(*env)
+}
+
func main() {
-
- oldPeers := GetOldPeers()
- fmt.Println("oldPeers: ", oldPeers)
- newPeers := GetNewPeers()
- fmt.Println("newPeers: ", newPeers)
- UpdatePeers(oldPeers, newPeers)
- time.Sleep(time.Second * 3)
- nowPeers := GetOldPeers()
- fmt.Println("nowPeers: ", nowPeers)
-}
-
-func GetOldPeers() string {
- str := "cat /opt/vasystem/seaweedfs_start.sh | grep peers="
- peers := RunScript(str)
- return peers
-}
-
-func GetNewPeers() string {
- getUrl := "http://192.168.20.10:9200/basicfs/_search"
- getJson := `{
- "query": {
- "bool": {
- "filter": [
- {
- "term": {
- "application":"nodeOperation"
- }
- }
- ]
- }
- },
- "size": 1
-}`
-
- getRes, _ := EsReq("POST", getUrl, []byte(getJson))
- fmt.Println(getRes)
- return ""
-}
-
-func UpdatePeers(oldPeers string, newPeers string) {
- str := "sed -ie 's/" + oldPeers + "/" + newPeers + "/g' /opt/vasystem/seaweedfs_start.sh"
- RunScript(str)
-}
-
-//鑴氭湰灏佽
-func RunScript(str string) string {
-
- cmd := exec.Command("sh", "-c", str)
- var out bytes.Buffer
- cmd.Stdout = &out
- err := cmd.Run()
- if err != nil {
- return "杩愯澶辫触"
- }
- return out.String()
-}
-
-//瑙ｆ瀽http
-func EsReq(method string, url string, parama []byte) (buf []byte, err error) {
- timeout := time.Duration(10 * time.Second)
- client := http.Client{
- Timeout: timeout,
- }
- 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
- }
-
- resp, err := client.Do(request)
- if err != nil {
- fmt.Println("request error: ", err)
- return nil, err
- }
-
- defer resp.Body.Close()
- body, err := ioutil.ReadAll(resp.Body)
- if err != nil {
- fmt.Println(err)
- return nil, err
- }
- return body, nil
+ r := router.NewRouter()
+ r.Run("0.0.0.0:7020")
}
--
Gitblit v1.8.0