如何手动实现一个具备“优先级调度”能力的简易版异步任务执行引擎

云磊同学_8126

云磊同学_8126

2026-04-15

712人浏览

原创

可直接用 asyncio.priorityqueue 搭建轻量可控的优先级任务引擎:将任务封装为 (priority, coro) 元组入队,由永驻调度协程循环取高优任务 await 执行,并通过 task_done()、异常捕获、超时控制及取消令牌实现健壮调度。

如何手动实现一个具备“优先级调度”能力的简易版异步任务执行引擎

可以直接用 asyncio.PriorityQueue 搭建一个轻量、可控的优先级任务引擎,核心是把“任务”封装成带优先级的协程,并由一个长期运行的调度协程按序取出执行。

用 PriorityQueue 构建任务分发中心

Python 的 asyncio.PriorityQueue 是线程安全、协程友好的优先队列,数值越小优先级越高。它天然适合作为任务入口:所有待执行的协程都以 (priority, coro) 元组形式入队。

  • 初始化一个全局队列:queue = asyncio.PriorityQueue()
  • 插入任务时指定优先级,例如:await queue.put((1, fetch_user()))(高优)、await queue.put((5, write_log()))(低优)
  • 注意:不能直接 put 协程对象,必须 await 它——所以实际应传入已 awaitable 的协程调用,如 fetch_user() 而非 fetch_user

写一个永驻调度协程负责执行

这个协程持续监听队列,每次取最高优先级任务并 await 执行,执行完标记完成。它不退出,靠 queue.join() 控制生命周期。

Soofy
Soofy

一款通过AI对话帮助用户学习语言的在线学习工具,将语言练习融入真实交流场景。

下载
  • 使用 while True 循环 + await queue.get() 阻塞等待新任务
  • 拿到 (priority, coro) 后,先 try/finally 包裹,确保无论成功失败都调用 queue.task_done()
  • 避免单个任务崩溃导致整个调度器停止,可加 except Exception 记录错误但不中断循环

启动任务与控制执行流

任务不是自动运行的,需要显式创建调度协程,并用 create_task 启动;外部通过 put 注入任务,用 join 等待全部结束。

  • 在 main() 中调用 asyncio.create_task(dispatcher(queue)) 启动调度器
  • 用 asyncio.gather() 或 asyncio.wait_for() 管理多个调度器实例(如按业务域隔离)
  • 若需暂停/恢复,可在调度协程中监听一个 asyncio.Event,await event.wait() 实现条件阻塞

扩展支持取消和超时

真实场景中,高优任务可能需要抢占或限时执行。可以在调度协程中加入超时控制,或对任务本身做包装。

  • 执行前用 asyncio.wait_for(coro, timeout=2.0) 限制最大耗时
  • 为任务附加取消令牌:传入 asyncio.CancelledError 捕获逻辑,或用 asyncio.shield() 保护关键步骤不被意外取消
  • 若需支持“取消排队中任务”,PriorityQueue 本身不提供 remove 接口,可改用自定义堆 + 标记删除,或换用 heapq + asyncio.Lock 手动维护

相关专题

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

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

2023.06.20

4986

5

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

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

2023.07.04

1229

3

js四舍五入
js四舍五入

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

2023.07.04

4924

6

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

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

2023.09.01

960

4

JavaScript转义字符
JavaScript转义字符

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

2023.09.04

1876

5

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

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

2023.09.04

3525

4

如何启用JavaScript
如何启用JavaScript

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

2023.09.12

4573

6

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

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

2023.09.20

3000

5

java访问控制修饰符介绍
java访问控制修饰符介绍

java访问控制修饰符有四种,分别是public、protected、private、默认访问修饰符。详细介绍:1、public,public是最宽松的访问控制修饰符,被修饰的类、方法和变量可以被任何其他类访问,当一个类、方法或变量被声明为public时,它们可以在任何地方被访问,无论是同一个包中的类还是不同包中的类;2、protected修饰符等等。

2023.09.20

948

7

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Swoole手册
Swoole手册

共0课时 | 0人学习