讲师中心 微信公众号
AI工具推荐 视频效率加速

Go语言限流算法实现_Golang令牌桶与漏桶原理详解

雨敏吖_5677

雨敏吖_5677

发布时间:2026-05-24 22:07:25

|

948人浏览过

|

来源于php中文网

原创

Go标准库无严格令牌桶实现,rate.Limiter是滑动窗口漏桶;其burst非容量而是最大延迟时间×rate;手写需用atomic.Int64、单调时钟和按需填充。

go语言限流算法实现_golang令牌桶与漏桶原理详解

Go 标准库没有严格意义上的令牌桶实现,golang.org/x/time/rate 是漏桶变体;真要按 RFC 2697 实现带容量上限、独立填充速率、可透支的令牌桶,得手写或引入第三方库。

rate.Limiter 不是令牌桶,是“滑动窗口式漏桶”

很多人误以为 rate.NewLimiter(10, 5) 创建的是“每秒 10 个令牌、桶容量 5”的令牌桶——其实它底层不维护令牌计数器,而是靠时间戳推算“最早能放行的时间”。它的行为等价于:允许最多 5 次瞬时突发,之后强制压到平均 10 QPS。这在多数 HTTP 限流场景够用,但若需动态调速、跨 goroutine 观察剩余令牌、或与外部系统同步水位(比如和 Redis 联动),就撑不住了。

  • 调用 Allow() 或 ReserveN() 时,它计算的是“从 now 往前推 burst 所需时间是否已过”,不是实时查桶里还有几个令牌
  • burst 参数本质是“最大允许延迟时间 × rate”,不是物理桶容量
  • 传入错误的时间值(如截断到秒的 time.Now().Unix())会导致窗口错位、放行过多

手写线程安全 TokenBucket 的三个关键点

自己实现一个符合经典定义的 TokenBucket,核心是解决原子更新、按需填充、非阻塞获取。别用锁包全局状态,也别用 float64 累加令牌——浮点误差几小时后就能偏差几十个。

Golang Naming
Golang Naming

Go(Golang)命名规范 — 包括包、构造函数、结构体、接口、常量、枚举、错误、布尔值、接收器、getter/setter、函数等。

下载
  • tokens 字段必须是 int64,用 atomic.LoadInt64 和 atomic.CompareAndSwapInt64 做 CAS 更新
  • 填充逻辑不能依赖 time.Ticker,而应记录上次填充时间,在 Take() 时按时间差补:当前令牌 = min(容量, 上次值 + rate × 时间差),单位统一用纳秒+int64
  • 拒绝时钟回拨:用 runtime.nanotime() 替代 time.Now().UnixNano(),它是单调时钟,不受 NTP 校正影响

漏桶算法必须“按需补桶”,不是“定时补桶”

用 time.Ticker 每秒往 channel 塞一个 token 看似简单,但完全失效——客户端 10ms 内连发 5 个请求,Ticker 还没 tick 第二次,桶就空了,全放行。真实漏桶必须每次请求都重算可用 token 数。

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

  • 结构体里存 lastRefill int64(纳秒时间戳)和 tokens int64,每次 Allow() 先更新:delta := (now - lastRefill) * rate / 1e9,再取 min(capacity, tokens + delta)
  • 更新 lastRefill 必须和 tokens 一起原子更新,否则并发下会丢失中间状态;推荐用 sync/atomic.Value 包裹整个状态 struct
  • 如果要用 time.Time,必须做 if now.After(last) { ... } 判断,否则服务器时间被 NTP 回拨会导致 token 暴增

真正难的不是写对单机版,而是把桶状态同步到分布式环境——比如多个 API 网关实例共享同一个桶水位。这时候 atomic 就没用了,得靠 Redis + Lua 原子脚本,或基于 Raft 的协调服务。单机 TokenBucket 写得再准,跨实例时桶还是各自为政。

热门AI工具

更多
SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

相关专题

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

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

499

2024.02.23

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

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

596

2024.02.23

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

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

1076

2024.02.23

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

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

771

2024.03.05

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

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

1606

2024.05.21

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

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

4474

2025.06.09

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

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

2034

2025.06.10

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

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

3946

2025.06.17

PixTV官网入口地址合集
PixTV官网入口地址合集

本专题汇总了 PixTV AI 一站式视频创作平台的官方入口与使用教程。无需下载软件,浏览器直接访问即可使用。平台将剧本、图像、视频、声音与剪辑整合在“无限画布”中,接入 GPT Image 2.5、Seedance 2.5 等头部模型。本专题整理了从新建画布、角色锚定、分镜拆分到视频生成与导出的完整操作指南,助你快速上手 AI 短剧与漫剧创作。

40

2026.10.10

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn