mirror of https://github.com/go-redis/redis.git
Merge pull request #586 from go-redis/fix/pool-fast-path
pool: add fast path
This commit is contained in:
commit
c8159532bc
|
@ -106,6 +106,9 @@ func (p *ConnPool) NewConn() (*Conn, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *ConnPool) PopFree() *Conn {
|
func (p *ConnPool) PopFree() *Conn {
|
||||||
|
select {
|
||||||
|
case p.queue <- struct{}{}:
|
||||||
|
default:
|
||||||
timer := timers.Get().(*time.Timer)
|
timer := timers.Get().(*time.Timer)
|
||||||
timer.Reset(p.poolTimeout)
|
timer.Reset(p.poolTimeout)
|
||||||
|
|
||||||
|
@ -120,6 +123,7 @@ func (p *ConnPool) PopFree() *Conn {
|
||||||
atomic.AddUint32(&p.stats.Timeouts, 1)
|
atomic.AddUint32(&p.stats.Timeouts, 1)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
p.freeConnsMu.Lock()
|
p.freeConnsMu.Lock()
|
||||||
cn := p.popFree()
|
cn := p.popFree()
|
||||||
|
@ -150,6 +154,9 @@ func (p *ConnPool) Get() (*Conn, bool, error) {
|
||||||
|
|
||||||
atomic.AddUint32(&p.stats.Requests, 1)
|
atomic.AddUint32(&p.stats.Requests, 1)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case p.queue <- struct{}{}:
|
||||||
|
default:
|
||||||
timer := timers.Get().(*time.Timer)
|
timer := timers.Get().(*time.Timer)
|
||||||
timer.Reset(p.poolTimeout)
|
timer.Reset(p.poolTimeout)
|
||||||
|
|
||||||
|
@ -164,6 +171,7 @@ func (p *ConnPool) Get() (*Conn, bool, error) {
|
||||||
atomic.AddUint32(&p.stats.Timeouts, 1)
|
atomic.AddUint32(&p.stats.Timeouts, 1)
|
||||||
return nil, false, ErrPoolTimeout
|
return nil, false, ErrPoolTimeout
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
for {
|
for {
|
||||||
p.freeConnsMu.Lock()
|
p.freeConnsMu.Lock()
|
||||||
|
|
Loading…
Reference in New Issue