Collect List<Object> to Map<String, List<Object>> in Java 8 using stream

Question

I am trying to do this iteration using java 8 stream instead of traditional for loop.

I have a List of Object List<Obligation> and the Obligation is having List of String within.

Obligation contains fields like

f1 and f2 of type String,

and

f3 of type List<String>

What I want to achieve is Map<String, List<Obligation>> such that, String used in key of Map is each value present in List<String> from each Obligation within List<Obligation>

This is how the code would look in traditional for loop:

// newly initialized Map
Map<String, List<Obligation>> map = new LinkedHashMap<String, List<Obligation>>();

// assume the list is populated
List<Obligation> obligations = someMethod();

                for(Obligation obligation : obligations) {
                    for(String license : obligation.getLicenseIDs()) {
                        if(map.containsKey(license)) {
                            map.get(license).add(obligation);
                        } else {
                            List<Obligation> list = new ArrayList<Obligation>();
                            list.add(obligation);
                            map.put(license, list);
                        }
                    }
                }

The question is well explained, good efforts from a newbie here. — Vishwa Ratna
– Vishwa Ratna, Commented Jul 17, 2019 at 12:34

Eran · Accepted Answer · 2019-07-17 05:41:38Z

5

Use flatMap to create a stream of pairs of Obligation and license IDs, and then use groupingBy to collect them by license ID.

Map<String, List<Obligation>> map =
    obligations.stream() // Stream<Obligation>
               .flatMap(o -> o.getLicenseIDs()
                              .stream()
                              .map(id -> new SimpleEntry<>(o,id))) // Stream<Map.Entry<Obligation,String>>
               .collect(Collectors.groupingBy(Map.Entry::getValue,
                                              Collectors.mapping(Map.Entry::getKey,
                                                                 Collectors.toList())));

edited Jul 17, 2019 at 5:41

answered Jul 17, 2019 at 5:36

Eran

395k57 gold badges726 silver badges793 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Collect List<Object> to Map<String, List<Object>> in Java 8 using stream

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related