当前位置:首页 > 文章列表 > 文章 > python教程 > asyncio.Queue 实现带优先级的队列(PriorityQueue)可以通过自定义一个类来封装 asyncio.Queue,并在放入元素时根据优先级排序。下面是一个简单的实现方式:实现思路使用 asyncio.Queue 作为底层数据结构。在放入元素时,将元素和优先级一起保存,并在队列中按优先级排序。取出元素时,总是取出优先级最高的元素。由于 asyncio.Queue 本身是先进先出(F

asyncio.Queue 实现带优先级的队列(PriorityQueue)可以通过自定义一个类来封装 asyncio.Queue,并在放入元素时根据优先级排序。下面是一个简单的实现方式:实现思路使用 asyncio.Queue 作为底层数据结构。在放入元素时,将元素和优先级一起保存,并在队列中按优先级排序。取出元素时,总是取出优先级最高的元素。由于 asyncio.Queue 本身是先进先出(F

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

asyncio.Queue 如何实现带优先级的队列(PriorityQueue)

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

✅ 使用 asyncio.PriorityQueue(最简单可靠)

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

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

示例:

import asyncio
<p>async def main():
pq = asyncio.PriorityQueue()</p><pre class="brush:python;toolbar:false;"># 先入低优先级(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
<p>class DeadlinePriorityQueue:
def <strong>init</strong>(self):
self._pq = asyncio.PriorityQueue()</p><pre class="brush:python;toolbar:false;">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学习网公众号。

QQ邮箱隐藏附件图标方法详解QQ邮箱隐藏附件图标方法详解
上一篇
QQ邮箱隐藏附件图标方法详解
腾讯企业邮共享邮箱权限无效解决方法
下一篇
腾讯企业邮共享邮箱权限无效解决方法
查看更多
最新文章
资料下载
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    516次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    500次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    485次学习
查看更多
AI推荐
  • ChatExcel酷表:告别Excel难题,北大团队AI助手助您轻松处理数据
    ChatExcel酷表
    ChatExcel酷表是由北京大学团队打造的Excel聊天机器人,用自然语言操控表格,简化数据处理,告别繁琐操作,提升工作效率!适用于学生、上班族及政府人员。
    4226次使用
  • Any绘本:开源免费AI绘本创作工具深度解析
    Any绘本
    探索Any绘本(anypicturebook.com/zh),一款开源免费的AI绘本创作工具,基于Google Gemini与Flux AI模型,让您轻松创作个性化绘本。适用于家庭、教育、创作等多种场景,零门槛,高自由度,技术透明,本地可控。
    4583次使用
  • 可赞AI:AI驱动办公可视化智能工具,一键高效生成文档图表脑图
    可赞AI
    可赞AI,AI驱动的办公可视化智能工具,助您轻松实现文本与可视化元素高效转化。无论是智能文档生成、多格式文本解析,还是一键生成专业图表、脑图、知识卡片,可赞AI都能让信息处理更清晰高效。覆盖数据汇报、会议纪要、内容营销等全场景,大幅提升办公效率,降低专业门槛,是您提升工作效率的得力助手。
    4464次使用
  • 星月写作:AI网文创作神器,助力爆款小说速成
    星月写作
    星月写作是国内首款聚焦中文网络小说创作的AI辅助工具,解决网文作者从构思到变现的全流程痛点。AI扫榜、专属模板、全链路适配,助力新人快速上手,资深作者效率倍增。
    6119次使用
  • MagicLight.ai:叙事驱动AI动画视频创作平台 | 高效生成专业级故事动画
    MagicLight
    MagicLight.ai是全球首款叙事驱动型AI动画视频创作平台,专注于解决从故事想法到完整动画的全流程痛点。它通过自研AI模型,保障角色、风格、场景高度一致性,让零动画经验者也能高效产出专业级叙事内容。广泛适用于独立创作者、动画工作室、教育机构及企业营销,助您轻松实现创意落地与商业化。
    4836次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码