Merge branch 'develop'

This commit is contained in:
Andy Pan 2018-07-24 22:30:37 +08:00
parent 5776f41c0d
commit 634682856f
2 changed files with 48 additions and 52 deletions

50
pool.go
View File

@ -63,32 +63,30 @@ type Pool struct {
func (p *Pool) periodicallyPurge() {
heartbeat := time.NewTicker(p.expiryDuration)
go func() {
for range heartbeat.C {
currentTime := time.Now()
p.lock.Lock()
idleWorkers := p.workers
if len(idleWorkers) == 0 && p.Running() == 0 && len(p.release) > 0 {
p.lock.Unlock()
return
}
n := 0
for i, w := range idleWorkers {
if currentTime.Sub(w.recycleTime) <= p.expiryDuration {
break
}
n = i
<-p.freeSignal
w.task <- nil
idleWorkers[i] = nil
}
if n > 0 {
n++
p.workers = idleWorkers[n:]
}
for range heartbeat.C {
currentTime := time.Now()
p.lock.Lock()
idleWorkers := p.workers
if len(idleWorkers) == 0 && p.Running() == 0 && len(p.release) > 0 {
p.lock.Unlock()
return
}
}()
n := 0
for i, w := range idleWorkers {
if currentTime.Sub(w.recycleTime) <= p.expiryDuration {
break
}
n = i
<-p.freeSignal
w.task <- nil
idleWorkers[i] = nil
}
if n > 0 {
n++
p.workers = idleWorkers[n:]
}
p.lock.Unlock()
}
}
// NewPool generates a instance of ants pool
@ -110,7 +108,7 @@ func NewTimingPool(size, expiry int) (*Pool, error) {
release: make(chan sig, 1),
expiryDuration: time.Duration(expiry) * time.Second,
}
p.periodicallyPurge()
go p.periodicallyPurge()
return p, nil
}
@ -208,7 +206,7 @@ func (p *Pool) getWorker() *Worker {
} else if w == nil {
w = &Worker{
pool: p,
task: make(chan f),
task: make(chan f, 1),
}
w.run()
}

View File

@ -64,32 +64,30 @@ type PoolWithFunc struct {
func (p *PoolWithFunc) periodicallyPurge() {
heartbeat := time.NewTicker(p.expiryDuration)
go func() {
for range heartbeat.C {
currentTime := time.Now()
p.lock.Lock()
idleWorkers := p.workers
if len(idleWorkers) == 0 && p.Running() == 0 && len(p.release) > 0 {
p.lock.Unlock()
return
}
n := 0
for i, w := range idleWorkers {
if currentTime.Sub(w.recycleTime) <= p.expiryDuration {
break
}
n = i
<-p.freeSignal
w.args <- nil
idleWorkers[i] = nil
}
if n > 0 {
n++
p.workers = idleWorkers[n:]
}
for range heartbeat.C {
currentTime := time.Now()
p.lock.Lock()
idleWorkers := p.workers
if len(idleWorkers) == 0 && p.Running() == 0 && len(p.release) > 0 {
p.lock.Unlock()
return
}
}()
n := 0
for i, w := range idleWorkers {
if currentTime.Sub(w.recycleTime) <= p.expiryDuration {
break
}
n = i
<-p.freeSignal
w.args <- nil
idleWorkers[i] = nil
}
if n > 0 {
n++
p.workers = idleWorkers[n:]
}
p.lock.Unlock()
}
}
// NewPoolWithFunc generates a instance of ants pool with a specific function
@ -112,7 +110,7 @@ func NewTimingPoolWithFunc(size, expiry int, f pf) (*PoolWithFunc, error) {
expiryDuration: time.Duration(expiry) * time.Second,
poolFunc: f,
}
p.periodicallyPurge()
go p.periodicallyPurge()
return p, nil
}
@ -213,7 +211,7 @@ func (p *PoolWithFunc) getWorker() *WorkerWithFunc {
} else if w == nil {
w = &WorkerWithFunc{
pool: p,
args: make(chan interface{}),
args: make(chan interface{}, 1),
}
w.run()
}