package GPool

import (
	"sync"
	"testing"
)

func BenchmarkGoPool(b *testing.B) {
	pool := NewGoPool(10)
	for i := 0; i < 1000; i++ {
		for j := 0; j < 1000; j++ {
			go pool.AddTask(add, []interface{}{i, j})
		}
	}
	b.StartTimer()
	pool.Run()
	pool.Wait()
	pool.Close()
	b.StopTimer()
}

func BenchmarkGoroutine(b *testing.B) {
	a := sync.WaitGroup{}
	b.StartTimer()
	for i := 0; i < 10000; i++ {
		for j := 0; j < 1000; j++ {
			a.Add(1)
			go func(i, j int) {
				add(i, j)
				a.Done()
			}(i, j)
		}
	}
	a.Wait()
	b.StopTimer()
}

func BenchmarkGoPoolForChain(b *testing.B) {
	pool := NewPoolForTaskChain(1000)
	for i := 0; i < 1000; i++ {
		for j := 0; j < 1000; j++ {
			pool.AddTaskToTQ(add, []interface{}{i, j})
		}
	}
	b.StartTimer()
	pool.Run_taskChain()
	b.StopTimer()
}

func BenchmarkWorkPool(b *testing.B) {
	pool := NewWorkPool(10000)
	var wg sync.WaitGroup
	b.StartTimer()
	for i := 0; i < 1000; i++ {
		for j := 0; j < 1000; j++ {
			wg.Add(1)
			np := &sub{
				//a: i,
				//b : j,
			}
			go func() {
				pool.Run(np)
				wg.Done()
			}()
		}
	}
	wg.Wait()
	pool.Close()
	b.StopTimer()
}

func BenchmarkOriginGoroutine(b *testing.B) {
	var wg sync.WaitGroup
	for i := 0; i < 1000; i++ {
		for j := 0; j < 1000; j++ {
			wg.Add(1)
			np := &sub{
				//a: i,
				//b : j,
			}
			go func() {
				np.Do()
				wg.Done()
			}()
		}
	}
	wg.Wait()
}