如何在Rust Rocket框架中运行周期性任务执行数据库待办作业

实现方案 #

你可以通过Rocket的Ignite生命周期Fairing拿到初始化完成的数据库连接池，再将周期性任务spawn到Tokio运行时后台执行，具体实现如下：

1. 实现自定义定时任务Fairing

use rocket::{fairing::{Fairing, Info, Kind}, Rocket, Build};
use std::time::Duration;
use tokio::time::interval;

pub struct CronJobFairing;

#[rocket::async_trait]
impl Fairing for CronJobFairing {
    fn info(&self) -> Info {
        Info {
            name: "Periodic Job Runner",
            kind: Kind::Ignite,
        }
    }

    async fn on_ignite(&self, rocket: Rocket<Build>) -> Result<Rocket<Build>, Rocket<Build>> {
        // 从已初始化的Rocket实例中获取数据库连接池
        let pool = DbConn::pool(&rocket)
            .expect("数据库连接池未初始化，请检查数据库配置");
        
        // 后台spawn周期性任务，不阻塞Rocket启动
        tokio::spawn(async move {
            // 可自定义执行间隔，示例为60秒执行一次
            let mut interval = interval(Duration::from_secs(60));
            loop {
                interval.tick().await;
                // 从连接池获取连接执行待处理任务
                if let Ok(conn) = pool.get().await {
                    let _ = conn.run(|c| {
                        // 此处实现你的业务逻辑：扫描数据库待执行的作业并处理
                        process_pending_block_jobs(c);
                    }).await;
                }
                // 可自行添加错误捕获、日志打印逻辑
            }
        });

        Ok(rocket)
    }
}

2. 注册Fairing到Rocket实例

修改你的main函数，挂载上面实现的定时任务Fairing即可：

#[rocket::main]
async fn main() {
    rocket::build()
        .attach(DbConn::fairing())
        .attach(CronJobFairing) // 新增此行挂载定时任务
        .mount(
            "/v1",
            routes![ post_block ],
        )
        .launch()
        .await
        .expect("could not launch rocket");
}

补充说明 #

• 如果需要支持cron表达式类的复杂调度规则，可以替换tokio::time::interval为对应三方调度库，核心获取连接池、后台spawn的逻辑不变。
• 若周期性任务需要共享其他业务状态，可以通过rocket.manage(状态实例)将状态存入Rocket的全局状态管理，在on_ignite回调中通过rocket.state::<状态类型>()即可获取。
• 建议为后台任务添加统一的panic捕获、错误日志逻辑，避免任务异常退出影响服务稳定性。

内容的提问来源于stack exchange，提问作者Matthias Wimmer