什么是速率限制器以及为什么使用它？-Golang-PHP中文网

什么是速率限制器以及为什么使用它？

Mary-Kate Olsen

发布： 2025-01-08 06:36:40

原创

877 人浏览过

Web 系统经常面临短时间内大量客户端请求的挑战。这可能会使服务器不堪重负，导致速度减慢或失败。 速率限制器通过控制每个客户端的请求频率来优雅地解决这个问题。它充当看门人的角色，将 API 或服务调用限制在指定的时间窗口内。

为什么要实施速率限制？

速率限制提供了几个重要的好处：

滥用预防： 缓解过度或恶意请求，例如拒绝服务 (DoS) 攻击，防止系统过载。
资源管理：确保有效的资源分配，防止一个客户端独占服务器资源并影响其他客户端。
性能增强：通过防止请求泛滥，即使在高流量负载下也能保持应用程序响应能力。
改善用户体验：防止用户因请求过多而锁定或性能下降。
安全加强：通过限制请求率来帮助阻止暴力攻击或利用尝试。

使用 Chi 进行实用的 Go 实现

让我们使用流行的 Chi 路由包和 golang.org/x/time/rate 包（采用令牌桶算法）来检查 Go 实现。此示例使用 http.Handler 来控制客户端请求率。

<code class="language-go">package main

import (
    "encoding/json"
    "net/http"
    "strings"
    "time"

    "github.com/go-chi/chi"
    "golang.org/x/time/rate"
)

func main() {
    r := chi.NewRouter()

    // Globally limits to 5 requests per second with a burst of 10
    r.Use(RateLimiter(rate.Limit(5), 10, 1*time.Second))

    // Test route
    r.Get("/", func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Request successful"))
    })

    http.ListenAndServe(":3000", r)
}

// RateLimiter middleware
func RateLimiter(limit rate.Limit, burst int, waitTime time.Duration) func(next http.Handler) http.Handler {
    limiterMap := make(map[string]*rate.Limiter)
    lastRequestMap := make(map[string]time.Time)

    return func(next http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            ip := strings.Split(r.RemoteAddr, ":")[0]

            limiter, exists := limiterMap[ip]
            if !exists {
                limiter = rate.NewLimiter(limit, burst)
                limiterMap[ip] = limiter
            }

            lastRequestTime, lastRequestExists := lastRequestMap[ip]
            if lastRequestExists && time.Since(lastRequestTime) < waitTime {
                //Handle exceeding rate limit
                http.Error(w, http.StatusText(http.StatusTooManyRequests), http.StatusTooManyRequests)
                return
            }

            if !limiter.Allow() {
                http.Error(w, http.StatusText(http.StatusTooManyRequests), http.StatusTooManyRequests)
                return
            }

            lastRequestMap[ip] = time.Now()
            next.ServeHTTP(w, r)
        })
    }
}</code>

登录后复制