naza/pkg/taskpool/taskpool_test.go

// Copyright 2019, Chef.  All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package taskpool

import (
	"sync"
	"testing"
	"time"

	"github.com/q191201771/naza/pkg/assert"
	"github.com/q191201771/naza/pkg/nazalog"
)

var (
	taskNum       = 1000 * 1000
	initWorkerNum = 1 //1000 * 20 //1000 * 10
)

func BenchmarkOriginGo(b *testing.B) {
	nazalog.Debug("> BenchmarkOriginGo")
	var wg sync.WaitGroup
	for j := 0; j < 1; j++ {
		wg.Add(taskNum)
		for i := 0; i < taskNum; i++ {
			go func() {
				time.Sleep(10 * time.Millisecond)
				wg.Done()
			}()
		}
		wg.Wait()
	}
	nazalog.Debug("< BenchmarkOriginGo")
}

func BenchmarkTaskPool(b *testing.B) {
	nazalog.Debug("> BenchmarkTaskPool")
	var wg sync.WaitGroup
	p, _ := NewPool(func(option *Option) {
		option.InitWorkerNum = initWorkerNum
	})
	//var ps []Pool
	//var poolNum = 1
	//for i := 0; i < poolNum; i++ {
	//	ps = append(ps, p)
	//}

	b.ResetTimer()
	for j := 0; j < 1; j++ {
		//b.StartTimer()
		wg.Add(taskNum)
		for i := 0; i < taskNum; i++ {
			p.Go(func() {
				time.Sleep(10 * time.Millisecond)
				wg.Done()
			})
		}
		wg.Wait()
		//b.StopTimer()
		//idle, busy := p.Status()
		//nazalog.Debugf("done, worker num. idle=%d, busy=%d", idle, busy) // 此时还有个别busy也是正常的，因为只是业务方的任务代码执行完了，可能还没回收到idle队列中
		//p.KillIdleWorkers()
		//idle, busy = p.Status()
		//nazalog.Debugf("killed, worker num. idle=%d, busy=%d", idle, busy)
	}
	nazalog.Debug("< BenchmarkTaskPool")
}

func TestTaskPool(t *testing.T) {
	var wg sync.WaitGroup
	p, _ := NewPool(func(option *Option) {
		option.InitWorkerNum = 1
	})

	go func() {
		idle, busy := p.Status()
		nazalog.Debugf("timer, worker num. idle=%d, busy=%d", idle, busy)
		time.Sleep(10 * time.Millisecond)
	}()

	n := 1000
	wg.Add(n)
	for i := 0; i < n; i++ {
		p.Go(func() {
			time.Sleep(10 * time.Millisecond)
			wg.Done()
		})
	}
	wg.Wait()
	idle, busy := p.Status()
	nazalog.Debugf("done, worker num. idle=%d, busy=%d", idle, busy) // 此时还有个别busy也是正常的，因为只是业务方的任务代码执行完了，可能还没回收到idle队列中
	p.KillIdleWorkers()
	idle, busy = p.Status()
	nazalog.Debugf("killed, worker num. idle=%d, busy=%d", idle, busy)

	time.Sleep(100 * time.Millisecond)

	wg.Add(n)
	for i := 0; i < n; i++ {
		p.Go(func() {
			time.Sleep(10 * time.Millisecond)
			wg.Done()
		})
	}
	wg.Wait()
	idle, busy = p.Status()
	nazalog.Debugf("done, worker num. idle=%d, busy=%d", idle, busy) // 此时还有个别busy也是正常的，因为只是业务方的任务代码执行完了，可能还没回收到idle队列中
}

func TestGlobal(t *testing.T) {
	err := Init()
	assert.Equal(t, nil, err)
	i, b := Status()
	assert.Equal(t, 0, i)
	assert.Equal(t, 0, b)
	Go(func() {
	})
	KillIdleWorkers()
}

func TestCorner(t *testing.T) {
	_, err := NewPool(func(option *Option) {
		option.InitWorkerNum = -1
	})
	assert.Equal(t, ErrTaskPool, err)
}