🚀 Support not only anonymous but also explicit functions

This commit is contained in:
Andy Pan 2019-02-02 10:24:01 +08:00
parent 1d33782287
commit 15507b021f
1 changed files with 2 additions and 4 deletions

View File

@ -28,8 +28,6 @@ import (
"time" "time"
) )
type f func()
// Pool accept the tasks from client,it limits the total // Pool accept the tasks from client,it limits the total
// of goroutines to a given number by recycling goroutines. // of goroutines to a given number by recycling goroutines.
type Pool struct { type Pool struct {
@ -123,7 +121,7 @@ func NewTimingPool(size, expiry int) (*Pool, error) {
//--------------------------------------------------------------------------- //---------------------------------------------------------------------------
// Submit submits a task to this pool. // Submit submits a task to this pool.
func (p *Pool) Submit(task f) error { func (p *Pool) Submit(task func()) error {
if 1 == atomic.LoadInt32(&p.release) { if 1 == atomic.LoadInt32(&p.release) {
return ErrPoolClosed return ErrPoolClosed
} }
@ -205,7 +203,7 @@ func (p *Pool) retrieveWorker() *Worker {
} else { } else {
w = &Worker{ w = &Worker{
pool: p, pool: p,
task: make(chan f, workerChanCap), task: make(chan func(), workerChanCap),
} }
} }
w.run() w.run()