release the pool in testing

This commit is contained in:
andy pan 2018-07-02 14:35:02 +08:00
parent 926f57c696
commit cca93c8dfc
2 changed files with 3 additions and 2 deletions

View File

@ -77,13 +77,14 @@ func BenchmarkGoroutineWithFunc(b *testing.B) {
} }
func BenchmarkAntsPoolWithFunc(b *testing.B) { func BenchmarkAntsPoolWithFunc(b *testing.B) {
defer ants.Release()
var wg sync.WaitGroup var wg sync.WaitGroup
p, _ := ants.NewPoolWithFunc(50000, func(i interface{}) error { p, _ := ants.NewPoolWithFunc(50000, func(i interface{}) error {
demoPoolFunc(i) demoPoolFunc(i)
wg.Done() wg.Done()
return nil return nil
}) })
p.Release()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
for j := 0; j < RunTimes; j++ { for j := 0; j < RunTimes; j++ {
wg.Add(1) wg.Add(1)

View File

@ -33,6 +33,7 @@ import (
var n = 1000000 var n = 1000000
func TestDefaultPool(t *testing.T) { func TestDefaultPool(t *testing.T) {
defer ants.Release()
var wg sync.WaitGroup var wg sync.WaitGroup
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
wg.Add(1) wg.Add(1)
@ -54,7 +55,6 @@ func TestDefaultPool(t *testing.T) {
} }
func TestNoPool(t *testing.T) { func TestNoPool(t *testing.T) {
defer ants.Release()
var wg sync.WaitGroup var wg sync.WaitGroup
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
wg.Add(1) wg.Add(1)