🐝 Performance improvement: use sync.Pool to cache active workers

Author: panjf2000

worker.go

@@ -23,6 +23,7 @@
 package ants
 
 import (
+	"log"
 	"time"
 )
 
@@ -43,26 +44,27 @@ type Worker struct {
 // run starts a goroutine to repeat the process
 // that performs the function calls.
 func (w *Worker) run() {
+	w.pool.incRunning()
 	go func() {
 		defer func() {
 			if p := recover(); p != nil {
 				w.pool.decRunning()
 				if w.pool.PanicHandler != nil {
 					w.pool.PanicHandler(p)
 				} else {
-					panic(p)
+					log.Printf("worker exits from a panic: %v", p)
 				}
 			}
 		}()
 
 		for f := range w.task {
 			if f == nil {
 				w.pool.decRunning()
-				w.pool.cachePool.Put(w)
+				w.pool.workerCache.Put(w)
 				return
 			}
 			f()
-			w.pool.putWorker(w)
+			w.pool.revertWorker(w)
 		}
 	}()
 }

worker_func.go

@@ -23,6 +23,7 @@
 package ants
 
 import (
+	"log"
 	"time"
 )
 
@@ -43,26 +44,27 @@ type WorkerWithFunc struct {
 // run starts a goroutine to repeat the process
 // that performs the function calls.
 func (w *WorkerWithFunc) run() {
+	w.pool.incRunning()
 	go func() {
 		defer func() {
 			if p := recover(); p != nil {
 				w.pool.decRunning()
 				if w.pool.PanicHandler != nil {
 					w.pool.PanicHandler(p)
 				} else {
-					panic(p)
+					log.Printf("worker exits from a panic: %v", p)
 				}
 			}
 		}()
 
 		for args := range w.args {
 			if args == nil {
 				w.pool.decRunning()
-				w.pool.cachePool.Put(w)
+				w.pool.workerCache.Put(w)
 				return
 			}
 			w.pool.poolFunc(args)
-			w.pool.putWorker(w)
+			w.pool.revertWorker(w)
 		}
 	}()
 }