impl Trait

如果函数返回实现了 MyTrait 的类型,可以将其返回类型编写为 -> impl MyTrait。这可以大大简化你的类型签名!

  1. use std::iter;
  2. use std::vec::IntoIter;
  3. // 该函数组合了两个 `Vec <i32>` 并在其上返回一个迭代器。
  4. // 看看它的返回类型多么复杂!
  5. fn combine_vecs_explicit_return_type(
  6. v: Vec<i32>,
  7. u: Vec<i32>,
  8. ) -> iter::Cycle<iter::Chain<IntoIter<i32>, IntoIter<i32>>> {
  9. v.into_iter().chain(u.into_iter()).cycle()
  10. }
  11. // 这是完全相同的函数,但其返回类型使用 `impl Trait`。
  12. // 看看它多么简单!
  13. fn combine_vecs(
  14. v: Vec<i32>,
  15. u: Vec<i32>,
  16. ) -> impl Iterator<Item=i32> {
  17. v.into_iter().chain(u.into_iter()).cycle()
  18. }
  19. fn main() {
  20. let v1 = vec![1, 2, 3];
  21. let v2 = vec![4, 5];
  22. let mut v3 = combine_vecs(v1, v2);
  23. assert_eq!(Some(1), v3.next());
  24. assert_eq!(Some(2), v3.next());
  25. assert_eq!(Some(3), v3.next());
  26. assert_eq!(Some(4), v3.next());
  27. assert_eq!(Some(5), v3.next());
  28. println!("all done");
  29. }

更重要的是,某些 Rust 类型无法写出。例如,每个闭包都有自己未命名的具体类型。在使用 impl Trait 语法之前,必须在堆上进行分配才能返回闭包。但是现在你可以像下面这样静态地完成所有操作:

  1. // 返回一个将输入和 `y` 相加的函数
  2. fn make_adder_function(y: i32) -> impl Fn(i32) -> i32 {
  3. let closure = move |x: i32| { x + y };
  4. closure
  5. }
  6. fn main() {
  7. let plus_one = make_adder_function(1);
  8. assert_eq!(plus_one(2), 3);
  9. }

您还可以使用 impl Trait 返回使用 mapfilter 闭包的迭代器!这使得使用 mapfilter 更容易。因为闭包类型没有名称,所以如果函数返回带闭包的迭代器,则无法写出显式的返回类型。但是有了 impl Trait,你就可以轻松地做到这一点:

```rust,editable fn double_positives<’a>(numbers: &’a Vec) -> impl Iterator + ‘a { numbers .iter() .filter(|x| x > &&0) .map(|x| x * 2) }