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

worker: decRunning when revertWorker is false 

Signed-off-by: Cholerae Hu <choleraehyq@gmail.com>
This commit is contained in:
Cholerae Hu 2019-09-11 16:39:55 +08:00 committed by Andy Pan
parent 280ac345a8
commit d6cd5a7e72
2 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
worker.go
View File

@ -48,9 +48,8 @@ func (w *goWorker) run() {
w.pool.incRunning()
go func() {
defer func() {
if p := recover(); p != nil {
w.pool.decRunning()
w.pool.workerCache.Put(w)
if p := recover(); p != nil {
if w.pool.panicHandler != nil {
w.pool.panicHandler(p)
} else {
@ -60,17 +59,16 @@ func (w *goWorker) run() {
log.Printf("worker exits from panic: %s\n", string(buf[:n]))
}
}
w.pool.workerCache.Put(w)
}()
for f := range w.task {
if f == nil {
w.pool.decRunning()
w.pool.workerCache.Put(w)
return
}
f()
if ok := w.pool.revertWorker(w); !ok {
break
return
}
}
}()

8
worker_func.go
View File

@ -48,9 +48,8 @@ func (w *goWorkerWithFunc) run() {
w.pool.incRunning()
go func() {
defer func() {
if p := recover(); p != nil {
w.pool.decRunning()
w.pool.workerCache.Put(w)
if p := recover(); p != nil {
if w.pool.panicHandler != nil {
w.pool.panicHandler(p)
} else {
@ -60,17 +59,16 @@ func (w *goWorkerWithFunc) run() {
log.Printf("worker with func exits from panic: %s\n", string(buf[:n]))
}
}
w.pool.workerCache.Put(w)
}()
for args := range w.args {
if args == nil {
w.pool.decRunning()
w.pool.workerCache.Put(w)
return
}
w.pool.poolFunc(args)
if ok := w.pool.revertWorker(w); !ok {
break
return
}
}
}()