How do I return a Filter iterator from a function?

by

Rust 1.26

fn filter_one(input: &[u8]) -> impl Iterator<Item = &u8> {
    input.iter().filter(|&&x| x == 1)
}

fn main() {
    let nums = vec![1, 2, 3, 1, 2, 3];
    let other: Vec<_> = filter_one(&nums).collect();
    println!("{:?}", other);
}

Rust 1.0

fn filter_one<'a>(input: &'a [u8]) -> Box<Iterator<Item = &'a u8> + 'a> {
    Box::new(input.iter().filter(|&&x| x == 1))
}

fn main() {
    let nums = vec![1, 2, 3, 1, 2, 3];
    let other: Vec<_> = filter_one(&nums).collect();
    println!("{:?}", other);
}

This solution requires additional allocation. We create a boxed trait object. Here, the size of the object is always known (it’s just a pointer or two), but the size of the object in the heap does not need to be known.

As Vladimir Matveev points out, if your predicate logic doesn’t need any information from the environment, you can use a function instead of a closure:

use std::{iter::Filter, slice::Iter};

fn filter_one<'a>(input: &'a [u8]) -> Filter<Iter<u8>, fn(&&u8) -> bool> {
    fn is_one(a: &&u8) -> bool {
        **a == 1
    }

    input.iter().filter(is_one)
}

fn main() {
    let nums = vec![1, 2, 3, 1, 2, 3];
    let other: Vec<_> = filter_one(&nums).collect();
    println!("{:?}", other);
}

See also:

More Related Contents:

  1. Writing a generic function that takes an iterable container as parameter in Rust
  2. “Expected type parameter” error in the constructor of a generic struct
  3. How do I require a generic type implement an operation like Add, Sub, Mul, or Div in a generic function?
  4. How to write a trait bound for adding two references of a generic type?
  5. How do I use integer number literals when using generic types?
  6. How do I write the lifetimes for references in a type constraint when one of them is a local reference?
  7. How can I avoid a ripple effect from changing a concrete struct to generic?
  8. The trait cannot be made into an object
  9. Why does a generic method inside a trait require trait object to be sized?
  10. Conflicting implementations of trait in Rust
  11. Matching a generic parameter to an associated type in an impl
  12. Why can’t I use a key function that returns a reference when sorting a vector with sort_by_key?
  13. Why can’t `&(?Sized + Trait)` be cast to `&dyn Trait`?
  14. Generics Error: expected type parameter, found struct
  15. Why can’t I add a blanket impl on a trait with a type parameter?
  16. Can I avoid eager ambiguity resolution for trait implementations with generics?
  17. What does the question mark mean in a type parameter bound?
  18. What does it mean to instantiate a Rust generic with an underscore?
  19. How do I unwrap an arbitrary number of nested Option types?
  20. Unable to create a polymorphic type because the trait cannot be made into an object
  21. What is the cited problem with using generic type parameters in trait objects?
  22. Scala: Abstract types vs generics
  23. Iterator returning items by reference, lifetime issue
  24. How do I get the minimum or maximum value of an iterator containing floating point numbers?
  25. Is there an owned version of String::chars?
  26. What is an idiomatic way to collect an iterator of &T into a collection of Ts?
  27. Difference between “*” and “Any” in Kotlin generics
  28. How do I perform iterator computations over iterators of Results without collecting to a temporary vector?
  29. Returning iterator of a Vec in a RefCell
  30. How to call a named constructor from a generic function in Dart/Flutter

Leave a Comment