本文深入探讨了在 Python 异步编程中实现优先级队列的两种核心方式：一是手动基于 `heapq` 和 `asyncio.Semaphore` 封装线程安全的自定义 `PriorityQueue` 类，二是直接使用标准库中专为异步场景设计的 `asyncio.PriorityQueue`——它作为 `asyncio.Queue` 的子类，内置堆排序、协程友好、线程安全，且用法简洁统一（如 `await queue.put((priority, item))`），优先级数值越小越先执行；文章明确推荐后者，因其更可靠、无需手动维护排序逻辑，是构建高效异步任务调度、消息处理等场景的首选方案。

asyncio.Queue 本身不支持优先级，但可以通过组合 asyncio.PriorityQueue（它已内置）或用 asyncio.Queue + 自定义优先级逻辑来实现。最直接、推荐的方式是直接使用 asyncio.PriorityQueue —— 它是标准库中专为异步场景设计的线程安全、协程友好的优先级队列。

✅ 使用 asyncio.PriorityQueue（最简单可靠）

asyncio.PriorityQueue 是 asyncio.Queue 的子类，内部基于 heapq 实现，支持按优先级（数值越小优先级越高）自动排序。所有方法（put()、get()、qsize() 等）与 Queue 一致，且天然支持 await。

• 入队时传入一个可比较的优先级项，通常为元组：(priority, item)
• 优先级可以是整数、浮点数，甚至自定义对象（只要支持 < 比较）
• 多个协程并发 get() 时，仍能保证每次取出当前最高优先级（最小 priority 值）的元素

示例：

import asyncio
async 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 + sorted() 手动维护（不推荐）

有人尝试用普通 asyncio.Queue 存储元组，再每次 get() 后手动排序剩余元素——这不仅破坏了队列的原子性，还无法保证并发安全，且 Queue 内部结构不可直接访问或修改。切勿这样做。

? 如需更灵活的优先级逻辑（如动态权重、多级策略）

可封装一个包装类，内部用 asyncio.PriorityQueue，但把 put() 的优先级计算逻辑隐藏起来：

• 例如：根据任务类型、超时时间、用户等级动态生成 priority 值
• 确保 priority 值始终可比较、稳定（相同逻辑下相同输入产生相同 priority）
• 避免在 priority 中嵌入不可比较对象（如未实现 __lt__ 的类实例）

示例（按截止时间优先）：

import asyncio
from datetime import datetime
class 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() 后检查并跳过已失效项

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于文章的相关知识，也可关注golang学习网公众号。