Golang如何测试API限流逻辑_Golang API请求限流测试方法

P粉602998670 2025-12-18 00:00:00 次阅读

答案：在Golang中测试API限流需模拟高频请求并验证系统行为，使用golang.org/x/time/rate实现令牌桶算法，结合httptest创建无网络依赖的服务端点，通过goroutine并发发送请求，检查成功与被限流的响应状态；为保证测试稳定，应封装限流器接口并引入可控时钟或mock存储，确保突发容量和初始状态正确影响测试结果。

在Golang中测试API限流逻辑，关键在于模拟高频请求并验证系统是否按预期限制访问。常用限流算法有令牌桶、漏桶等，Go语言标准库中的 golang.org/x/time/rate 包提供了简洁的实现。测试时需结合HTTP测试工具和并发控制，确保限流策略生效。

使用 net/http/httptest 模拟请求

Go内置的 httptest 可创建无网络依赖的HTTP服务端点，适合单元测试。

示例：搭建一个带限流的简单Handler

func rateLimitedHandler() http.Handler {
    limiter := rate.NewLimiter(2, 4) // 每秒2个令牌，突发4个
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if !limiter.Allow() {
            w.WriteHeader(http.StatusTooManyRequests)
            w.Write([]byte("too many requests"))
            return
        }
        w.WriteHeader(http.StatusOK)
        w.Write([]byte("success"))
    })
}

并发发起多请求验证限流

通过 goroutine 并发发送多个请求，观察返回状态码是否符合预期。

测试代码示例：

func TestRateLimit(t *testing.T) {
    server := httptest.NewServer(rateLimitedHandler())
    defer server.Close()

    var success, blocked int32

    var wg sync.WaitGroup
    client := &http.Client{}

    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            resp, err := client.Get(server.URL)
            if err != nil {
                t.Error(err)
                return
            }
            defer resp.Body.Close()

            if resp.StatusCode == http.StatusOK {
                atomic.AddInt32(&success, 1)
            } else if resp.StatusCode == http.StatusTooManyRequests {
                atomic.AddInt32(&blocked, 1)
            }
        }()
    }

    wg.Wait()

    if success > 4 {
        t.Errorf("expected at most 4 successes due to burst limit, got %d", success)
    }
    if blocked == 0 {
        t.Errorf("expected some requests to be blocked, but none were")
    }
}