您的位置:首页 >asyncio.Queue 实现带优先级的队列(PriorityQueue)可以通过自定义一个类来包装 asyncio.Queue,并在入队时根据优先级排序。下
发布于2026-02-07 阅读(0)
扫一扫,手机访问
推荐直接使用 asyncio.PriorityQueue 实现优先级队列,它是标准库中线程安全、协程友好的内置类,基于 heapq 自动排序,支持 await,优先级数值越小越先执行;避免手动维护排序。

asyncio.Queue 本身不支持优先级,但可以通过组合 asyncio.PriorityQueue(它已内置)或用 asyncio.Queue + 自定义优先级逻辑来实现。最直接、推荐的方式是直接使用 asyncio.PriorityQueue —— 它是标准库中专为异步场景设计的线程安全、协程友好的优先级队列。
asyncio.PriorityQueue 是 asyncio.Queue 的子类,内部基于 heapq 实现,支持按优先级(数值越小优先级越高)自动排序。所有方法(put()、get()、qsize() 等)与 Queue 一致,且天然支持 await。
(priority, item)< 比较)get() 时,仍能保证每次取出当前最高优先级(最小 priority 值)的元素示例:
import asyncioasync def main(): pq = asyncio.PriorityQueue()
# 先入低优先级(3),再入高优先级(1),最后入中等(2) await pq.put((3, "task_c")) await pq.put((1, "task_a")) # 最先被 get await pq.put((2, "task_b")) print(await pq.get()) # (1, 'task_a') print(await pq.get()) # (2, 'task_b') print(await pq.get()) # (3, 'task_c')asyncio.run(main())
有人尝试用普通 asyncio.Queue 存储元组,再每次 get() 后手动排序剩余元素——这不仅破坏了队列的原子性,还无法保证并发安全,且 Queue 内部结构不可直接访问或修改。切勿这样做。
可封装一个包装类,内部用 asyncio.PriorityQueue,但把 put() 的优先级计算逻辑隐藏起来:
__lt__ 的类实例)示例(按截止时间优先):
import asyncio from datetime import datetimeclass DeadlinePriorityQueue: def init(self): self._pq = asyncio.PriorityQueue()
async def put(self, item, deadline: datetime): # 越早截止,priority 越小 → 越先执行 priority = deadline.timestamp() await self._pq.put((priority, item)) async def get(self): _, item = await self._pq.get() return item使用
async def demo(): q = DeadlinePriorityQueue() await q.put("urgent", datetime(2024, 1, 1, 10, 0)) await q.put("later", datetime(2024, 1, 1, 12, 0)) print(await q.get()) # "urgent"
asyncio.PriorityQueue 在 Python 3.7+ 中可用;3.6 及更早版本需自行实现或升级put_nowait() / get_nowait() —— 它们不支持优先级调度,会绕过 heapq 排序逻辑,导致行为异常PriorityQueue 不提供 remove 接口,常见做法是加个“取消标记”字段,在 get() 后检查并跳过已失效项
售后无忧
立即购买>office旗舰店
售后无忧
立即购买>office旗舰店
售后无忧
立即购买>office旗舰店
售后无忧
立即购买>office旗舰店
正版软件
正版软件
正版软件
正版软件
正版软件
1
2
3
7
9