Java Stream How to - Truncate a stream with limit








Question

We would like to know how to truncate a stream with limit.

Answer

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
//www . j  a v  a 2s.c om
public class Main {

  public static void main(String[] args) throws Exception{
    List<Dish> menu =
        Arrays.asList( new Dish("pork", false, 800, Dish.Type.MEAT),
                       new Dish("beef", false, 700, Dish.Type.MEAT),
                       new Dish("chicken", false, 400, Dish.Type.MEAT),
                       new Dish("rice", true, 350, Dish.Type.OTHER),
                       new Dish("pizza", true, 550, Dish.Type.OTHER),
                       new Dish("prawns", false, 400, Dish.Type.FISH),
                       new Dish("salmon", false, 450, Dish.Type.FISH));
    // Truncating a stream
    List<Dish> dishesLimit3 =
        menu.stream()
            .filter(d -> d.getCalories() > 300)
            .limit(3)
            .collect(Collectors.toList());

    dishesLimit3.forEach(System.out::println);

  }

}

class Dish {

  private final String name;
  private final boolean vegetarian;
  private final int calories;
  private final Type type;

  public Dish(String name, boolean vegetarian, int calories, Type type) {
      this.name = name;
      this.vegetarian = vegetarian;
      this.calories = calories;
      this.type = type;
  }

  public String getName() {
      return name;
  }

  public boolean isVegetarian() {
      return vegetarian;
  }

  public int getCalories() {
      return calories;
  }

  public Type getType() {
      return type;
  }

  public enum Type { MEAT, FISH, OTHER }

  @Override
  public String toString() {
      return name;
  }


}

The code above generates the following result.