| | |
| | | "io/ioutil" |
| | | "net" |
| | | "os" |
| | | "sync" |
| | | |
| | | //"os" |
| | | "strings" |
| | |
| | | switch ev := event.(type) { |
| | | case serf.UserEvent: |
| | | fmt.Println(string(ev.Payload)) |
| | | var tmpstringslice []string |
| | | tmpstringslice = append(tmpstringslice, string(ev.Payload)) |
| | | fmt.Println(tmpstringslice) |
| | | results, err := ExecuteWriteSql(tmpstringslice) |
| | | |
| | | for _, result := range results { |
| | | fmt.Println(result, "results err: ", err) |
| | | } |
| | | var sqlArr []string |
| | | sqlArr = append(sqlArr, string(ev.Payload)) |
| | | //results, err := ExecuteWriteSql(sqlArr) |
| | | flag, _ := ExecuteSqlByGorm(sqlArr) |
| | | fmt.Println("userEvent exec ",sqlArr,",Result:",flag) |
| | | |
| | | case *serf.Query: |
| | | |
| | |
| | | fmt.Println("Query tableNames unmarshal err") |
| | | return |
| | | } |
| | | fmt.Println("Query tableNames:",tableNames) |
| | | datas, err := ExecuteQueryByGorm(tableNames) |
| | | if err !=nil { |
| | | fmt.Println("queryByGorm err") |
| | | return |
| | | } |
| | | bytesReturn, err := json.Marshal(datas) |
| | | fmt.Println("results: ", bytesReturn) |
| | | fmt.Println("results.len: ", len(bytesReturn)) |
| | | if query, ok := event.(*serf.Query); ok { |
| | | if err := query.Respond(bytesReturn); err != nil { |
| | | fmt.Printf("err: %s\n", err) |
| | |
| | | |
| | | //GetDbFromCluster get the newest database after join cluster |
| | | //dbPathWrite the path where to write after got a database, |
| | | func (a *Agent) GetTableDataFromCluster(tableNames []string) error { |
| | | func (a *Agent) GetTableDataFromCluster(tableNames []string) (*[]string,error) { |
| | | //members: get name of first member |
| | | mbs := a.GroupMembers(a.conf.ClusterID) |
| | | var specmembername string |
| | | for _, m := range mbs { |
| | | if m.Addr.String() != a.conf.BindAddr { |
| | | fmt.Println("m",m) |
| | | if m.Name != a.conf.NodeName { |
| | | specmembername = m.Name |
| | | break |
| | | } |
| | | } |
| | | fmt.Println(specmembername) |
| | | fmt.Println("mbs:",mbs,"a.conf.BindAddr:",a.conf.BindAddr,"specmembername:",specmembername) |
| | | |
| | | //query: get db file. |
| | | params := serf.QueryParam{ |
| | |
| | | if err == nil || !strings.Contains(err.Error(), "cannot contain") { |
| | | fmt.Println("err: ", err) |
| | | } |
| | | fmt.Println("Query.resp.err:",err,"resp:",resp) |
| | | |
| | | |
| | | var wg sync.WaitGroup |
| | | wg.Add(1) |
| | | var dumpSqls []string |
| | | go func() { |
| | | defer wg.Done() |
| | | respCh := resp.ResponseCh() |
| | | for { |
| | | select { |
| | | case r := <-respCh: |
| | | fmt.Println("x length is: ", len(r.Payload)) |
| | | // // byte to file. |
| | | var dumpSqls []string |
| | | fmt.Println("Query response's len:", len(r.Payload)) |
| | | err := json.Unmarshal(r.Payload, &dumpSqls) |
| | | if err ==nil { |
| | | if flag,_ := ExecuteSqlByGorm(dumpSqls);flag{ |
| | | fmt.Println("data dump success") |
| | | } |
| | | fmt.Println("dumpSql:",dumpSqls) |
| | | fmt.Println("data dump success") |
| | | } |
| | | return |
| | | } |
| | | } |
| | | }() |
| | | return nil |
| | | wg.Wait() |
| | | return &dumpSqls,nil |
| | | |
| | | //r, err = c.Query([]string{query}, false, false) |
| | | //if err != nil { |