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 攻击带来的影响。不过要注意，这只是整个安全防护体系的一部分，不能单独依赖它来解决所有问题。
基本上就这些。
