From 2066ff5acb3b64f888b481ab5071589c530be4f9 Mon Sep 17 00:00:00 2001
From: Xiang Li <xiang.li@coreos.com>
Date: Wed, 28 May 2014 11:55:07 -0700
Subject: [PATCH] raft: candidate should become leader directly if it is a one
 node cluster

---
 raft/raft.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/raft/raft.go b/raft/raft.go
index 9c5744bae..9a5f87814 100644
--- a/raft/raft.go
+++ b/raft/raft.go
@@ -204,7 +204,6 @@ func (sm *stateMachine) becomeCandidate() {
 	sm.term++
 	sm.vote = sm.addr
 	sm.state = stateCandidate
-	sm.poll(sm.addr, true)
 }
 
 func (sm *stateMachine) becomeLeader() {
@@ -228,6 +227,10 @@ func (sm *stateMachine) Step(m Message) {
 	switch m.Type {
 	case msgHup:
 		sm.becomeCandidate()
+		if sm.q() == sm.poll(sm.addr, true) {
+			sm.becomeLeader()
+			return
+		}
 		for i := 0; i < sm.k; i++ {
 			if i == sm.addr {
 				continue