From 5dace354b740cdb05cdbfc9f4da699844b38a379 Mon Sep 17 00:00:00 2001 From: Evan Hazlett Date: Sat, 9 Jun 2018 13:28:25 -0400 Subject: [PATCH] add subscribe for node events Signed-off-by: Evan Hazlett --- agent.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/agent.go b/agent.go index 0581d65..0bab358 100644 --- a/agent.go +++ b/agent.go @@ -50,3 +50,15 @@ func NewAgent(cfg *Config) (*Agent, error) { grpcServer: grpcServer, }, nil } + +// Subscribe subscribes to the node event channel +func (a *Agent) Subscribe(ch chan *NodeEvent) { + go func() { + for { + select { + case evt := <-a.nodeEventChan: + ch <- evt + } + } + }() +}