Question

我有以下代码（playground）：

struct A {
    pub vec: Vec<u64>,
}

impl A {
    fn perform_for_all<F: Fn(&mut u64)>(&mut self, f: F) {
        for mut i in &mut self.vec {
            f(i);
        }
    }
}
fn main() {
    let mut a = A {
        vec: vec![1, 3, 44, 2, 4, 5, 6],
    };

    let mut done = false;

    a.perform_for_all(|v| {
        println!("value: {:?}", v);
        done = true;
    });

    if !done {
        a.perform_for_all(|v| {
            println!("value {:?}", v);
        });
    }
}

发生以下错误：

error[E0594]: cannot assign to `done`, as it is a captured variable in a `Fn` closure
  --> src/main.rs:21:9
   |
21 |         done = true;
   |         ^^^^^^^^^^^ cannot assign
   |
help: consider changing this to accept closures that implement `FnMut`
  --> src/main.rs:19:23
   |
19 |       a.perform_for_all(|v| {
   |  _______________________^
20 | |         println!("value: {:?}", v);
21 | |         done = true;
22 | |     });
   | |_____^

我有一个已加载对象的列表和一个数据库中的对象列表。我需要一个函数，它接受一个闭包并在加载的对象上执行它，如果我们没有列表中的对象，则在数据库的对象列表上执行它。

该功能如下：

pub fn perform_for_match_with_mark<F>(&mut self, mark: MatchMark, f: F)
where
    F: Fn(&mut GameMatch),
{
    self.perform_for_all_matches(
        |m| {
            // runtime list
            if let Game::Match(ref mut gm) = *m {
                if gm.match_stamp().mark == mark {
                    f(gm);
                }
            }
        },
        None,
    );
    // if we have called `f` above - don't execute lines below.
    let tx = self.match_tx.clone();
    GamesDatabase::perform_for_match_with_mark(mark, |ms| {
        // database
        self.perform_for_all_matches(
            |m| {
                if let Game::Match(ref gm) = *m {
                    if gm.match_stamp().id == ms.id {
                        f(&mut GameMatch::new_with_match_stamp(
                            tx.clone(),
                            ms.clone(),
                            gm.needs_server_set,
                            gm.server_id,
                        ))
                    }
                }
            },
            None,
        );
    });
}

只有当我们无法在运行时列表中找到它们时，我们才必须对数据库中的对象进行操作。这就是为什么我决定制作一个变量，说明＆＃34;我们已经在列表中找到了这些对象，只留下数据库＆＃34;。

Answer 1

将您的Guest Operations功能改为使用FnMut而不是Fn：

perform_for_all

As Peter said，有一些编译器魔法正在进行。

Fn::call的签名是：

fn perform_for_all<F>(&mut self, mut f: F)
where
    F: FnMut(&mut u64),
{
    for mut i in &mut self.vec {
        f(&mut i);
    }
}

这会引用extern "rust-call" fn call(&self, args: Args) -> Self::Output的不可变引用，这就是为什么你不能修改任何捕获的变量。

FnMut::call_mut的签名允许您改变变量，因为它需要self：

&mut self

通过将关闭从extern "rust-call" fn call_mut(&mut self, args: Args) -> Self::Output更改为Fn，允许它修改其捕获的变量，前提是您传递给它的引用是可变的。

Answer 2

只是为了扩大对SplittyDev的回答。

当你使用一个闭包时，编译器会让闭包在它的环境中访问变量。实际上，它将创建一个新结构，其成员是您尝试访问的变量。

这不完全是这个（它实际上没有编译），但它在概念上是一个合理的近似值：

struct Closure_1 {
    done: bool
}

impl FnMut<&mut u64> for Closure_1 {
    fn call_mut(&mut self, v: &mut u64) {
        println!("value: {:?}", v);                                                                 
        self.done = true;         
    }
}

当您调用它时，这些变量将被借用或复制（或者如果您使用move关键字，则会被移动。）

let mut c1 = Closure_1 { done : done };
a.perform_for_all(|v| c1.call(&v)); 
done = c1.done;

当闭包修改其环境时，它不能是Fn，因为它还必须改变变量本身：

impl Fn<&mut u64> for Closure_1 {
    fn call(&self, v: &mut u64) {
        println!("value: {:?}", v);                                                                 
        self.done = true; // Can't do this because self is not a mutable ref
    }
}

有关详细信息，请参阅 Rust编程语言 section on closures and their environment。

如何从Rust中的Fn闭包内部更改变量？

2 个答案: