conrenttest.go

package utils

import (
	"fmt"
	"sync"
	"sync/atomic"
	"time"
)

var (
	tps             int64 = 0
	total           int64
	goroutine_count int = 1000
	ops_count       int = 1000
)

type TestFun func() error

func CurrentTest(do TestFun, args ...int) {
	fmt.Println("======== START ======")

	if len(args) > 0 && args[0] > 0 {
		goroutine_count = args[0]
	}
	if len(args) > 1 && args[1] > 0 {
		ops_count = args[1]
	}
	fmt.Printf("Concurrent=%v	Each send=%v\n", goroutine_count, ops_count)
	wg := sync.WaitGroup{}
	wg.Add(goroutine_count)

	for cs := 0; cs < goroutine_count; cs++ {
		go func() {
			for ps := 0; ps < ops_count; ps++ {
				if err := do(); err != nil {
					fmt.Errorf("err=%v", err)
					break
				}
				atomic.AddInt64(&tps, 1)
			}
			wg.Done()
		}()
	}
	go listen()
	wg.Wait()
	fmt.Println("======== END ======")
}

func listen() {
	ticket := time.NewTicker(time.Second)
	for range ticket.C {
		tp := atomic.SwapInt64(&tps, 0)
		fmt.Println("tps=", tp)
		atomic.AddInt64(&total, tp)
		if tp == 0 && total > 0 {
			fmt.Println("========end total=", total)
			break
		}
	}
}