diff options
Diffstat (limited to 'go/weed')
| -rw-r--r-- | go/weed/weed_server/master_server.go | 11 | ||||
| -rw-r--r-- | go/weed/weed_server/raft_server.go | 4 |
2 files changed, 15 insertions, 0 deletions
diff --git a/go/weed/weed_server/master_server.go b/go/weed/weed_server/master_server.go index d1f7914f1..738484ff0 100644 --- a/go/weed/weed_server/master_server.go +++ b/go/weed/weed_server/master_server.go @@ -5,6 +5,7 @@ import ( "code.google.com/p/weed-fs/go/replication" "code.google.com/p/weed-fs/go/sequence" "code.google.com/p/weed-fs/go/topology" + "github.com/goraft/raft" "github.com/gorilla/mux" "net/http" "net/http/httputil" @@ -72,6 +73,16 @@ func NewMasterServer(r *mux.Router, version string, port int, metaFolder string, func (ms *MasterServer) SetRaftServer(raftServer *RaftServer) { ms.raftServer = raftServer + ms.raftServer.raftServer.AddEventListener(raft.LeaderChangeEventType, func(e raft.Event) { + ms.topo.IsLeader = ms.IsLeader() + glog.V(0).Infoln("[", ms.raftServer.Name(), "]", ms.raftServer.Leader(), "becomes leader.") + }) + ms.topo.IsLeader = ms.IsLeader() + if ms.topo.IsLeader { + glog.V(0).Infoln("[", ms.raftServer.Name(), "]", "I am the leader!") + } else { + glog.V(0).Infoln("[", ms.raftServer.Name(), "]", ms.raftServer.Leader(), "is the leader.") + } } func (ms *MasterServer) IsLeader() bool { diff --git a/go/weed/weed_server/raft_server.go b/go/weed/weed_server/raft_server.go index a44936413..b481df68d 100644 --- a/go/weed/weed_server/raft_server.go +++ b/go/weed/weed_server/raft_server.go @@ -87,6 +87,10 @@ func NewRaftServer(r *mux.Router, version string, peers []string, httpAddr strin return s } +func (s *RaftServer) Name() string { + return s.raftServer.Name() +} + func (s *RaftServer) IsLeader() bool { return s.Leader() == s.raftServer.Name() } |