0

0

使用Golang实现Rate Limiting限流_基于令牌桶算法的API保护

P粉602998670

P粉602998670

发布时间:2026-02-25 16:55:33

|

527人浏览过

|

来源于php中文网

原创

golang.org/x/time/rate 是官方维护的轻量、无锁、并发安全令牌桶限流库,需复用实例而非每次新建,支持动态调速;http中间件中应优先用 reserven 配合上下文检查,避免误用 allow 或 wait 导致逻辑错误。

使用golang实现rate limiting限流_基于令牌桶算法的api保护

Go 里用 golang.org/x/time/rate 实现令牌桶限流

标准库没有原生限流器,但 golang.org/x/time/rate 是官方维护、生产可用的令牌桶实现。它轻量、无锁、支持动态调整速率,比自己手写 time.Ticker + channel 更可靠。

常见错误是直接在 HTTP handler 里 new 一个 rate.Limiter,导致每个请求都新建限流器,完全失效。正确做法是复用同一个实例,按用户 ID、IP 或路由路径做 key 分组管理。

  • rate.NewLimiter(rate.Limit(10), 5) 表示:最大允许突发 5 个令牌,长期速率为每秒 10 个请求
  • 第一个参数是 rate.Limit 类型(本质是 float64),不是整数;传 10.010 更明确
  • 第二个参数是 burst,必须 ≥ 1;设为 1 就是严格匀速,设太大则失去限流意义
  • 注意:该包不处理分布式场景,单机有效

HTTP 中间件里怎么安全调用 limiter.Wait

Wait 会阻塞直到拿到令牌或上下文超时,适合对延迟敏感度低、且希望“等一等再试”的场景;而 AllowTryConsume 是非阻塞的,适合快速失败。

中间件里别直接用 limiter.Wait(r.Context()) —— 如果客户端断连或超时,r.Context() 可能已取消,Wait 会立即返回错误,但你可能没检查这个错误就继续执行了 handler,造成逻辑错乱。

立即学习go语言免费学习笔记(深入)”;

知元AI
知元AI

AI智能语音聊天 对讲问答 AI绘画 AI写作 AI创作助手工具

下载
  • 务必检查 err:如果 err == context.Canceled || err == context.DeadlineExceeded,说明请求本身已无效,不该再往下走
  • 更稳妥的做法是用 limiter.ReserveN(r.Context(), 1),它返回 *rate.Reservation,可调用 OK() 判断是否成功,且支持预占多个令牌
  • 别在 http.ServeHTTP 外层 defer 调用 Reservation.Cancel() —— 它只应在未使用该 reservation 时才调,否则可能误释放

如何给不同用户/路径配不同限流策略

硬编码一个全局 rate.Limiter 只适用于全站统一限流。真实场景中,登录用户和游客、VIP 和普通用户、/api/pay 和 /api/status 的容忍度完全不同。

典型做法是用 sync.Map 或带 TTL 的内存缓存(如 github.com/bluele/gcache)按 key 缓存 *rate.Limiter 实例,key 可以是 "ip:" + clientIP"user:" + userID

  • 避免 key 泛滥:限制 key 总数,比如只对前 10 万 IP 做个性化限流,其余走默认策略
  • 注意 rate.Limiter 不是线程安全的?错,它是并发安全的,但 SetLimitAndBurst 是原子更新,可用于运行时动态调速
  • 不要在每次请求里 new rate.Limiter 并丢弃——构造开销小,但 GC 压力会上升,尤其高 QPS 下
  • 测试时用 limiter.SetLimitAndBurst(rate.Limit(1000), 1000) 临时放开,别改代码注释

为什么 Allow 返回 true 还可能被拒绝

Allow 看起来最简单,但它只反映「此刻是否能拿一个令牌」,不保证后续操作一定成功。真正容易踩坑的是:它不感知上下文取消,也不阻塞等待,所以常被误用于需要强一致性的场景。

比如你在 Allow 返回 true 后去查数据库、发 MQ,结果 DB 挂了、MQ 超时——整个请求失败了,但令牌已被扣掉,用户白丢一次配额。

  • 更适合的模式是:ReserveN(ctx, 1) → 检查 OK() → 执行业务 → 成功则 CancelAt() 不调用(自动消耗),失败则显式 Cancel()
  • Allow 适合日志采样、指标上报这类“丢了也不关键”的操作
  • 注意:所有方法都基于本地时钟,系统时间跳变(如 NTP 校准)会导致令牌生成节奏紊乱,生产环境确保 ntpdchronyd 稳定运行

令牌桶真正的复杂点不在代码几行,而在怎么定义“谁该被限”、什么时候该“重置桶”、以及下游服务挂了时要不要把令牌还回去——这些没法靠 rate.Limiter 自动解决。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

207

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

242

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

350

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

212

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

405

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

365

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

200

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

1111

2025.06.17

batoto漫画官网入口与网页版访问指南
batoto漫画官网入口与网页版访问指南

本专题系统整理batoto漫画官方网站最新可用入口,涵盖最新官网地址、网页版登录页面及防走失访问方式说明,帮助用户快速找到batoto漫画官方平台,稳定在线阅读各类漫画内容。

65

2026.02.25

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 5.5万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号