forked from mirror/redis
Merge pull request #201 from go-redis/fix/sporadic-pool-timeout
Fix sporadic pool timeouts with IdleTimeout != 0. Fixes #195.
This commit is contained in:
commit
54a9acc11f
|
@ -146,6 +146,7 @@ func (c *ClusterClient) process(cmd Cmder) {
|
||||||
pipe.Process(NewCmd("ASKING"))
|
pipe.Process(NewCmd("ASKING"))
|
||||||
pipe.Process(cmd)
|
pipe.Process(cmd)
|
||||||
_, _ = pipe.Exec()
|
_, _ = pipe.Exec()
|
||||||
|
pipe.Close()
|
||||||
ask = false
|
ask = false
|
||||||
} else {
|
} else {
|
||||||
client.Process(cmd)
|
client.Process(cmd)
|
||||||
|
|
36
pool.go
36
pool.go
|
@ -163,8 +163,12 @@ func (p *connPool) First() *conn {
|
||||||
select {
|
select {
|
||||||
case cn := <-p.freeConns:
|
case cn := <-p.freeConns:
|
||||||
if p.isIdle(cn) {
|
if p.isIdle(cn) {
|
||||||
p.conns.Remove(cn)
|
var err error
|
||||||
continue
|
cn, err = p.replace(cn)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("redis: replace failed: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return cn
|
return cn
|
||||||
default:
|
default:
|
||||||
|
@ -181,8 +185,12 @@ func (p *connPool) wait() *conn {
|
||||||
select {
|
select {
|
||||||
case cn := <-p.freeConns:
|
case cn := <-p.freeConns:
|
||||||
if p.isIdle(cn) {
|
if p.isIdle(cn) {
|
||||||
p.Remove(cn)
|
var err error
|
||||||
continue
|
cn, err = p.replace(cn)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("redis: replace failed: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return cn
|
return cn
|
||||||
case <-deadline:
|
case <-deadline:
|
||||||
|
@ -257,16 +265,24 @@ func (p *connPool) Put(cn *conn) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *connPool) Remove(cn *conn) error {
|
func (p *connPool) replace(cn *conn) (*conn, error) {
|
||||||
// Replace existing connection with new one and unblock waiter.
|
|
||||||
newcn, err := p.new()
|
newcn, err := p.new()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("redis: new failed: %s", err)
|
_ = p.conns.Remove(cn)
|
||||||
return 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
|
p.freeConns <- newcn
|
||||||
return err
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Len returns total number of connections.
|
// Len returns total number of connections.
|
||||||
|
|
Loading…
Reference in New Issue