convert a list objects with list to map in Java 8 by stream

Question

I have the following classes:

class A {
   String id;
   List<B> b;
}

class B {
   String id;
}

And I have a list of a's that I would like to convert to a map with the following logic:

List<A> aList;

Map<String, String> map; 

for (A a:aList)
{
   for (B b:aList.b)
   {
      map.put(b.id, a.id)
   }
}

What is the best way to do it in one line of stream method?

Thanks Elad

score 2 · Answer 1 · answered Jun 14 '20 at 17:23

You can stream the each list of B class from A class object, and then collect the id combinations into Map.Entry objects

Map<String,String> map = aList.stream()
                              .flatMap(al->al.getB()
                                             .stream()
                                             .map(bl-new AbstractMap.SimpleEntry<String, String>(bl.getId(), al.getId())))
                              .collect(Collectors.toMap(Map.Entry::getKey,Map.Entry::getValue));

Thanks @Deadpool! Actually you can use Map.entry instead of new AbstractMap.SimpleEntry — Elad Cohen, Jun 14 '20 at 18:10

convert a list objects with list to map in Java 8 by stream

1 Answers1