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

Question

我在嵌套的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

我可以理解为什么会发生这种情况,我可以通过将rowthrough 的值线程化到内部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<_>>();

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

Answer 1

这里的技巧是闭包如何捕获它们的变量:如果闭包的内容允许它们,它们将接受对它们的引用,而不考虑它们的使用方式,以保持闭包表达式的局部推断和可预测性.在这种情况下,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<_>>();

(我在Rust帖子中发现的关闭更详细地挖掘了闭包,就像Rust的书一样.)