From 9e9867a0f667b77648ac4a7881f7a596c363536e Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期二, 08 十月 2019 21:04:07 +0800
Subject: [PATCH] tcp receive data from chan

---
 agent.go |   67 ++++++++++++++++++++++++++-------
 1 files changed, 53 insertions(+), 14 deletions(-)

diff --git a/agent.go b/agent.go
index 7483a81..fe65539 100644
--- a/agent.go
+++ b/agent.go
@@ -81,6 +81,9 @@
 		return nil, err
 	}
 	serfConf.MemberlistConfig.Keyring = keyring
+
+	serfConf.MemberlistConfig.Delegate = &UserDelegate{}
+
 	logger.Info("[INFO] agent: Restored keyring with %d keys from %s",
 		len(conf.EncryptKey), conf.EncryptKey)
 
@@ -131,11 +134,15 @@
 				return
 			}
 			if sqlUe.Owner != a.conf.NodeName {
-				//results, err := ExecuteWriteSql(sqlArr)
+
+				evTime := uint64(ev.LTime)
+				logger.Info("ev.LTime:",evTime,",SqlUserEvent.sql:",sqlUe.Sql)
+
 				flag, _ := ExecuteSqlByGorm(sqlUe.Sql)
 				logger.Info("userEvent exec ",sqlUe.Sql,",Result:",flag)
 			}
 		} else if ev.Name == UserEventSyncDbTablePersonCache {
+			logger.Info("LTime:",ev.LTime,",ev.Payload.len:",len(ev.Payload))
 			SyncDbTablePersonCacheChan <- ev.Payload
 		}
 
@@ -185,29 +192,46 @@
 			//for _, r := range rows {
 			//	rowsReturn = append(rowsReturn, *r)
 			//}
-			var tableNames []string
-			err := json.Unmarshal(ev.Payload, &tableNames)
+			var fromP QueryTableDataParam
+			err := json.Unmarshal(ev.Payload, &fromP)
 			if err !=nil {
 				logger.Error("Query tableNames unmarshal err")
 				return
 			}
-			logger.Info("Query tableNames:",tableNames)
-			datas, err := ExecuteQueryByGorm(tableNames)
+			logger.Info("Query tableNames:",fromP.Tables)
+			datas, err := ExecuteQueryByGorm(fromP.Tables)
 			if err !=nil {
 				logger.Error("queryByGorm err")
 				return
 			}
 			bytesReturn, err := json.Marshal(datas)
 			logger.Info("results.len: ", len(bytesReturn))
-			if query, ok := event.(*serf.Query); ok {
-				if err := query.Respond(bytesReturn); err != nil {
-					logger.Error("err: %s\n", err)
-					return
+
+			var targetNode *memberlist.Node
+			nodes := a.Serf().Memberlist().Members()
+			if nodes != nil && len(nodes) > 0 {
+				for _,n :=range nodes {
+					if n.Name == fromP.From {
+						targetNode = n
+						break
+					}
 				}
 			}
+			if targetNode !=nil {
+				sendErr := a.Serf().Memberlist().SendToTCP(targetNode, bytesReturn)
+				if sendErr != nil {
+					logger.Debug("sendToTcp err:",sendErr)
+				}
+			} else {
+				logger.Debug("targetNode is nil")
+			}
 
-			//var res []*Rows
-			//json.Unmarshal(bytesReturn, &res)
+			//if query, ok := event.(*serf.Query); ok {
+			//	if err := query.Respond(bytesReturn); err != nil {
+			//		logger.Error("err: %s\n", err)
+			//		return
+			//	}
+			//}
 		}
 	case serf.MemberEvent:
 		if event.EventType() == serf.EventMemberLeave {
@@ -441,6 +465,12 @@
 	}()
 }
 
+type QueryTableDataParam struct {
+	Tables []string `json:"tables"`
+	From string `json:"from"`
+}
+
+var QueryTcpResponseChan = make(chan []byte)
 //GetDbFromCluster get the newest database after join cluster
 //dbPathWrite the path where to write after got a database,
 func (a *Agent) GetTableDataFromCluster(tableNames []string) (*[]string,error) {
@@ -468,8 +498,12 @@
 		FilterNodes: strings.Fields(specmembername),
 	}
 
-	//SQL
-	tBytes, _ := json.Marshal(tableNames)
+	//get db tables
+	var fromP = QueryTableDataParam{
+		Tables: tableNames,
+		From: a.conf.NodeName,
+	}
+	tBytes, _ := json.Marshal(fromP)
 
 	resp, err := a.Query(QueryEventUpdateDBData, tBytes, &params)
 	if err == nil || !strings.Contains(err.Error(), "cannot contain") {
@@ -477,9 +511,10 @@
 	}
 	logger.Info("Query.resp.err:",err,"resp:",resp)
 
+	var dumpSqls []string
+
 	var wg sync.WaitGroup
 	wg.Add(1)
-	var dumpSqls []string
 	go func() {
 		defer wg.Done()
 		respCh := resp.ResponseCh()
@@ -493,7 +528,11 @@
 					logger.Error("data dump success")
 				}
 				return
+			case msg := <-QueryTcpResponseChan:
+				logger.Debug("QueryTcpResponseChan receive msg len:",len(msg))
+				return
 			}
+
 		}
 	}()
 	wg.Wait()

--
Gitblit v1.8.0