Rust 如何在嵌套map中使用闭包

发布于06月05日

我试图从一个创建每个元素的函子中生成一个二维矩阵，并将其存储为一个平面Vec(每行连接).

我使用嵌套map(实际上是flat_map和嵌套map)来创建每一行并将其连接起来.Here is what I tried:

fn make<T, F>(n: usize, m: usize, f: F) -> Vec<T>
where
    F: Fn(usize, usize) -> T,
{
    (0..m).flat_map(|y| (0..n).map(|x| f(x, y))).collect()
}

fn main() {
    let v = make(5, 5, |x, y| x + y);

    println!("{:?}", v);
}

不幸的是，我在编译过程中遇到了一个错误:

error[E0597]: `y` does not live long enough
 --> src/main.rs:5:45
  |
5 |     (0..m).flat_map(|y| (0..n).map(|x| f(x, y))).collect()
  |                                    ---      ^ -          - borrowed value needs to live until here
  |                                    |        | |
  |                                    |        | borrowed value only lives until here
  |                                    |        borrowed value does not live long enough
  |                                    capture occurs here

如何在嵌套映射中使用闭包？我研究了这个问题，但我仍然对答案感兴趣.

推荐答案

在您的例子中，内部闭包|x| f(x,y)是borrow 闭包，它通过引用获取其环境(y和f).

按照.flat_map(..)的工作方式，它禁止您保留对y的引用，这不是外部范围.因此，我们需要让你的闭包按价值来衡量它的环境，这对于y是usize是Copy不是一个问题:

(0..m).flat_map(|y| (0..n).map(move |x| f(x, y))).collect()

然而，现在又出现了另一个问题:

error[E0507]: cannot move out of captured outer variable in an `FnMut` closure
 --> src/main.rs:5:36
  |
1 | fn make<T, F>(n: usize, m: usize, f: F) -> Vec<T>
  |                                   - captured outer variable
...
5 |     (0..m).flat_map(|y| (0..n).map(move |x| f(x,y))).collect()
  |                                    ^^^^^^^^ cannot move out of captured outer variable in an `FnMut` closure

在这里，我们试图将f也移动到闭包中，这是绝对不可能的(除非m是1，但编译器无法知道).

既然f是Fn(usize, usize) -> T，我们也可以显式地将&引用传递给它，&引用就是Copy:

fn make<T, F>(n: usize, m: usize, f: F) -> Vec<T>
where
    F: Fn(usize, usize) -> T,
{
    let f_ref = &f;
    (0..m)
        .flat_map(|y| (0..n).map(move |x| f_ref(x, y)))
        .collect()
}

在本例中，闭包按值获取其环境，该环境由y和f_ref组成，两者都是Copy，一切正常.