Rocketry项目教程：运行时控制与任务调度管理-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00664/article/details/148578375

Rocketry项目教程：运行时控制与任务调度管理

rocketry Modern scheduling library for Python 项目地址: https://gitcode.com/gh_mirrors/ro/rocketry

前言

在现代任务调度系统中，动态控制运行时行为是核心需求之一。Rocketry作为一个功能强大的Python任务调度库，提供了丰富的运行时控制能力。本文将深入探讨如何在Rocketry项目中实现运行时控制，包括调度会话管理、任务动态操作以及任务队列实现等高级功能。

调度会话(Session)控制

获取Session实例

在Rocketry中，所有运行时操作都通过session对象进行。获取session有两种推荐方式：

通过依赖注入方式获取（推荐）：

from rocketry.args import Session

@app.task()
def do_things(session=Session()):
    # 使用session进行操作
    ...

直接通过应用实例获取：

@app.task()
def do_things():
    session = app.session
    # 使用session进行操作
    ...

第一种方式更适合大型应用，因为它遵循了依赖注入原则，使代码更易于测试和维护。

关键会话操作

重启调度器

当需要重新加载所有任务和配置时，可以调用重启方法：

@app.task()
def restart_scheduler(session=Session()):
    session.restart()
    print("调度器已成功重启")

关闭调度器

安全关闭调度器的实现方式：

@app.task()
def graceful_shutdown(session=Session()):
    session.shut_down()
    print("调度器正在优雅关闭...")

任务动态管理

运行时任务操作

移除任务

动态移除不再需要的任务：

@app.task()
def remove_old_task(session=Session()):
    try:
        session.remove_task("obsolete_task")
        print("任务已成功移除")
    except KeyError:
        print("指定任务不存在")

创建任务

动态创建新任务的示例：

from rocketry.conds import hourly

def new_operation():
    print("执行新操作")

@app.task()
def add_dynamic_task(session=Session()):
    session.create_task(
        func=new_operation,
        name="dynamic_operation",
        start_cond=hourly
    )
    print("新任务已动态添加")

任务状态监控

获取任务实例

可以通过两种方式获取任务实例：

通过Task参数注入：

from rocketry.args import Task

@app.task()
def monitor_task(target_task=Task("do_things")):
    print(f"任务状态: {target_task.status}")

通过session获取：

@app.task()
def check_task_status(session=Session()):
    task = session["critical_task"]
    print(f"最后执行时间: {task.last_run}")

强制运行任务

有时需要手动触发任务执行：

@app.task()
def run_task_manually(session=Session()):
    task = session["backup_task"]
    if not task.is_running:
        task.run()
        print("任务已手动启动")

高级任务队列实现

Rocketry支持构建复杂的任务队列系统，包括参数化任务执行。以下是实现示例：

from rocketry.args import Session
import asyncio

def generate_tasks():
    """任务生成器"""
    yield 'data_processing', {'priority': 'high'}
    yield 'report_generation', {'format': 'pdf'}
    yield 'cleanup', {'scope': 'temp'}

@app.task(on_startup=True, execution="async")
async def task_processor(session=Session()):
    for task_name, params in generate_tasks():
        task = session[task_name]
        
        # 执行任务并传入参数
        task.run(**params)
        
        # 等待任务完成
        while task.is_running:
            await asyncio.sleep(1)
        
        if not task.status == "success":
            print(f"任务 {task_name} 执行失败")
            break
            
    print("所有队列任务处理完成")