你如何将Rust函数作为参数传递？

Question

你如何将Rust函数作为参数传递？

166

我能把函数当做参数传递吗？如果不能，有什么好的替代方案吗？

我尝试了一些不同的语法，但都没有找到正确的。我知道我可以这样做：

fn example() {
    let fun: fn(value: i32) -> i32;
    fun = fun_test;
    fun(5i32);
}

fn fun_test(value: i32) -> i32 {
    println!("{}", value);
    value
}

但这不是将函数作为参数传递给另一个函数：

fn fun_test(value: i32, (some_function_prototype)) -> i32 {
    println!("{}", value);
    value
}

- Angel Angel

2个回答

60

Fn，FnMut和FnOnce是闭包类型，这些内容在其他答案中已经解释过。它们是那些闭合其作用域的函数类型。

除了传递闭包之外，Rust还支持传递简单（非闭包）函数，例如：

fn times2(value: i32) -> i32 {
    2 * value
}

fn fun_test(value: i32, f: fn(i32) -> i32) -> i32 {
    println!("{}", f (value));
    value
}

fn main() {
    fun_test (2, times2);
}

fn(i32) -> i32 这里是一个函数指针类型。

如果您不需要一个完整的闭包，那么使用函数类型通常更简单，因为它不必处理那些闭包生命周期的细节。

- ArtemGr

@IvanTemchenko 也许可以试试这段代码：https://play.rust-lang.org/?version=stable&mode=debug&edition=2018&gist=41d2aa2c947dbebe539e5effc7310ccd - ArtemGr

这并不是我想要的 =) 找到了解决方法，返回动态闭包以捕获self的状态，因此我不需要传递实例引用... - Ivan Temchenko

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- rodrigo · Accepted Answer

当然可以：

fn fun_test(value: i32, f: &dyn Fn(i32) -> i32) -> i32 {
    println!("{}", f(value));
    value
}

fn times2(value: i32) -> i32 {
    2 * value
}

fn main() {
    fun_test(5, &times2);
}

由于这是 Rust，你必须考虑到闭包的所有权和生命周期。

简而言之，基本上有三种类型的闭包（可调用对象）：

Fn：它不能修改捕获的对象。
FnMut：它可以修改捕获的对象。
FnOnce：最受限制的。只能被调用一次，因为当它被调用时，它会消耗自己和它的捕获物。

请参阅闭包何时实现 Fn、FnMut 和 FnOnce？以获取更多详细信息。

如果你使用像指向函数的简单指针这样的闭包，则捕获设置为空，且你具有 Fn 类型。

如果你想要做更高级的操作，则必须使用 lambda 函数。

在Rust中，有适当的函数指针，与C语言中的指针类似。它们的类型例如为fn(i32) -> i32。 Fn(i32) -> i32，FnMut(i32) -> i32和FnOnce(i32) -> i32实际上是特征（traits）。函数指针始终实现这三个特征，但是Rust还具有闭包，可以将其转换为指针（取决于捕获集是否为空），但它们确实实现了其中一些特征。

因此，例如，上面的示例可以扩展：

fn fun_test_impl(value: i32, f: impl Fn(i32) -> i32) -> i32 {
    println!("{}", f(value));
    value
}
fn fun_test_dyn(value: i32, f: &dyn Fn(i32) -> i32) -> i32 {
    println!("{}", f(value));
    value
}
fn fun_test_ptr(value: i32, f: fn(i32) -> i32) -> i32 {
    println!("{}", f(value));
    value
}

fn times2(value: i32) -> i32 {
    2 * value
}

fn main() {
    let y = 2;
    //static dispatch
    fun_test_impl(5, times2);
    fun_test_impl(5, |x| 2*x);
    fun_test_impl(5, |x| y*x);
    //dynamic dispatch
    fun_test_dyn(5, &times2);
    fun_test_dyn(5, &|x| 2*x);
    fun_test_dyn(5, &|x| y*x);
    //C-like pointer to function
    fun_test_ptr(5, times2);
    fun_test_ptr(5, |x| 2*x); //ok: empty capture set
    fun_test_ptr(5, |x| y*x); //error: expected fn pointer, found closure
}