kratos/pkg/container/queue/aqm/codel_test.go

package aqm

import (
	"context"
	"fmt"
	"sync"
	"sync/atomic"
	"testing"
	"time"

	"github.com/bilibili/Kratos/pkg/ecode"
)

var testConf = &Config{
	Target:   20,
	Internal: 500,
}

var qps = time.Microsecond * 2000

func TestCoDel1200(t *testing.T) {
	q := New(testConf)
	drop := new(int64)
	tm := new(int64)
	delay := time.Millisecond * 3000
	testPush(q, qps, delay, drop, tm)
	fmt.Printf("qps %v process time %v drop %d timeout %d \n", int64(time.Second/qps), delay, *drop, *tm)
	time.Sleep(time.Second)
}

func TestCoDel200(t *testing.T) {
	q := New(testConf)
	drop := new(int64)
	tm := new(int64)
	delay := time.Millisecond * 2000
	testPush(q, qps, delay, drop, tm)
	fmt.Printf("qps %v process time %v drop %d timeout %d \n", int64(time.Second/qps), delay, *drop, *tm)
	time.Sleep(time.Second)

}

func TestCoDel100(t *testing.T) {
	q := New(testConf)
	drop := new(int64)
	tm := new(int64)
	delay := time.Millisecond * 1000
	testPush(q, qps, delay, drop, tm)
	fmt.Printf("qps %v process time %v drop %d timeout %d \n", int64(time.Second/qps), delay, *drop, *tm)

}

func TestCoDel50(t *testing.T) {
	q := New(testConf)
	drop := new(int64)
	tm := new(int64)
	delay := time.Millisecond * 500
	testPush(q, qps, delay, drop, tm)
	fmt.Printf("qps %v process time %v drop %d timeout %d \n", int64(time.Second/qps), delay, *drop, *tm)
}

func testPush(q *Queue, sleep time.Duration, delay time.Duration, drop *int64, tm *int64) {
	var group sync.WaitGroup
	for i := 0; i < 5000; i++ {
		time.Sleep(sleep)
		group.Add(1)
		go func() {
			defer group.Done()
			ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Millisecond*1000))
			defer cancel()
			if err := q.Push(ctx); err != nil {
				if err == ecode.LimitExceed {
					atomic.AddInt64(drop, 1)
				} else {
					atomic.AddInt64(tm, 1)
				}
			} else {
				time.Sleep(delay)
				q.Pop()
			}
		}()
	}
	group.Wait()
}

func BenchmarkAQM(b *testing.B) {
	q := Default()
	b.RunParallel(func(p *testing.PB) {
		for p.Next() {
			ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Millisecond*5))
			err := q.Push(ctx)
			if err == nil {
				q.Pop()
			}
			cancel()
		}
	})
}