Golang如何实现请求限流和防刷-Golang-PHP中文网

Golang如何实现请求限流和防刷

P粉602998670

发布： 2025-10-20 13:45:02

原创

487人浏览过

使用令牌桶或Redis实现限流，单机用rate.Limiter，分布式用Redis+Lua，结合中间件防刷，保障服务稳定。

golang如何实现请求限流和防刷

在高并发场景下，Golang 实现请求限流和防刷是保障服务稳定性的关键手段。通过限制单位时间内客户端的请求次数，可以有效防止恶意刷接口、爬虫攻击或突发流量压垮后端服务。常见的实现方式包括令牌桶、漏桶算法，结合内存或 Redis 存储进行控制。

使用令牌桶算法限流

Go 标准库 golang.org/x/time/rate 提供了基于令牌桶的限流器实现，简单高效，适合单机场景。

基本用法如下：

package main
<p>import (
"fmt"
"time"
"golang.org/x/time/rate"
)</p><p>func main() {
// 每秒允许 5 个请求，最多容纳 10 个突发请求
limiter := rate.NewLimiter(5, 10)</p><pre class='brush:php;toolbar:false;'>for i := 0; i < 20; i++ {
    if !limiter.Allow() {
        fmt.Println("请求被限流")
        continue
    }
    fmt.Printf("处理请求 %d\n", i)
    time.Sleep(100 * time.Millisecond)
}

登录后复制

}

立即学习“go语言免费学习笔记（深入）”；

这种方式适用于单个服务实例，但无法跨节点共享状态，不适用于分布式环境。

基于 Redis 的分布式限流

在微服务或多实例部署中，需使用 Redis 实现分布式限流。常用算法是滑动窗口或固定窗口计数。

示例：使用 Redis + Lua 脚本实现 IP 级别每分钟最多 60 次请求：

-- limit.lua
local key = KEYS[1]
local limit = tonumber(ARGV[1])
local expire_time = ARGV[2]
<p>local current = redis.call("INCR", key)
if current == 1 then
redis.call("EXPIRE", key, expire_time)
end
return current <= limit and 1 or 0</p>

登录后复制

Go 中调用：

import (
    "context"
    "github.com/go-redis/redis/v8"
)
<p>var ctx = context.Background()</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/1089">
                            <img src="https://img.php.cn/upload/ai_manual/001/503/042/68b6c82c43031769.png" alt="Quinvio AI">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/1089">Quinvio AI</a>
                            <p>AI辅助下快速创建视频，虚拟代言人</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="Quinvio AI">
                                <span>59</span>
                            </div>
                        </div>
                        <a href="/ai/1089" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="Quinvio AI">
                        </a>
                    </div>
                <p>func isAllowed(client *redis.Client, ip string) bool {
script := redis.NewScript(<code> local key = KEYS[1] local limit = tonumber(ARGV[1]) local expire_time = ARGV[2] local current = redis.call("INCR", key) if current == 1 then redis.call("EXPIRE", key, expire_time) end return current <= limit and 1 or 0 </code>)</p><pre class='brush:php;toolbar:false;'>result, err := script.Run(ctx, client, []string{"rate_limit:" + ip}, 60, 60).Int()
return err == nil && result == 1

登录后复制

}

立即学习“go语言免费学习笔记（深入）”；

该方法可保证多个服务实例共享限流状态，适合生产环境。

结合中间件实现 HTTP 接口防刷

在 Web 服务中，可通过 Gin 或其他框架的中间件统一拦截请求并做限流判断。

例如 Gin 中间件示例：

func RateLimitMiddleware(client *redis.Client) gin.HandlerFunc {
    return func(c *gin.Context) {
        ip := c.ClientIP()
        if !isAllowed(client, ip) {
            c.JSON(429, gin.H{"error": "请求过于频繁，请稍后再试"})
            c.Abort()
            return
        }
        c.Next()
    }
}
<p>// 使用
r := gin.Default()
r.Use(RateLimitMiddleware(redisClient))
r.GET("/api/data", getDataHandler)
r.Run(":8080")</p>

登录后复制

可根据业务需求扩展为按用户 ID、API Key、设备指纹等维度限流。