From 1d801237e476f40af8a6974b871afe85cdca5441 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 09 八月 2019 11:20:39 +0800
Subject: [PATCH] test

---
 agent.go |   97 ++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 85 insertions(+), 12 deletions(-)

diff --git a/agent.go b/agent.go
index 78af264..1f032cb 100644
--- a/agent.go
+++ b/agent.go
@@ -159,21 +159,32 @@
 				}
 			}
 		} else if ev.Name == QueryEventUpdateDBData {
-			fmt.Println(string(ev.Payload))
-			var tmpstringslice []string
-			tmpstringslice = append(tmpstringslice, string(ev.Payload))
-			fmt.Println(tmpstringslice)
-			rows, err := ExecuteQuerySql(tmpstringslice)
-			if err != nil {
-				fmt.Println("err: ", err)
+			//fmt.Println(string(ev.Payload))
+			//var tmpstringslice []string
+			//tmpstringslice = append(tmpstringslice, string(ev.Payload))
+			//fmt.Println(tmpstringslice)
+			//rows, err := ExecuteQuerySql(tmpstringslice)
+			//if err != nil {
+			//	fmt.Println("err: ", err)
+			//	return
+			//}
+			//var rowsReturn []Rows
+			//for _, r := range rows {
+			//	rowsReturn = append(rowsReturn, *r)
+			//}
+			var tableNames []string
+			err := json.Unmarshal(ev.Payload, &tableNames)
+			if err !=nil {
+				fmt.Println("Query tableNames unmarshal err")
 				return
 			}
-			var rowsReturn []Rows
-			for _, r := range rows {
-				rowsReturn = append(rowsReturn, *r)
+			fmt.Println("Query tableNames:",tableNames)
+			datas, err := ExecuteQueryByGorm(tableNames)
+			if err !=nil {
+				fmt.Println("queryByGorm err")
+				return
 			}
-
-			bytesReturn, err := json.Marshal(rowsReturn)
+			bytesReturn, err := json.Marshal(datas)
 			fmt.Println("results: ", bytesReturn)
 			if query, ok := event.(*serf.Query); ok {
 				if err := query.Respond(bytesReturn); err != nil {
@@ -405,6 +416,68 @@
 	}()
 }
 
+//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 {
+	//members: get name of first member
+	mbs := a.GroupMembers(a.conf.ClusterID)
+	var specmembername string
+	for _, m := range mbs {
+		fmt.Println("m",m)
+		if m.Name != a.conf.NodeName {
+			specmembername = m.Name
+			break
+		}
+	}
+	fmt.Println("mbs:",mbs,"a.conf.BindAddr:",a.conf.BindAddr,"specmembername:",specmembername)
+
+	//query: get db file.
+	params := serf.QueryParam{
+		FilterNodes: strings.Fields(specmembername),
+	}
+
+	//SQL
+	tBytes, _ := json.Marshal(tableNames)
+
+	resp, err := a.Query(QueryEventUpdateDBData, tBytes, &params)
+	if err == nil || !strings.Contains(err.Error(), "cannot contain") {
+		fmt.Println("err: ", err)
+	}
+	fmt.Println("Query.resp.err:",err,"resp:",resp)
+
+	//go func() {
+	//	respCh := resp.ResponseCh()
+	//	for {
+	//		select {
+	//		case r := <-respCh:
+	//			fmt.Println("x length is: ", len(r.Payload))
+	//			// // byte to file.
+	//			var dumpSqls []string
+	//			err := json.Unmarshal(r.Payload, &dumpSqls)
+	//			if err ==nil {
+	//				if flag,_ := ExecuteSqlByGorm(dumpSqls);flag{
+	//					fmt.Println("data dump success")
+	//				}
+	//			}
+	//			return
+	//		}
+	//	}
+	//}()
+	return err
+
+	//r, err = c.Query([]string{query}, false, false)
+	//if err != nil {
+	//	return err
+	//}
+	//for _, x := range r[0].Values {
+	//	y := fmt.Sprintf("%s;\n", x[0].(string))
+	//	if _, err := w.Write([]byte(y)); err != nil {
+	//		return err
+	//	}
+	//}
+
+}
+
 //SyncSql boardcast sql to cluster
 func (a *Agent) SyncSql(sqlOp string) {
 	// event : use to send command to operate db.

--
Gitblit v1.8.0