Java Stream How to - Map int to Custom Object and store in List

• Back to Stream Map  ↑

Question

We would like to know how to map int to Custom Object and store in List.

Answer

//from  www.  ja  v  a 2 s .c  o  m
import java.util.ArrayList;
import java.util.List;
import java.util.stream.IntStream;

public class Main {

  public static void main(String[] args) throws Exception {

    List<Foo> foos = new ArrayList<>();

    IntStream
        .range(1, 4)
        .forEach(num -> foos.add(new Foo("Foo" + num)));

    foos.forEach(f ->
        IntStream
            .range(1, 4)
            .forEach(num -> f.bars.add(new Bar("Bar" + num + " <- " + f.name))));

    foos.stream()
        .flatMap(f -> f.bars.stream())
        .forEach(b -> System.out.println(b.name));
  }

  static class Foo {
    String name;
    List<Bar> bars = new ArrayList<>();

    Foo(String name) {
      this.name = name;
    }
  }

  static class Bar {
    String name;

    Bar(String name) {
      this.name = name;
    }
  }
}

The code above generates the following result.

• Back to Stream Map  ↑