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 |   82 +++++++++++++++++++++++++---------------
 1 files changed, 51 insertions(+), 31 deletions(-)

diff --git a/agent.go b/agent.go
index 1994157..fe65539 100644
--- a/agent.go
+++ b/agent.go
@@ -81,12 +81,11 @@
 		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)
-
-	ltLock.Lock()
-	curLTime = QueryLTimeFromDbByGorm()
-	ltLock.Unlock()
 
 	return &Agent{
 		Agent:   serfAgent,
@@ -118,8 +117,6 @@
 }
 
 var SyncDbTablePersonCacheChan = make(chan []byte,0)
-var curLTime uint64
-var ltLock sync.RWMutex
 
 // HandleEvent Handles serf.EventMemberJoin events,
 // which will wait for members to join until the number of group members is equal to "groupExpect"
@@ -129,8 +126,6 @@
 
 	switch ev := event.(type) {
 	case serf.UserEvent:
-		ltLock.Lock()
-		defer ltLock.Unlock()
 		if ev.Name == UserEventSyncSql {
 			var sqlUe SqlUserEvent
 			err := json.Unmarshal(ev.Payload, &sqlUe)
@@ -139,19 +134,12 @@
 				return
 			}
 			if sqlUe.Owner != a.conf.NodeName {
+
 				evTime := uint64(ev.LTime)
-				logger.Info("ev.LTime:",evTime,",curLTime:",curLTime,",SqlUserEvent.sql:",sqlUe.Sql)
-				if curLTime !=0 && evTime < curLTime{//鏄鐞嗚繃鐨勪簨浠�
-					logger.Info("already executed event,ev.LTime:",evTime,"SqlUserEvent.sql:",sqlUe.Sql)
-					return
-				}
+				logger.Info("ev.LTime:",evTime,",SqlUserEvent.sql:",sqlUe.Sql)
+
 				flag, _ := ExecuteSqlByGorm(sqlUe.Sql)
 				logger.Info("userEvent exec ",sqlUe.Sql,",Result:",flag)
-				if flag {
-					curLTime = evTime
-
-					ExecuteSqlByGorm([]string{"update sync_serf set lamport_time='"+strconv.FormatUint(curLTime,10)+"'"})
-				}
 			}
 		} else if ev.Name == UserEventSyncDbTablePersonCache {
 			logger.Info("LTime:",ev.LTime,",ev.Payload.len:",len(ev.Payload))
@@ -204,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 {
@@ -460,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) {
@@ -487,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") {
@@ -496,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()
@@ -512,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