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

雨敏吖_5677

雨敏吖_5677

2026-05-24

948人浏览

原创

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 数。

  • 结构体里存 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 写得再准,跨实例时桶还是各自为政。

golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

golang go语言

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2024.02.23

499

5

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

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

2024.02.23

596

5

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

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

2024.02.23

1076

5

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

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

2024.03.05

771

5

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

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

2024.05.21

1586

6

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

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

2025.06.09

4454

18

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

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

2025.06.10

2034

13

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

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

2025.06.17

3946

17

golang数组大全
golang数组大全

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

2025.06.20

4746

22

热门下载

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

精品课程

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