Golang网络编程如何防御DDoS攻击使用rate.Limiter实现请求限流-Golang-PHP中文网

Golang网络编程如何防御DDoS攻击使用rate.Limiter实现请求限流

P粉602998670

发布： 2025-07-22 09:53:01

原创

562人浏览过

rate.limiter是go语言中基于令牌桶算法实现的限流工具，用于控制事件频率。其核心参数包括速率（每秒允许的请求数）和突发容量（短时间内可承受的最大请求数）。在http服务中使用时，通常通过中间件对每个请求进行判断，超过限制则返回429错误。实际部署需注意ip粒度选择、内存管理问题以及结合cdn、防火墙等其他防护手段共同防御ddos攻击。

Golang网络编程如何防御DDoS攻击使用rate.Limiter实现请求限流

在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次突发请求一次性到达。

ListenLeap

AI辅助通过播客学英语

101

查看详情

如何在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段
- 使用负载均衡和自动扩容机制应对流量激增