Tokio中使用JoinSet/spawn时scylla生命周期不足问题

问题代码 #

pub async fn foobar(scylla: Data<scylla::Session>) {
    let mut queries = tokio::task::JoinSet::new();
    let task = scylla.query("hello", &[]);
    queries.spawn(task);
    queries.join_next().await;
}

编译器错误 #

`scylla` does not live long enough
borrowed value does not live long enough (rustc) Click for full compiler diagnostic
post.rs(46, 1): `scylla` dropped here while still borrowed

原因分析 #

scylla.query()返回的Future会借用传入的scylla实例，但JoinSet::spawn（包括tokio::spawn）要求任务的Future必须是'static生命周期——编译器无法静态保证异步任务会在当前函数结束前执行完毕，因此不允许任务持有对函数局部变量/参数的借用，否则可能出现悬空引用。

你之前尝试克隆无效，大概率是没把克隆后的实例move进异步任务块，导致任务仍然借用原scylla实例。

解决方案 #

scylla::Session内部基于Arc实现，Data<Session>的克隆是轻量的引用计数操作，只需克隆实例并通过async move将其移入任务，让任务持有自己的引用即可：

修正后的单任务代码 #

pub async fn foobar(scylla: Data<scylla::Session>) {
    let mut queries = tokio::task::JoinSet::new();
    
    // 轻量克隆Session实例
    let scylla_clone = scylla.clone();
    // 将克隆实例move进async块，避免借用原实例
    let task = async move {
        scylla_clone.query("hello", &[]).await
    };
    
    queries.spawn(task);
    queries.join_next().await;
}

多查询并行示例 #

如果需要同时发起多个查询，重复克隆并生成任务即可：

pub async fn foobar(scylla: Data<scylla::Session>) {
    let mut queries = tokio::task::JoinSet::new();
    
    // 并行发起3个查询
    for sql in ["SELECT * FROM table1", "SELECT * FROM table2", "SELECT * FROM table3"] {
        let scylla_clone = scylla.clone();
        let sql_clone = sql.to_string();
        let task = async move {
            scylla_clone.query(&sql_clone, &[]).await
        };
        queries.spawn(task);
    }
    
    // 逐个处理任务结果
    while let Some(task_res) = queries.join_next().await {
        match task_res {
            Ok(query_res) => match query_res {
                Ok(rows) => println!("查询结果: {:?}", rows),
                Err(e) => eprintln!("查询失败: {}", e),
            },
            Err(e) => eprintln!("任务执行异常: {}", e),
        }
    }
}

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