Java 8 Nested (Multi level) group by

by

You can’t group a single item by multiple keys, unless you accept the item to potentially appear in multiple groups. In that case, you want to perform a kind of flatMap operation.

One way to achieve this, is to use Stream.flatMap with a temporary pair holding the combinations of Item and SubItem before collecting. Due to the absence of a standard pair type, a typical solution is to use Map.Entry for that:

Map<T, Map<V, List<SubItem>>> result = pojo.getItems().stream()
    .flatMap(item -> item.subItems.stream()
        .map(sub -> new AbstractMap.SimpleImmutableEntry<>(item.getKey1(), sub)))
    .collect(Collectors.groupingBy(AbstractMap.SimpleImmutableEntry::getKey,
                Collectors.mapping(Map.Entry::getValue,
                    Collectors.groupingBy(SubItem::getKey2))));

An alternative not requiring these temporary objects would be performing the flatMap operation right in the collector, but unfortunately, flatMapping won’t be there until Java 9.

With that, the solution would look like

Map<T, Map<V, List<SubItem>>> result = pojo.getItems().stream()
    .collect(Collectors.groupingBy(Item::getKey1,
                Collectors.flatMapping(item -> item.getSubItems().stream(),
                    Collectors.groupingBy(SubItem::getKey2))));

and if we don’t want to wait for Java 9 for that, we may add a similar collector to our code base, as it’s not so hard to implement:

static <T,U,A,R> Collector<T,?,R> flatMapping(
    Function<? super T,? extends Stream<? extends U>> mapper,
    Collector<? super U,A,R> downstream) {

    BiConsumer<A, ? super U> acc = downstream.accumulator();
    return Collector.of(downstream.supplier(),
        (a, t) -> { try(Stream<? extends U> s=mapper.apply(t)) {
            if(s!=null) s.forEachOrdered(u -> acc.accept(a, u));
        }},
        downstream.combiner(), downstream.finisher(),
        downstream.characteristics().toArray(new Collector.Characteristics[0]));
}

More Related Contents:

  1. How to force max to return ALL maximum values in a Java Stream?
  2. Retrieving a List from a java.util.stream.Stream in Java 8
  3. Java 8 stream map to list of keys sorted by values
  4. Java 8 Distinct by property
  5. Java 8 List into Map
  6. How can I throw CHECKED exceptions from inside Java 8 streams?
  7. Java 8 lambdas, Function.identity() or t->t
  8. Copy a stream to avoid “stream has already been operated upon or closed”
  9. Java 8: performance of Streams vs Collections
  10. Using Java 8’s Optional with Stream::flatMap
  11. Java 8 Streams: multiple filters vs. complex condition
  12. Lambda expression to convert array/List of String to array/List of Integers
  13. How do I keep the iteration order of a List when using Collections.toMap() on a stream?
  14. Filter Java Stream to 1 and only 1 element
  15. Java 8, Streams to find the duplicate elements
  16. How can I get a List from some class properties with Java 8 Stream?
  17. Why doesn’t java.util.Collection implement the new Stream interface?
  18. Does Java SE 8 have Pairs or Tuples?
  19. How to use a Java8 lambda to sort a stream in reverse order?
  20. Java 8 Collectors.toMap SortedMap
  21. Why does Stream.allMatch() return true for an empty stream?
  22. Java 8 grouping using custom collector?
  23. What Java 8 Stream.collect equivalents are available in the standard Kotlin library?
  24. How can I throw CHECKED exceptions from inside Java 8 lambdas/streams?
  25. How to apply multiple predicates to a java.util.Stream?
  26. Parallel stream from a HashSet doesn’t run in parallel
  27. Picking elements of a list until condition is met with Java 8 Lambdas
  28. Non-interference exact meaning in Java 8 streams
  29. In Java 8 how do I transform a Map to another Map using a lambda?
  30. Group sequences of values

Leave a Comment