You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.1 KiB
97 lines
2.1 KiB
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()
|
|
}
|
|
})
|
|
}
|
|
|