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

🦠 Adapt code for explicit functions

This commit is contained in:
Andy Pan 2019-02-02 10:32:51 +08:00
parent b49328d4e6
commit 64b763c558
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pool_func.go
View File

@ -28,8 +28,6 @@ import (
"time" "time"
) )
type pf func(interface{})
// PoolWithFunc accept the tasks from client,it limits the total // PoolWithFunc 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 PoolWithFunc struct { type PoolWithFunc struct {
@ -54,8 +52,8 @@ type PoolWithFunc struct {
// cond for waiting to get a idle worker. // cond for waiting to get a idle worker.
cond *sync.Cond cond *sync.Cond
// pf is the function for processing tasks. // poolFunc is the function for processing tasks.
poolFunc pf poolFunc func(interface{})
// once makes sure releasing this pool will just be done for one time. // once makes sure releasing this pool will just be done for one time.
once sync.Once once sync.Once
@ -102,12 +100,12 @@ func (p *PoolWithFunc) periodicallyPurge() {
} }
// NewPoolWithFunc generates an instance of ants pool with a specific function. // NewPoolWithFunc generates an instance of ants pool with a specific function.
func NewPoolWithFunc(size int, f pf) (*PoolWithFunc, error) { func NewPoolWithFunc(size int, pf func(interface{})) (*PoolWithFunc, error) {
return NewTimingPoolWithFunc(size, DefaultCleanIntervalTime, f) return NewTimingPoolWithFunc(size, DefaultCleanIntervalTime, pf)
} }
// NewTimingPoolWithFunc generates an instance of ants pool with a specific function and a custom timed task. // NewTimingPoolWithFunc generates an instance of ants pool with a specific function and a custom timed task.
func NewTimingPoolWithFunc(size, expiry int, f pf) (*PoolWithFunc, error) { func NewTimingPoolWithFunc(size, expiry int, pf func(interface{})) (*PoolWithFunc, error) {
if size <= 0 { if size <= 0 {
return nil, ErrInvalidPoolSize return nil, ErrInvalidPoolSize
} }
@ -117,7 +115,7 @@ func NewTimingPoolWithFunc(size, expiry int, f pf) (*PoolWithFunc, error) {
p := &PoolWithFunc{ p := &PoolWithFunc{
capacity: int32(size), capacity: int32(size),
expiryDuration: time.Duration(expiry) * time.Second, expiryDuration: time.Duration(expiry) * time.Second,
poolFunc: f, poolFunc: pf,
} }
p.cond = sync.NewCond(&p.lock) p.cond = sync.NewCond(&p.lock)
go p.periodicallyPurge() go p.periodicallyPurge()