借用嵌套lambda中变量不够长的借用检查错误

时间:2015-11-09 00:46:39

标签: rust closures lifetime borrow-checker

我在嵌套的lambda中遇到错误。

let rows = vec![
    vec![3, 6, 2, 8, 9, 0],
    vec![0, 0, 1, 4, 5, 1],
];

let pair_sums = rows.iter()
    .flat_map(|row| {
        (0 ..= row.len()).map(|i| row[i] + row[i + 1])
    })
    .collect::<Vec<_>>();

println!("{:?}", pair_sums);
error[E0597]: `row` does not live long enough
  --> src/main.rs:9:40
   |
9  |             (0..row.len() - 1).map(|i| row[i] + row[i + 1])
   |                                    --- ^^^ does not live long enough
   |                                    |
   |                                    capture occurs here
10 |         })
   |         - borrowed value only lives until here
11 |         .collect::<Vec<_>>();
   |                            - borrowed value needs to live until here

我可以理解为什么会这样,我可以通过将row的值线程化到内部lambda来解决它:

let pair_sums = rows.iter()
    .flat_map(|row| { 
        (0 ..= row.len()).zip(iter::repeat(row))
            .map(|(i, row)| row[i] + row[i + 1])
    })
    .collect::<Vec<_>>();

这太可怕了,不是最好的解决方案。如何在父作用域中引用变量而不必显式传递它们?

1 个答案:

答案 0 :(得分:10)

这里的技巧是闭包如何捕获它们的变量:如果闭包的内容允许它们,它们将接受对它们的引用,而不考虑它们的使用方式,以保持闭包表达式的局部推断和可预测性。在这种情况下,row变量只能通过引用使用,因此可以通过引用捕获它;也就是说,传递给map的闭包对象包含对row的引用。因此,该对象不能离开声明row变量(即flat_map的闭包)的范围,因为该引用将指向无效的内存。返回.map(closure)将违反此规则,因为.map创建了一个存储闭包的惰性迭代器,并且仅在请求元素时调用它。

此处的修复是强制row变量不被引用捕获,因此闭包可以离开作用域。这可以使用move关键字:

来完成
let pair_sums = rows.iter()
    .flat_map(|row| { 
        (0..row.len() - 1)
            .map(move |i| row[i] + row[i + 1])
    })
    .collect::<Vec<_>>();

换句话说,原始代码等同于:

let pair_sums = rows.iter()
    .flat_map(|row: &Vec<i32>| { 
        let row_ref: &&Vec<i32> = &row;
        (0..row.len() - 1)
            .map(move |i| (*row_ref)[i] + (*row_ref)[i + 1])
    })
    .collect::<Vec<_>>();

(我的Finding Closure in Rust帖子更详细地了解了闭包,Rust book也是如此。)