25 lines
530 B
Go
25 lines
530 B
Go
package limit
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestRateLimiter(t *testing.T) {
|
|
// 测试限流，连续多次请求 id=10
|
|
//p := &Parameters{
|
|
// Key: "key",
|
|
// Burst: 1,
|
|
// Interval: time.Second * 5,
|
|
//}
|
|
//n := NewRateLimiter()
|
|
//for i := 0; i < 11; i++ {
|
|
// err := n.RateLimit(p)
|
|
// if err != nil {
|
|
// t.Errorf("请求被限流，处理 id=%d ,错误信息：%v", i, err)
|
|
// } else {
|
|
// t.Logf("请求成功，处理 id=%d 的请求\n", i)
|
|
// }
|
|
// time.Sleep(1 * time.Second) // 每 1 秒请求一次
|
|
//}
|
|
}
|