Start gossip agent in a separate goroutine

BBVA · Nov 29, 2018 · 91e6507 · 91e6507
1 parent 6b47c3e
commit 91e6507
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/cmd/agent_auditor.go b/cmd/agent_auditor.go
@@ -50,7 +50,6 @@ func newAgentAuditorCommand(ctx *agentContext) *cobra.Command {
 			}
 			log.Debugf("Number of nodes contacted: %d (%v)", contacted, agentConfig.StartJoin)
 
-			agent.Start()
 			defer agent.Shutdown()
 			util.AwaitTermSignal(agent.Leave)
 		},

diff --git a/cmd/agent_monitor.go b/cmd/agent_monitor.go
@@ -58,7 +58,6 @@ func newAgentMonitorCommand(ctx *agentContext) *cobra.Command {
 			}
 			log.Debugf("Number of nodes contacted: %d", contacted)
 
-			agent.Start()
 			defer agent.Shutdown()
 			util.AwaitTermSignal(agent.Leave)
 		},

diff --git a/cmd/agent_publisher.go b/cmd/agent_publisher.go
@@ -52,7 +52,6 @@ func newAgentPublisherCommand(ctx *agentContext) *cobra.Command {
 			}
 			log.Debugf("Number of nodes contacted: %d", contacted)
 
-			agent.Start()
 			defer agent.Shutdown()
 			util.AwaitTermSignal(agent.Leave)
 		},

diff --git a/gossip/agent.go b/gossip/agent.go
@@ -104,10 +104,14 @@ func NewAgent(conf *Config, p []Processor) (agent *Agent, err error) {
 		RetransmitMult: 2,
 	}
 
+	if p != nil {
+		go agent.start()
+	}
+
 	return agent, nil
 }
 
-func (a *Agent) Start() {
+func (a *Agent) start() {
 
 	outTicker := time.NewTicker(2 * time.Second)
 	alertTicker := time.NewTicker(1 * time.Second)

diff --git a/gossip/agent_test.go b/gossip/agent_test.go
@@ -9,10 +9,6 @@ import (
 	"github.com/bbva/qed/gossip/member"
 )
 
-func TestStart(t *testing.T) {
-
-}
-
 func TestJoin(t *testing.T) {
 	conf := DefaultConfig()
 	conf.NodeName = "testNode"

diff --git a/gossip/delegate.go b/gossip/delegate.go
@@ -48,6 +48,9 @@ func (e *eventDelegate) NotifyLeave(n *memberlist.Node) {
 // updated, usually involving the meta data.
 func (e *eventDelegate) NotifyUpdate(n *memberlist.Node) {
 	// ignore
+	peer := member.ParsePeer(n)
+	e.agent.Topology.Update(peer)
+	log.Debugf("member updated: %+v ", peer)
 }
 
 type agentDelegate struct {