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

optimization for timed task to clear idle workers

This commit is contained in:
Andy Pan 2018-07-08 10:29:12 +08:00
parent 8e369baa4c
commit 2e7d823141
7 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -84,7 +84,7 @@ func main() {
// use the pool with a function // use the pool with a function
// set 10 the size of goroutine pool and 1 second for expired duration // set 10 the size of goroutine pool and 1 second for expired duration
p, _ := ants.NewPoolWithFunc(10, 1, func(i interface{}) error { p, _ := ants.NewPoolWithFunc(10, func(i interface{}) error {
myFunc(i) myFunc(i)
wg.Done() wg.Done()
return nil return nil

2
README_ZH.md
View File

@ -85,7 +85,7 @@ func main() {
// use the pool with a function // use the pool with a function
// set 10 the size of goroutine pool and 1 second for expired duration // set 10 the size of goroutine pool and 1 second for expired duration
p, _ := ants.NewPoolWithFunc(10, 1, func(i interface{}) error { p, _ := ants.NewPoolWithFunc(10, func(i interface{}) error {
myFunc(i) myFunc(i)
wg.Done() wg.Done()
return nil return nil

2
ants.go
View File

@ -36,7 +36,7 @@ const (
) )
// Init a instance pool when importing ants // Init a instance pool when importing ants
var defaultPool, _ = NewPool(DefaultPoolSize, DefaultCleanIntervalTime) var defaultPool, _ = NewPool(DefaultPoolSize)
// Submit submit a task to pool // Submit submit a task to pool
func Submit(task f) error { func Submit(task f) error {

4
ants_benchmark_test.go
View File

@ -78,7 +78,7 @@ func BenchmarkGoroutineWithFunc(b *testing.B) {
func BenchmarkAntsPoolWithFunc(b *testing.B) { func BenchmarkAntsPoolWithFunc(b *testing.B) {
var wg sync.WaitGroup var wg sync.WaitGroup
p, _ := ants.NewPoolWithFunc(50000, 1, func(i interface{}) error { p, _ := ants.NewPoolWithFunc(50000, func(i interface{}) error {
demoPoolFunc(i) demoPoolFunc(i)
wg.Done() wg.Done()
return nil return nil
@ -104,7 +104,7 @@ func BenchmarkGoroutine(b *testing.B) {
} }
func BenchmarkAntsPool(b *testing.B) { func BenchmarkAntsPool(b *testing.B) {
p, _ := ants.NewPoolWithFunc(50000, 1, demoPoolFunc) p, _ := ants.NewPoolWithFunc(50000, demoPoolFunc)
defer p.Release() defer p.Release()
b.ResetTimer() b.ResetTimer()

2
examples/main.go
View File

@ -67,7 +67,7 @@ func main() {
// use the pool with a function // use the pool with a function
// set 10 the size of goroutine pool and 1 second for expired duration // set 10 the size of goroutine pool and 1 second for expired duration
p, _ := ants.NewPoolWithFunc(10, 1, func(i interface{}) error { p, _ := ants.NewPoolWithFunc(10, func(i interface{}) error {
myFunc(i) myFunc(i)
wg.Done() wg.Done()
return nil return nil

11
pool.go
View File

@ -62,9 +62,9 @@ type Pool struct {
} }
func (p *Pool) monitorAndClear() { func (p *Pool) monitorAndClear() {
heartbeat := time.NewTicker(p.expiryDuration)
go func() { go func() {
for { for range heartbeat.C{
time.Sleep(p.expiryDuration)
currentTime := time.Now() currentTime := time.Now()
p.lock.Lock() p.lock.Lock()
idleWorkers := p.workers idleWorkers := p.workers
@ -88,7 +88,12 @@ func (p *Pool) monitorAndClear() {
} }
// NewPool generates a instance of ants pool // NewPool generates a instance of ants pool
func NewPool(size, expiry int) (*Pool, error) { func NewPool(size int) (*Pool, error) {
return NewTimingPool(size, DefaultCleanIntervalTime)
}
// NewTimingPool generates a instance of ants pool with a custom timed task
func NewTimingPool(size, expiry int) (*Pool, error) {
if size <= 0 { if size <= 0 {
return nil, ErrPoolSizeInvalid return nil, ErrPoolSizeInvalid
} }

9
pool_func.go
View File

@ -88,8 +88,13 @@ func (p *PoolWithFunc) monitorAndClear() {
}() }()
} }
// NewPoolWithFunc generates a instance of ants pool with a specific function. // NewTimingPoolWithFunc generates a instance of ants pool with a specific function and a custom timed task
func NewPoolWithFunc(size, expiry int, f pf) (*PoolWithFunc, error) { func NewPoolWithFunc(size int, f pf) (*PoolWithFunc, error) {
return NewTimingPoolWithFunc(size, DefaultCleanIntervalTime, f)
}
// NewPoolWithFunc generates a instance of ants pool with a specific function
func NewTimingPoolWithFunc(size, expiry int, f pf) (*PoolWithFunc, error) {
if size <= 0 { if size <= 0 {
return nil, ErrPoolSizeInvalid return nil, ErrPoolSizeInvalid
} }