From 7f1eb05ba827fde075d2b0d139e79e78955064e6 Mon Sep 17 00:00:00 2001
From: Vladimir Mihailenco <vladimir.webdev@gmail.com>
Date: Sun, 10 May 2015 16:01:38 +0300
Subject: [PATCH] cluster: don't reset command when there no attempts left.

---
 cluster.go      |  9 +++++++--
 cluster_test.go | 26 ++++++++++++++++++++------
 2 files changed, 27 insertions(+), 8 deletions(-)

diff --git a/cluster.go b/cluster.go
index 0ed0954..20ecbe9 100644
--- a/cluster.go
+++ b/cluster.go
@@ -130,6 +130,10 @@ func (c *ClusterClient) process(cmd Cmder) {
 	}
 
 	for attempt := 0; attempt <= c.opt.getMaxRedirects(); attempt++ {
+		if attempt > 0 {
+			cmd.reset()
+		}
+
 		if ask {
 			pipe := client.Pipeline()
 			pipe.Process(NewCmd("ASKING"))
@@ -152,7 +156,6 @@ func (c *ClusterClient) process(cmd Cmder) {
 			if err != nil {
 				return
 			}
-			cmd.reset()
 			continue
 		}
 
@@ -167,7 +170,6 @@ func (c *ClusterClient) process(cmd Cmder) {
 			if err != nil {
 				return
 			}
-			cmd.reset()
 			continue
 		}
 
@@ -282,6 +284,9 @@ type ClusterOptions struct {
 }
 
 func (opt *ClusterOptions) getMaxRedirects() int {
+	if opt.MaxRedirects == -1 {
+		return 0
+	}
 	if opt.MaxRedirects == 0 {
 		return 16
 	}
diff --git a/cluster_test.go b/cluster_test.go
index 6b06687..2ba8812 100644
--- a/cluster_test.go
+++ b/cluster_test.go
@@ -40,14 +40,16 @@ func (s *clusterScenario) slaves() []*redis.Client {
 	return result
 }
 
-func (s *clusterScenario) clusterClient() *redis.ClusterClient {
+func (s *clusterScenario) clusterClient(opt *redis.ClusterOptions) *redis.ClusterClient {
 	addrs := make([]string, len(s.ports))
 	for i, port := range s.ports {
 		addrs[i] = net.JoinHostPort("127.0.0.1", port)
 	}
-	return redis.NewClusterClient(&redis.ClusterOptions{
-		Addrs: addrs,
-	})
+	if opt == nil {
+		opt = &redis.ClusterOptions{}
+	}
+	opt.Addrs = addrs
+	return redis.NewClusterClient(opt)
 }
 
 func startCluster(scenario *clusterScenario) error {
@@ -228,7 +230,7 @@ var _ = Describe("Cluster", func() {
 		var client *redis.ClusterClient
 
 		BeforeEach(func() {
-			client = scenario.clusterClient()
+			client = scenario.clusterClient(nil)
 		})
 
 		AfterEach(func() {
@@ -301,6 +303,18 @@ var _ = Describe("Cluster", func() {
 			Expect(cmds[27].(*redis.DurationCmd).Val()).To(BeNumerically("~", 7*time.Hour, time.Second))
 		})
 
+		It("should return error when there are no attempts left", func() {
+			client = scenario.clusterClient(&redis.ClusterOptions{
+				MaxRedirects: -1,
+			})
+
+			slot := redis.HashSlot("A")
+			Expect(client.SwapSlot(slot)).To(Equal([]string{"127.0.0.1:8224", "127.0.0.1:8221"}))
+
+			err := client.Get("A").Err()
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(ContainSubstring("MOVED"))
+		})
 	})
 })
 
@@ -317,7 +331,7 @@ func BenchmarkRedisClusterPing(b *testing.B) {
 		b.Fatal(err)
 	}
 	defer stopCluster(scenario)
-	client := scenario.clusterClient()
+	client := scenario.clusterClient(nil)
 	defer client.Close()
 
 	b.ResetTimer()