如何利用 await 结合令牌桶算法实现对高频并发 async 任务的最大并发速度硬性管控

秋墨吖_2725

秋墨吖_2725

2026-06-16

321人浏览

原创

await 无法直接实现令牌桶硬性速率管控,需将“取令牌”改造为可 await 的异步操作,通过动态补发令牌、asyncio.lock 保护状态、支持超时与异常反馈,实现精准 qps 限流。

如何利用 await 结合令牌桶算法实现对高频并发 async 任务的最大并发速度硬性管控

直接用 await 无法实现令牌桶的硬性速率管控,必须配合协程等待逻辑与令牌状态检查。核心在于:把“取令牌”变成一个可 await 的异步操作,让无令牌的协程挂起,直到新令牌生成或超时。

令牌桶需支持异步等待

同步版令牌桶(如 Go 示例)调用 Allow() 是阻塞判断,不适合 asyncio。必须改造为:
– 维护一个内部令牌计数器和上次填充时间
– 提供 acquire() 方法,返回 Awaitable[bool]
– 若无令牌,自动计算等待时长并 await asyncio.sleep(),而非忙等或立即拒绝

关键实现要点

  • 使用 asyncio.Lock 保护桶状态读写,避免并发修改导致令牌透支
  • 填充逻辑不依赖定时器,而是在每次 acquire() 时按时间差动态补发,更精准、无资源泄漏
  • 支持可选超时(timeout: float = None),防止协程无限等待,便于熔断或降级
  • 拒绝路径应返回明确信号(如抛出 RateLimitExceeded 异常),而非静默失败

典型调用方式

不是在任务里简单写 await token_bucket.acquire() 就完事,而是要嵌入到实际业务协程中:

python 查询技能
python 查询技能

查询客流数据,输出JSON格式,可直接导入Bitable等可视化工具

下载
async def api_handler(request):
    try:
        await rate_limiter.acquire()  # 等待令牌,可能挂起
        return await do_actual_work(request)
    except RateLimitExceeded:
        return web.json_response({"error": "429 Too Many Requests"}, status=429)

这样,即使瞬间涌入 1000 个请求,真正执行 do_actual_work 的速率也严格受限于令牌填充速度(例如每秒 10 个),其余协程在 acquire() 处排队或超时退出。

与 BoundedSemaphore 的本质区别

– BoundedSemaphore(10) 控制的是**最大并发数**(同时最多 10 个任务在跑),不限制单位时间请求数;
– 令牌桶控制的是**长期平均速率**(如 10 QPS),允许短时突发(桶容量决定弹性),更适合 API 限速场景。
两者可叠加使用:先用令牌桶控速率,再用 Semaphore 控资源占用(如连接数),形成双层防护。

相关文章

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

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

下载

相关标签:

python

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

相关专题

更多
js获取数组长度的方法
js获取数组长度的方法

在js中,可以利用array对象的length属性来获取数组长度,该属性可设置或返回数组中元素的数目,只需要使用“array.length”语句即可返回表示数组对象的元素个数的数值,也就是长度值。php中文网还提供JavaScript数组的相关下载、相关课程等内容,供大家免费下载使用。

2023.06.20

4686

5

js刷新当前页面
js刷新当前页面

js刷新当前页面的方法:1、reload方法,该方法强迫浏览器刷新当前页面,语法为“location.reload([bForceGet]) ”;2、replace方法,该方法通过指定URL替换当前缓存在历史里(客户端)的项目,因此当使用replace方法之后,不能通过“前进”和“后退”来访问已经被替换的URL,语法为“location.replace(URL) ”。php中文网为大家带来了js刷新当前页面的相关知识、以及相关文章等内容

2023.07.04

1169

3

js四舍五入
js四舍五入

js四舍五入的方法:1、tofixed方法,可把 Number 四舍五入为指定小数位数的数字;2、round() 方法,可把一个数字舍入为最接近的整数。php中文网为大家带来了js四舍五入的相关知识、以及相关文章等内容

2023.07.04

4624

6

js删除节点的方法
js删除节点的方法

js删除节点的方法有:1、removeChild()方法,用于从父节点中移除指定的子节点,它需要两个参数,第一个参数是要删除的子节点,第二个参数是父节点;2、parentNode.removeChild()方法,可以直接通过父节点调用来删除子节点;3、remove()方法,可以直接删除节点,而无需指定父节点;4、innerHTML属性,用于删除节点的内容。

2023.09.01

920

4

JavaScript转义字符
JavaScript转义字符

JavaScript中的转义字符是反斜杠和引号,可以在字符串中表示特殊字符或改变字符的含义。本专题为大家提供转义字符相关的文章、下载、课程内容,供大家免费下载体验。

2023.09.04

1836

5

js生成随机数的方法
js生成随机数的方法

js生成随机数的方法有:1、使用random函数生成0-1之间的随机数;2、使用random函数和特定范围来生成随机整数;3、使用random函数和round函数生成0-99之间的随机整数;4、使用random函数和其他函数生成更复杂的随机数;5、使用random函数和其他函数生成范围内的随机小数;6、使用random函数和其他函数生成范围内的随机整数或小数。

2023.09.04

3345

4

如何启用JavaScript
如何启用JavaScript

JavaScript启用方法有内联脚本、内部脚本、外部脚本和异步加载。详细介绍:1、内联脚本是将JavaScript代码直接嵌入到HTML标签中;2、内部脚本是将JavaScript代码放置在HTML文件的`<script>`标签中;3、外部脚本是将JavaScript代码放置在一个独立的文件;4、外部脚本是将JavaScript代码放置在一个独立的文件。

2023.09.12

4333

6

Js中Symbol类详解
Js中Symbol类详解

javascript中的Symbol数据类型是一种基本数据类型,用于表示独一无二的值。Symbol的特点:1、独一无二,每个Symbol值都是唯一的,不会与其他任何值相等;2、不可变性,Symbol值一旦创建,就不能修改或者重新赋值;3、隐藏性,Symbol值不会被隐式转换为其他类型;4、无法枚举,Symbol值作为对象的属性名时,默认是不可枚举的。

2023.09.20

2840

5

java访问控制修饰符介绍
java访问控制修饰符介绍

java访问控制修饰符有四种,分别是public、protected、private、默认访问修饰符。详细介绍:1、public,public是最宽松的访问控制修饰符,被修饰的类、方法和变量可以被任何其他类访问,当一个类、方法或变量被声明为public时,它们可以在任何地方被访问,无论是同一个包中的类还是不同包中的类;2、protected修饰符等等。

2023.09.20

908

7

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Scrapy 官方文档与教程
Scrapy 官方文档与教程

共0课时 | 0人学习

Conan 2 安装指南
Conan 2 安装指南

共0课时 | 0人学习