From 622f52952028ce4e6c6620b025184e322be43b74 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期四, 08 八月 2019 13:26:37 +0800
Subject: [PATCH] Merge branch 'master' of http://192.168.1.14:10010/r/syncdb

---
 agent.go |   95 +++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 83 insertions(+), 12 deletions(-)

diff --git a/agent.go b/agent.go
index 78af264..70ab249 100644
--- a/agent.go
+++ b/agent.go
@@ -159,21 +159,31 @@
 				}
 			}
 		} 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)
+			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 +415,67 @@
 	}()
 }
 
+//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 {
+		if m.Addr.String() != a.conf.BindAddr {
+			specmembername = m.Name
+			break
+		}
+	}
+	fmt.Println(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)
+	}
+
+
+	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 nil
+
+	//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