From fb44c891ddccc854c418bca59d23bc58f751761a Mon Sep 17 00:00:00 2001
From: Vladimir Mihailenco <vladimir.webdev@gmail.com>
Date: Fri, 27 Nov 2015 13:35:30 +0200
Subject: [PATCH] Fix sporadic pool timeouts with IdleTimeout != 0. Fixes #195.

---
 cluster.go |  1 +
 pool.go    | 36 ++++++++++++++++++++++++++----------
 2 files changed, 27 insertions(+), 10 deletions(-)

diff --git a/cluster.go b/cluster.go
index 463cae4..ab06890 100644
--- a/cluster.go
+++ b/cluster.go
@@ -146,6 +146,7 @@ func (c *ClusterClient) process(cmd Cmder) {
 			pipe.Process(NewCmd("ASKING"))
 			pipe.Process(cmd)
 			_, _ = pipe.Exec()
+			pipe.Close()
 			ask = false
 		} else {
 			client.Process(cmd)
diff --git a/pool.go b/pool.go
index 03bb1a0..2c387e9 100644
--- a/pool.go
+++ b/pool.go
@@ -163,8 +163,12 @@ func (p *connPool) First() *conn {
 		select {
 		case cn := <-p.freeConns:
 			if p.isIdle(cn) {
-				p.conns.Remove(cn)
-				continue
+				var err error
+				cn, err = p.replace(cn)
+				if err != nil {
+					log.Printf("redis: replace failed: %s", err)
+					continue
+				}
 			}
 			return cn
 		default:
@@ -181,8 +185,12 @@ func (p *connPool) wait() *conn {
 		select {
 		case cn := <-p.freeConns:
 			if p.isIdle(cn) {
-				p.Remove(cn)
-				continue
+				var err error
+				cn, err = p.replace(cn)
+				if err != nil {
+					log.Printf("redis: replace failed: %s", err)
+					continue
+				}
 			}
 			return cn
 		case <-deadline:
@@ -257,16 +265,24 @@ func (p *connPool) Put(cn *conn) error {
 	return nil
 }
 
-func (p *connPool) Remove(cn *conn) error {
-	// Replace existing connection with new one and unblock waiter.
+func (p *connPool) replace(cn *conn) (*conn, error) {
 	newcn, err := p.new()
 	if err != nil {
-		log.Printf("redis: new failed: %s", err)
-		return p.conns.Remove(cn)
+		_ = p.conns.Remove(cn)
+		return nil, err
+	}
+	_ = p.conns.Replace(cn, newcn)
+	return newcn, nil
+}
+
+func (p *connPool) Remove(cn *conn) error {
+	// Replace existing connection with new one and unblock waiter.
+	newcn, err := p.replace(cn)
+	if err != nil {
+		return err
 	}
-	err = p.conns.Replace(cn, newcn)
 	p.freeConns <- newcn
-	return err
+	return nil
 }
 
 // Len returns total number of connections.