Java stream flatMap with keeping first and second level objects in stream

Question

I was wondering how I could use Java Stream API to flatten a structure where I have an object and a collection of the same type objects nested in it.

What I mean is that I have a Component class which has a field of type List<Component>. What I would like to do is find a neat, stream solution that would the the same as the following code (I need to end up with a list of all components and nested subcomponents).

List<Component> components = getComponents(id); List<Component> componentsAndSubcomponents = new ArrayList<>(); for (Component component : components) { componentsAndSubcomponents.add(component); componentsAndSubcomponents.addAll(component.getSubComponents()); }

Eran · Accepted Answer · 2019-11-27 14:11:49Z

You can use flatMap with Stream concatenation:

List<Component> componentsAndSubcomponents = components.stream() .flatMap(c -> Stream.concat(Stream.of(c),c.getSubComponents().stream())) .collect(Collectors.toList());

This will map each Component into a Stream that contains that Component followed by all of its sub-components, and flatten all these Streams into a flat Stream<Component>, to be collected into a List<Component>.

ernest_k · Accepted Answer · 2019-11-27 13:55:23Z

An easy solution is to create an inner stream on the fly, as in:

List<Component> result = components.stream() .flatMap(comp -> Stream.concat(Stream.of(comp), comp.getSubComponents().stream())) .collect(Collectors.toList());

Collectives™ on Stack Overflow

Java stream flatMap with keeping first and second level objects in stream

2 Answers 2

Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Related