如何使用Rust的Tokio创建自生成任务？

要把原来的循环逻辑改成自生成任务的形式，核心思路是将循环内的操作封装为异步函数，在函数执行完核心逻辑后，通过tokio::spawn启动一个新的自身任务副本，以此实现持续重复执行的效果。

改造后的代码示例 #

use tokio::time::Duration;
use std::sync::Arc;

// 请根据实际场景替换YourHandleType为你的句柄类型
async fn repeat_task(handle_clone: Arc<YourHandleType>) {
    // 执行原循环内的核心逻辑
    if let Ok(mut handle) = handle_clone.driver_only.try_lock() {
        handle.park_timeout(&handle_clone.driver, Duration::from_millis(0));
    }

    // 在当前任务结束前，生成自身的任务副本并提交给Tokio调度器
    tokio::spawn(repeat_task(handle_clone));
}

// 启动初始任务
// let your_handle_arc = Arc::new(your_handle_instance);
// tokio::spawn(repeat_task(your_handle_arc));

关键注意点 #

• 共享资源安全：必须用Arc（或其他线程安全的共享容器）包裹handle_clone，因为每个新生成的任务都需要持有该资源的引用，且Tokio任务是运行在多线程环境下的。
• 错误处理：原代码中的try_lock().unwrap()容易触发panic，建议换成if let Ok(...)的方式优雅处理锁获取失败的场景，避免程序崩溃。
• 任务调度差异：循环是在同一个Tokio任务内重复执行，而自生成任务会每次创建新任务，Tokio调度器会对这些任务进行独立调度，这种差异可能会影响性能或执行顺序，需要结合你的设计需求确认是否符合预期。
• 任务停止机制：如果需要主动停止任务，可以引入一个线程安全的停止标记（比如Arc<AtomicBool>），在函数开头检查标记状态，若需停止则不再生成下一个任务。

内容的提问来源于stack exchange，提问作者Jainil Patel