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

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

千萱小哥_2100

千萱小哥_2100

发布时间:2026-06-27 11:51:47

|

933人浏览过

|

来源于php中文网

原创

必须将令牌桶算法深度嵌入协程调度链路,使每个await前显式申请令牌:用asyncio.Lock+asyncio.sleep实现原子填充,禁用同步操作,多进程需Redis Lua保证一致性。

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

要对高频并发的 async 任务做硬性速率限制,不能只靠“等一等”或“数个数”,必须把令牌桶算法深度嵌入协程调度链路中——核心是让每个 await 操作在真正执行前,先向令牌桶申请许可,无令牌则挂起,不阻塞事件循环。

令牌桶需与事件循环协同工作

Python 中不能用普通线程安全计数器(如 threading.Lock + int)实现令牌桶,因为 async 任务不共享线程上下文。必须使用 asyncio 兼容的原子操作原语:

  • 用 asyncio.Semaphore 管理并发槽位数(对应桶容量),但它只控制“同时运行几个”,不解决“单位时间放多少令牌”问题
  • 真正可行的是基于 asyncio.Queue 或自定义 TokenBucket 类,内部用 asyncio.Event + asyncio.create_task 启动后台填充协程,按固定间隔(如每秒 10 个)向队列注入令牌
  • 关键细节:填充协程必须用 asyncio.sleep() 而非 time.sleep(),否则会阻塞整个事件循环

await 前必须显式调用 acquire_token()

不能把限流逻辑藏在装饰器或中间件里“自动生效”,必须让每个高风险 async 任务显式 await 一个令牌获取操作。示例结构如下:

class TokenBucket:
    def __init__(self, capacity: int, refill_rate: float):
        self.capacity = capacity
        self.tokens = capacity
        self.refill_rate = refill_rate
        self.last_refill = asyncio.get_event_loop().time()
        self._lock = asyncio.Lock()
<pre class="brush:php;toolbar:false;">async def acquire(self):
    while True:
        async with self._lock:
            now = asyncio.get_event_loop().time()
            elapsed = now - self.last_refill
            new_tokens = int(elapsed * self.refill_rate)
            if new_tokens > 0:
                self.tokens = min(self.capacity, self.tokens + new_tokens)
                self.last_refill = now
            if self.tokens > 0:
                self.tokens -= 1
                return True
        # 无令牌时,等待 10ms 后重试,避免忙等
        await asyncio.sleep(0.01)

调用方写法必须是:await bucket.acquire() —— 这个 await 就是硬性管控点,它不返回就卡住当前协程,但不阻塞其他任务。

避免常见陷阱:同步操作污染事件循环

很多团队失败是因为在 acquire() 内部混入了同步耗时操作,例如:

  • 调用 redis.incr()(未用 aioredis)→ 实际是同步阻塞调用 → 整个事件循环卡死
  • 用 datetime.now() 替代 asyncio.get_event_loop().time() → 时间精度低,且无实际危害但暴露设计粗糙
  • 令牌桶状态存在本地内存 → 多进程部署时失效 → 必须用 Redis 的 Lua 脚本保证原子性(EVAL "local tokens = ..." 2 rate limit_key)

与业务逻辑解耦但强绑定

令牌桶本身不关心任务内容,但它的 acquire() 必须成为任务入口的强制前置步骤。推荐模式:

  • 定义统一的 async 工厂函数:async def run_with_rate_limit(task_coro, bucket)
  • 内部先 await bucket.acquire(),再 await task_coro(),异常时也确保令牌归还(可选)
  • 禁止直接 await 业务协程,所有高频调用路径必须走该工厂函数
  • FastAPI 中可封装为依赖项(Depends),让路由函数签名强制携带 bucket 实例

不复杂但容易忽略:硬性管控不是加个装饰器就能生效,而是把令牌获取变成 await 链上不可跳过的环节。

热门AI工具

更多
WorkBuddy

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

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

DeepSeek

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

Atoms
Atoms Hot

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

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

火山引擎

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

豆包大模型

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

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

相关专题

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

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

4266

2023.06.20

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

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

1089

2023.07.04

js四舍五入
js四舍五入

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

4184

2023.07.04

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

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

860

2023.09.01

JavaScript转义字符
JavaScript转义字符

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

1736

2023.09.04

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

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

3105

2023.09.04

如何启用JavaScript
如何启用JavaScript

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

4033

2023.09.12

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

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

2580

2023.09.20

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

120

2026.09.23

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Python 并发编程实战
Python 并发编程实战

共12课时 | 1万人学习

Django 教程
Django 教程

共8课时 | 11.2万人学习

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

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