当前位置:

首页 > 编程开发 > asyncio.Queue 实现带优先级的队列(PriorityQueue)可以通过自定义一个类来包装 asyncio.Queue,并在入队时根据优先级排序。下

asyncio.Queue 实现带优先级的队列(PriorityQueue)可以通过自定义一个类来包装 asyncio.Queue,并在入队时根据优先级排序。下

推荐直接使用asyncio.PriorityQueue实现优先级队列,它是标准库中线程安全、协程友好的内置类,基于heapq自动排序,支持await,优先级数值越小越先执行;避免手动维护排序。

本文目录

    推荐直接使用 asyncio.PriorityQueue 实现优先级队列,它是标准库中线程安全、协程友好的内置类,基于 heapq 自动排序,支持 await,优先级数值越小越先执行;避免手动维护排序。

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

    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() 后检查并跳过已失效项
    本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
    作者最新文章
    编程开发
    相关文章 更多
    opencode安装skills教程及本地配置方法
    opencode安装skills教程及本地配置方法

    遇到OpenCode Skills无法加载或配置无效?本文手把手教你完成Skills的安装、环境变量设置及本地调试,避开常见坑点,让AI助手更懂你的代码习惯。

    如何开启Visual Studio2022实时代码分析警告提示
    如何开启Visual Studio2022实时代码分析警告提示

    解决 Visual Studio 2022 编辑时不显示代码分析警告的问题。指南涵盖 .NET 项目实时分析设置、Roslyn 分析器范围调整及错误列表来源筛选,确保开发者能高效利用实时代码诊断功能。

    如何在Visual Studio2022查看变量监视窗口调试程序
    如何在Visual Studio2022查看变量监视窗口调试程序

    详解 Visual Studio 2022 调试技巧:包括断点设置、监视窗口开启方法、变量添加步骤以及通过 F10 单步执行观察变量值变化的全过程。

    Visual Studio2017设置默认编码为UTF-8不带BOM
    Visual Studio2017设置默认编码为UTF-8不带BOM

    针对 Visual Studio 2017 缺乏全局默认编码选项的情况,详解如何利用 .editorconfig 设定项目级 UTF-8 无 BOM 规则,以及手动转换现有文件并校验字节的方法。

    Visual Studio2022F5调试提示启动项目未配置怎么修复
    Visual Studio2022F5调试提示启动项目未配置怎么修复

    解决 Visual Studio 2022 F5 调试报错问题,通过设置启动项目、检查解决方案属性及正确区分类库与应用项目,快速恢复调试功能。

    Visual Studio2022NuGet包还原失败离线处理方案
    Visual Studio2022NuGet包还原失败离线处理方案

    解决Visual Studio 2022离线环境中NuGet包还原失败的难题,涵盖依赖链分析、本地源配置、包集收集及隔离验证步骤,帮助开发者构建稳定的离线开发环境。

    Visual Studio2017大项目加载缓慢,优化加载性能技巧
    Visual Studio2017大项目加载缓慢,优化加载性能技巧

    解决Visual Studio 2017大型项目加载慢的实用指南。涵盖性能监测、扩展管理、文档恢复优化及版本兼容性检查,帮助开发者精准定位瓶颈并提升工作效率。

    Visual Studio2022设置多启动项目同时调试运行
    Visual Studio2022设置多启动项目同时调试运行

    详解 Visual Studio 2022 多启动项目配置方法,包括启动动作选择、顺序调整及调试技巧,帮助开发者高效调试分布式或前后端分离应用。

    fastapi下载文件接口实现教程
    fastapi下载文件接口实现教程

    深入解析在FastAPI框架中实现文件下载接口的完整方案,涵盖基于FileResponse的本地静态文件传输、基于StreamingResponse的分块流式传输、RFC 5987中文文件名编码规范、路径穿越安全防御及BackgroundTasks临时文件清理,并配备详细代码示例与架构解析图。

    码上飞 codeflying 平台功能介绍
    码上飞 codeflying 平台功能介绍

    了解码上飞CodeFlying如何利用自然语言生成应用。通过物品借用登记实例,学习需求解析、代码生成、测试优化及对话修改的核心操作流程。

    查看更多
    精品专题 更多
    装机必备
    装机必备

    正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

    Windows
    Windows

    正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

    PDF教程
    PDF教程

    正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

    Mac软件 更多
    讯飞智作 网页版
    讯飞智作 网页版
    Windows/ macOS/ Linux

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    火山引擎 网页版
    火山引擎 网页版
    Windows/ macOS/ Linux

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    Aperty macOS版
    Aperty macOS版
    macOS

    Aperty 是 Skylum 推出的 AI 人像修图软件,支持自然肤质修饰、瑕疵处理、面部与身体调整、预设及批量同步。

    WINDOWS 更多
    讯飞智作 网页版
    讯飞智作 网页版
    Windows/ macOS/ Linux

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    火山引擎 网页版
    火山引擎 网页版
    Windows/ macOS/ Linux

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    PixTV 网页版
    PixTV 网页版
    Windows

    PixTV 是一款在浏览器中使用的 AI 视觉内容创作平台,提供图片、视频、音频、文字生成、AI 分镜、局部重绘、提示词反推和无限画布等工具,适合短剧、漫剧、广告及社交媒体内容制作。