如何在Golang中实现HTTP请求限流-Golang-PHP中文网

如何在Golang中实现HTTP请求限流

P粉602998670

发布： 2025-10-26 08:54:02

原创

144人浏览过

使用golang.org/x/time/rate包中的rate.Limiter可基于令牌桶算法实现HTTP请求限流，支持全局限流或按客户端IP独立限流，结合中间件和定期清理机制保障服务稳定性。

如何在golang中实现http请求限流

在Golang中实现HTTP请求限流，主要是为了防止服务被过多请求压垮，保障系统稳定性。常用的方式是使用令牌桶或漏桶算法，结合 golang.org/x/time/rate 包来实现简单高效的限流控制。

使用 golang.org/x/time/rate 实现限流

rate.Limiter 是 Go 官方提供的限流工具，基于令牌桶算法，使用简单且性能良好。你可以为每个请求者（如IP）分配一个限流器，也可以全局限流。

示例：对所有请求统一限流

package main
<p>import (
"golang.org/x/time/rate"
"net/http"
"time"
)</p><p>var limiter = rate.NewLimiter(10, 50) // 每秒10个令牌，最多容纳50个突发请求</p><p><span>立即学习</span>“<a href="https://pan.quark.cn/s/00968c3c2c15" style="text-decoration: underline !important; color: blue; font-weight: bolder;" rel="nofollow" target="_blank">go语言免费学习笔记（深入）</a>”；</p><p>func limit(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
if !limiter.Allow() {
http.StatusText(http.StatusTooManyRequests)
w.WriteHeader(http.StatusTooManyRequests)
w.Write([]byte("too many requests"))
return
}
next(w, r)
}
}</p><p>func handler(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello, limited world!"))
}</p><p>func main() {
mux := http.NewServeMux()
mux.HandleFunc("/", limit(handler))
http.ListenAndServe(":8080", mux)
}</p>

登录后复制

基于客户端IP的独立限流

更常见的场景是为每个客户端IP设置独立的限流策略，避免某个用户影响其他用户。

可以使用 map + sync.Mutex 或 sync.Map 来存储每个IP对应的限流器。

如知AI笔记

如知笔记——支持markdown的在线笔记，支持ai智能写作、AI搜索，支持DeepseekR1满血大模型

查看详情

var visitors = make(map[string]*rate.Limiter)
var mu sync.RWMutex
<p>func getVisitorLimiter(ip string) *rate.Limiter {
mu.RLock()
limiter, exists := visitors[ip]
mu.RUnlock()
if exists {
return limiter
}</p><pre class='brush:php;toolbar:false;'>mu.Lock()
// 双检确认，避免重复创建
if limiter, exists = visitors[ip]; exists {
    mu.Unlock()
    return limiter
}
limiter = rate.NewLimiter(2, 5) // 每秒2次请求，最多5个突发
visitors[ip] = limiter
mu.Unlock()
return limiter

登录后复制

}

func ipLimit(next http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { ip := r.RemoteAddr // 注意：反向代理时可能需要读取 X-Forwarded-For limiter := getVisitorLimiter(ip)

    if !limiter.Allow() {
        http.StatusText(http.StatusTooManyRequests)
        w.WriteHeader(http.StatusTooManyRequests)
        w.Write([]byte("too many requests"))
        return
    }
    next(w, r)
}

登录后复制

}

定期清理过期的限流器

如果不限期清理 map 中的旧IP记录，内存会持续增长。可以通过启动一个后台协程定期清理长时间未活动的IP。

例如：每5分钟清理超过30分钟无请求的IP

func cleanupVisitors() {
    for {
        time.Sleep(5 * time.Minute)
        mu.Lock()
        for ip, limiter := range visitors {
            // 假设我们通过 LastSeen 时间判断，这里简化为检查是否“空闲”
            // 更完整可扩展结构体包含 lastSeen 字段
            if limiter.Tokens() == float64(limiter.Burst()) {
                delete(visitors, ip)
            }
        }
        mu.Unlock()
    }
}
<p>// 在 main 中启动清理
func main() {
go cleanupVisitors()
// ...
}</p>

登录后复制

基本上就这些。使用 rate.Limiter 配合中间件方式，可以灵活实现全局或按IP的HTTP请求限流。关键是根据业务需求设置合理的速率和突发值，并注意内存管理。

以上就是如何在Golang中实现HTTP请求限流的详细内容，更多请关注php中文网其它相关文章！