Rust异步并行获取项目：返回引用数据生命周期问题求助

问题场景 #

我需要并行执行异步代码，等待所有任务完成后合并所有结果返回。写了如下并行实现，但遇到了所有权相关的编译错误：

pub async fn get_projects(organizations: &Vec<Organization>) -> Vec<Project> {
    let mut result = Vec::new();
    let mut tasks = Vec::new();

    for organization in organizations.iter() {
        if !organization.is_pat_valid {
            continue;
        }
        tasks.push(repositories::devops::get_projects(
            &organization.pat,
            &organization.name,
        ));
    }

    let task_results = join_all(tasks).await;

    for task_result in task_results.iter() {
        match task_result {
            Ok(projects) => result.extend(projects),
            Err(error) => println!("Error: {}", error),
        }
    }    

    return result;
}

具体错误 #

处理任务结果时，projects的类型是&Project而非Project，无法直接加入Vec<Project>；如果将result改为Vec<&Project>，编译器会报错：

return result;
   |   ^^^^^^ returns a value referencing data owned by the current function

原串行实现可正常运行但性能较差：

pub async fn get_projects(organizations: Vec<Organization>) -> Vec<Project> {
    let mut result: Vec<Project> = vec![];
    for organization in organizations.iter() {
        if (!organization.is_pat_valid) {
            continue;
        }
        let organization_projects =
            repositories::devops::get_projects(&organization.pat, &organization.name).await;
        match organization_projects {
            Ok(projects) => {
                result.extend(projects);
            }
            Err(error) => {
                println!("Error: {}", error);
            }
        }
    }
    return result;
}

问题根源 #

1. 用task_results.iter()遍历拿到的是引用：&Result<Vec<Project>, E>，因此Ok(projects)中的projects是&Vec<Project>，调用extend时会尝试插入&Project，与Vec<Project>类型不匹配。
2. 若改为Vec<&Project>，这些引用指向的是task_results内部的Vec<Project>，但task_results会在函数结束时被销毁，导致引用悬空，触发编译器的生命周期检查错误。

解决方案 #

核心是获取任务结果的所有权，而非引用：

pub async fn get_projects(organizations: &Vec<Organization>) -> Vec<Project> {
    let mut tasks = Vec::new();

    for organization in organizations.iter() {
        if !organization.is_pat_valid {
            continue;
        }
        tasks.push(repositories::devops::get_projects(
            &organization.pat,
            &organization.name,
        ));
    }

    let task_results = join_all(tasks).await;
    let mut result = Vec::new();

    // 直接遍历task_results，获取每个Result的所有权
    for task_result in task_results {
        match task_result {
            // projects是Vec<Project>的所有权，extend会转移所有元素到result中
            Ok(projects) => result.extend(projects),
            Err(error) => println!("Error: {}", error),
        }
    }

    result
}

原理说明 #

• 去掉iter()后，task_result是Result<Vec<Project>, E>的完整所有权，Ok(projects)中的projects是独立的Vec<Project>。
• Vec::extend方法会将projects中的每个Project元素的所有权转移到result中，最终result拥有所有元素的所有权，符合返回值Vec<Project>的要求，不会出现悬空引用问题。

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