如何为迭代器创建 rust 过滤谓词？

Question

pub struct S{
    a: u32
}
fn main() {
    let ve = vec![S{a: 1}, S{a: 2}];
    ve.iter().filter(filter_function);
}

fn filter_function(s: &S) -> bool {
    todo!()
}

给

error[E0631]: type mismatch in function arguments
   --> src/main.rs:6:22
    |
6   |     ve.iter().filter(filter_function);
    |               ------ ^^^^^^^^^^^^^^^ expected signature of `for<'r> fn(&'r &S) -> _`
    |               |
    |               required by a bound introduced by this call
...
9   | fn filter_function(s: &S) -> bool {
    | --------------------------------- found signature of `for<'r> fn(&'r S) -> _`
    |
note: required by a bound in `filter`

https://play.rust-lang.org/?version=stable&mode=debug&edition=2021&gist=caf542348c2b744abace8e7b6b30d766

我不知道为什么。 签名是一样的。 怎么了？

Answer 1

签名不太一样。 预期的类型是 function ，它采用&'r &S ，而您的 function 采用&'r S 。 一个有一个额外的间接层。

您需要一个 function 采用双引用并使用单引用调用filter_function 。 我们可以通过显式取消引用来做到这一点。

ve.iter().filter(|x| filter_function(*x));

不过，一般来说，Rust 对 lambda 中这类解引用问题的梳理还是不错的，所以我们可以简单地写

ve.iter().filter(|x| filter_function(x));

Rust 足够聪明，可以为我们处理双重参考。

值得一提的是，额外的 lambda 应该不会影响性能。 filter采用通用参数，这意味着它将被静态分派，因此优化器很容易内联。 所以性能应该相当于拥有一个顶级的function。