re
/
ants
forked from mirror/ants
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

📚Refactor the blocking logic in retrieveWorker function

This commit is contained in:
Andy Pan 2019-07-27 13:08:09 +08:00
parent 9cfc9fa643
commit 61660e2109
2 changed files with 18 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pool.go
View File

@ -211,7 +211,6 @@ func (p *Pool) retrieveWorker() *Worker {
p.lock.Lock()
idleWorkers := p.workers
n := len(idleWorkers) - 1
RESUME:
if n >= 0 {
w = idleWorkers[n]
idleWorkers[n] = nil
@ -229,21 +228,16 @@ RESUME:
}
w.run()
} else {
for {
if p.Running() == 0 {
goto RESUME
}
p.cond.Wait()
l := len(p.workers) - 1
if l < 0 {
continue
}
w = p.workers[l]
p.workers[l] = nil
p.workers = p.workers[:l]
break
Reentry:
p.cond.Wait()
l := len(p.workers) - 1
if l < 0 {
goto Reentry
}
p.lock.Unlock()
w = p.workers[l]
p.workers[l] = nil
p.workers = p.workers[:l]
p.lock.Unlock()
}
return w
}

24
pool_func.go
View File

@ -216,7 +216,6 @@ func (p *PoolWithFunc) retrieveWorker() *WorkerWithFunc {
p.lock.Lock()
idleWorkers := p.workers
n := len(idleWorkers) - 1
RESUME:
if n >= 0 {
w = idleWorkers[n]
idleWorkers[n] = nil
@ -234,21 +233,16 @@ RESUME:
}
w.run()
} else {
for {
if p.Running() == 0 {
goto RESUME
}
p.cond.Wait()
l := len(p.workers) - 1
if l < 0 {
continue
}
w = p.workers[l]
p.workers[l] = nil
p.workers = p.workers[:l]
break
Reentry:
p.cond.Wait()
l := len(p.workers) - 1
if l < 0 {
goto Reentry
}
p.lock.Unlock()
w = p.workers[l]
p.workers[l] = nil
p.workers = p.workers[:l]
p.lock.Unlock()
}
return w
}