Loco-RS 任务调度器新增启动时执行任务功能解析

背景介绍

Loco-RS 是一个 Rust 语言的 Web 框架，其内置的任务调度系统(Scheduler)为开发者提供了定时执行任务的便捷功能。在实际开发中，开发者经常遇到这样的需求：某些定时任务不仅需要在固定周期运行，还需要在应用启动时立即执行一次。

问题分析

传统的任务调度配置通常只支持设置执行周期，比如"每24小时执行一次"。但很多业务场景下，我们希望：

1. 应用启动后立即执行一次任务
2. 然后按照设定的周期继续执行

例如：

• 数据缓存预热
• 系统初始化检查
• 启动时数据同步

解决方案

Loco-RS 通过 YAML 配置文件新增了 run_on_start 配置项，完美解决了这个问题。开发者可以明确指定哪些任务需要在调度器启动时立即执行。

配置示例

scheduler:
  output: stdout
  jobs:
    write_content:
      shell: true
      run: "echo loco >> ./scheduler.txt"
      schedule: run every 1 second
      output: silent
      tags: ['base', 'infra']

    run_task:
      run: "foo"
      schedule: "at 10:00 am"

    list_users:
      run: "user_report"
      shell: true
      schedule: "* 2 * * * *"
      tags: ['base', 'users']
      
run_on_start:
  - write_content
  - list_users

技术实现原理

1. 配置解析：调度器启动时会解析 YAML 配置
2. 任务初始化：识别 run_on_start 下列出的任务名称
3. 立即执行：在调度器初始化完成后，立即执行这些任务
4. 周期调度：然后按照各自的 schedule 配置进行周期性调度

优势对比

方案	优点	缺点
手动执行	简单直接	需要额外操作，容易遗漏
before_start 钩子	自动执行	会影响 worker/server 启动
run_on_start	配置清晰，精准控制	需要框架支持

最佳实践建议

1. 关键任务标记：将系统初始化等关键任务标记为 run_on_start
2. 性能考虑：启动时执行的任务不宜过多或耗时过长
3. 错误处理：确保启动任务有完善的错误处理机制
4. 日志记录：详细记录启动任务的执行情况

总结

Loco-RS 的这项功能增强使得任务调度更加灵活和实用，特别适合需要初始化执行的场景。通过简单的 YAML 配置，开发者可以轻松实现"立即执行+周期执行"的任务模式，大大提升了开发效率和系统可靠性。