rate.limiter是go语言中基于令牌桶算法实现的限流工具，用于控制事件频率。其核心参数包括速率（每秒允许的请求数）和突发容量（短时间内可承受的最大请求数）。在http服务中使用时，通常通过中间件对每个请求进行判断，超过限制则返回429错误。实际部署需注意ip粒度选择、内存管理问题以及结合cdn、防火墙等其他防护手段共同防御ddos攻击。
在golang网络服务中，防御DDoS攻击的一个常用手段是限制客户端的请求频率。通过golang.org/x/time/rate包中的rate.Limiter结构体，我们可以很方便地实现基于令牌桶算法的限流机制。
什么是 rate.Limiter？
rate.Limiter 是 Go 标准库之外提供的一个限流工具，它使用令牌桶算法来控制事件发生的频率。你可以设定每秒允许处理多少个请求（速率），以及最多能容纳多少个“积压”的请求（突发容量）。
简单来说：
立即学习“go语言免费学习笔记（深入）”；
- 速率（r）：每秒允许的请求数
- 突发容量（b）：短时间内可以承受的最大请求数
比如设置为 rate.Every(100*time.Millisecond) 和突发容量5，表示每秒允许10次请求，但允许最多5次突发请求一次性到达。
如何在HTTP服务中使用 rate.Limiter 做限流？
通常我们会将限流逻辑放在中间件中，对所有进入的请求进行统一控制。
package main import ( "fmt" "net/http" "sync" "time" "golang.org/x/time/rate" ) type RateLimiter struct { visitors map[String]*rate.Limiter mu sync.RWMutex } func (rl *RateLimiter) getVisitor(ip string) *rate.Limiter { rl.mu.Lock() defer rl.mu.Unlock() limiter, exists := rl.visitors[ip] if !exists { // 每秒允许20个请求，突发容量为5 limiter = rate.NewLimiter(20, 5) rl.visitors[ip] = limiter } return limiter } func limit(next http.HandlerFunc, limiter *RateLimiter) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { ip := r.RemoteAddr l := limiter.getVisitor(ip) if !l.Allow() { http.Error(w, http.StatusText(http.StatusTooManyRequests), http.StatusTooManyRequests) return } next(w, r) } }
这样就能在每个请求进来时判断是否超过频率限制。
实际部署中需要注意的问题
虽然用rate.Limiter做限流很简单，但在实际部署中还是有一些细节容易被忽略：
-
IP粒度的选择
如果你担心代理或NAT用户被误伤，可以考虑使用更粗粒度的限流方式，例如按用户ID、API Key等来做区分。 -
内存管理问题
上面的例子中我们用了map[string]*rate.Limiter来存储每个IP的限流器，但如果访问量很大，这个map会无限增长，需要定期清理长时间未活跃的IP。 -
结合其他手段一起使用
单纯靠限流无法完全抵御大规模DDoS攻击，建议配合以下措施：- 使用CDN（如Cloudflare）进行前端过滤
- 配合防火墙规则，封禁异常IP段
- 使用负载均衡和自动扩容机制应对流量激增
小结
使用rate.Limiter是 Golang 网络服务中实现请求限流的一种轻量而有效的方式。它可以帮助我们在一定程度上缓解 DDoS 攻击带来的影响。不过要注意，这只是整个安全防护体系的一部分，不能单独依赖它来解决所有问题。
基本上就这些。
