From c41ef86c3cb3096f17c4cc6a9ef6abff38409e36 Mon Sep 17 00:00:00 2001
From: Can ZHANG <zhangcan@letv.com>
Date: Tue, 7 Jul 2015 19:29:30 +0800
Subject: [PATCH] Remove unnecessary timeout in WaitForLeader()

If the timer expires and a leader is still not selected, the local data node will never be created.
---
 meta/store.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/meta/store.go b/meta/store.go
index 985ba5dba10..5a67e0efc15 100644
--- a/meta/store.go
+++ b/meta/store.go
@@ -352,7 +352,7 @@ func (s *Store) init() {
 // Writes the id of the node to file on success.
 func (s *Store) createLocalNode() error {
 	// Wait for leader.
-	if err := s.WaitForLeader(5 * time.Second); err != nil {
+	if err := s.WaitForLeader(0); err != nil {
 		return fmt.Errorf("wait for leader: %s", err)
 	}
 
@@ -382,6 +382,7 @@ func (s *Store) Snapshot() error {
 }
 
 // WaitForLeader sleeps until a leader is found or a timeout occurs.
+// timeout == 0 means to wait forever.
 func (s *Store) WaitForLeader(timeout time.Duration) error {
 	if s.raft.Leader() != "" {
 		return nil
@@ -399,7 +400,9 @@ func (s *Store) WaitForLeader(timeout time.Duration) error {
 		case <-s.closing:
 			return errors.New("closing")
 		case <-timer.C:
-			return errors.New("timeout")
+			if timeout != 0 {
+				return errors.New("timeout")
+			}
 		case <-ticker.C:
 			if s.raft.Leader() != "" {
 				return nil