From 940e9644d5b72c5083eacf28c65eb07154d87f67 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期二, 27 八月 2019 17:42:07 +0800
Subject: [PATCH] deal EventMemberLeave

---
 agent.go |   56 ++++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 42 insertions(+), 14 deletions(-)

diff --git a/agent.go b/agent.go
index 4d44c31..9081e8c 100644
--- a/agent.go
+++ b/agent.go
@@ -40,6 +40,8 @@
 const (
 	QueryEventGetDB        = "GetDatabase"
 	QueryEventUpdateDBData = "UpdateDBData"
+	UserEventSyncSql	   = "SyncSql"
+	UserEventSyncDbTablePersonCache	   = "SyncCache"
 )
 
 // Agent warps the serf agent
@@ -110,6 +112,8 @@
 	go a.BroadcastMemberlist(BroadcastInterval * time.Second)
 }
 
+var SyncDbTablePersonCacheChan = make(chan []byte,0)
+
 // HandleEvent Handles serf.EventMemberJoin events,
 // which will wait for members to join until the number of group members is equal to "groupExpect"
 // when the startup mode is "ModeCluster",
@@ -118,18 +122,22 @@
 
 	switch ev := event.(type) {
 	case serf.UserEvent:
-		//fmt.Println(string(ev.Payload))
-		var sqlUe SqlUserEvent
-		err := json.Unmarshal(ev.Payload, &sqlUe)
-		if err !=nil {
-			fmt.Println("sqlUe unmarshal err:",err)
-			return
+		if ev.Name == UserEventSyncSql {
+			var sqlUe SqlUserEvent
+			err := json.Unmarshal(ev.Payload, &sqlUe)
+			if err !=nil {
+				fmt.Println("sqlUe unmarshal err:",err)
+				return
+			}
+			if sqlUe.Owner != a.conf.NodeName {
+				//results, err := ExecuteWriteSql(sqlArr)
+				flag, _ := ExecuteSqlByGorm(sqlUe.Sql)
+				fmt.Println("userEvent exec ",sqlUe.Sql,",Result:",flag)
+			}
+		} else if ev.Name == UserEventSyncDbTablePersonCache {
+			SyncDbTablePersonCacheChan <- ev.Payload
 		}
-		if sqlUe.Owner != a.conf.NodeName {
-			//results, err := ExecuteWriteSql(sqlArr)
-			flag, _ := ExecuteSqlByGorm(sqlUe.Sql)
-			fmt.Println("userEvent exec ",sqlUe.Sql,",Result:",flag)
-		}
+
 
 	case *serf.Query:
 
@@ -200,6 +208,18 @@
 			//var res []*Rows
 			//json.Unmarshal(bytesReturn, &res)
 		}
+	case serf.MemberEvent:
+		if event.EventType() == serf.EventMemberLeave {
+			if ev.Members !=nil && len(ev.Members) ==1 {
+				leaveMember := ev.Members[0]
+				leaveSql := "delete from cluster_node where node_id='"+leaveMember.Name+"'"
+				ExecuteSqlByGorm([]string{ leaveSql })
+
+				fmt.Println("EventMemberLeave,current Members:",ev.Members)
+			}
+			return
+		}
+
 
 	default:
 		fmt.Printf("Unknown event type: %s\n", ev.EventType().String())
@@ -501,9 +521,17 @@
 		fmt.Println("sqlUE marshal err:",err)
 		return
 	}
-	err = a.UserEvent("SyncSql", ueB, false)
+	err = a.UserEvent(UserEventSyncSql, ueB, false)
 	if err == nil || !strings.Contains(err.Error(), "cannot contain") {
 		fmt.Println("err: ", err)
+	}
+}
+
+//鏇存柊鍚屾搴撶殑姣斿缂撳瓨
+func (a *Agent) SyncDbTablePersonCache(b []byte) {
+	err := a.UserEvent(UserEventSyncDbTablePersonCache, b, false)
+	if err !=nil{
+		fmt.Println("UserEventSyncDbTablePersonCache err:",err)
 	}
 }
 
@@ -573,8 +601,8 @@
 
 	n, err := a.Agent.Join(nodes, true)
 	if err != nil || n == 0 {
-		a.Stop()
-		fmt.Println("Stop node")
+		//a.Stop()
+		//fmt.Println("Stop node")
 		return fmt.Errorf("Error Encrypt Key!")
 	}
 

--
Gitblit v1.8.0