make waffleListener a donut.MonitoredListener

dforsyth · Mar 31, 2012 · e99bb04 · e99bb04
1 parent beab6f1
commit e99bb04
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/graph.go b/graph.go
@@ -198,3 +198,7 @@ func (g *Graph) compute() {
 func (g *Graph) Write() error {
 	return g.job.Write(g)
 }
+
+func (g *Graph) information() map[string]interface{} {
+	return make(map[string]interface{})
+}
diff --git a/listener.go b/listener.go
@@ -11,7 +11,6 @@ type waffleListener struct {
 
 	clusterName string
 
-	graph   *Graph
 	done    chan byte
 	zk      *gozk.ZooKeeper
 	job     Job
@@ -22,8 +21,7 @@ type waffleListener struct {
 func (l *waffleListener) OnJoin(zk *gozk.ZooKeeper) {
 	log.Println("waffle onjoin")
 	l.zk = zk
-	l.graph = newGraph(l.job, l.coordinator)
-	l.coordinator.graph = l.graph
+	l.coordinator.graph = newGraph(l.job, l.coordinator)
 	l.coordinator.donutConfig = l.config
 	if err := l.coordinator.start(zk); err != nil {
 		l.cluster.Shutdown()
@@ -43,3 +41,7 @@ func (l *waffleListener) OnLeave() {
 		l.done <- 1
 	}()
 }
+
+func (l *waffleListener) Information() map[string]interface{} {
+	return l.coordinator.graph.information()
+}